Fix conditional use of pinned allocator on devices

Signed-off-by: Joseph Schuchart <[email protected]>
TESSEorg · Aug 14, 2024 · a15f724 · a15f724
1 parent c0cabfa
commit a15f724
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/spmm/spmm.cc b/examples/spmm/spmm.cc
@@ -68,9 +68,8 @@ using blk_t = DeviceTensor<scalar_t, btas::DEFAULT::range,
                            btas::mohndle<btas::varray<scalar_t
 #if HAVE_SPMM_DEVICE
                                                       , TiledArray::device_pinned_allocator<scalar_t>
-#else  // HAVE_SPMM_DEVICE
-                                                      >,
 #endif  // HAVE_SPMM_DEVICE
+                                                      >,
                                          btas::Handle::shared_ptr>>;
 //#include <atomic>
 //static std::atomic<uint64_t> reduce_count = 0;