fix: properly use GPU if possible

filecoin-project · Oct 26, 2021 · f8b6533 · f8b6533
1 parent a42a933
commit f8b6533
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/storage-proofs-porep/src/stacked/vanilla/proof.rs b/storage-proofs-porep/src/stacked/vanilla/proof.rs
@@ -824,7 +824,9 @@ impl<'a, Tree: 'static + MerkleTreeTrait, G: 'static + Hasher> StackedDrg<'a, Tr
         start: usize,
         end: usize,
     ) -> Result<TreeRElementData<Tree>> {
-        if SETTINGS.use_gpu_tree_builder {
+        if SETTINGS.use_gpu_tree_builder
+            && TypeId::of::<Tree::Hasher>() == TypeId::of::<PoseidonHasher>()
+        {
             use fr32::bytes_into_fr;
 
             let mut layer_bytes = vec![0u8; (end - start) * std::mem::size_of::<Fr>()];