Fix endinanness on cuda (#353)

risc0 · Jan 27, 2023 · 7995ca6 · 7995ca6
1 parent 98cb897
commit 7995ca6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/risc0/zkp/kernels/cuda/sha256.h b/risc0/zkp/kernels/cuda/sha256.h
@@ -194,6 +194,10 @@ __device__ inline ShaDigest shaHashInner(const T* data, size_t size, size_t stri
     compress(state, words);
   }
 
+  for (size_t i = 0; i < 8; i++) {
+    state.words[i] = impl::convertU32(state.words[i]);
+  }
+
   return state;
 }
 
@@ -224,6 +228,10 @@ __device__ inline ShaDigest shaHashPair(ShaDigest x, ShaDigest y) {
   ShaDigest state = impl::initState();
   impl::compress(state, words);
 
+  for (size_t i = 0; i < 8; i++) {
+    state.words[i] = impl::convertU32(state.words[i]);
+  }
+
   // Return the results
   return state;
 }