skip bf16

5e5b497d · zhuwenwen · 7e1d5e53 · 5e5b497d
Commit 5e5b497d authored Feb 19, 2024 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

csrc/cache_kernels.cu csrc/cache_kernels.cu +4 -4

No files found.
--- a/csrc/cache_kernels.cu
+++ b/csrc/cache_kernels.cu
@@ -13,10 +13,10 @@
 #include <map>
 #include <vector>

-#ifdef USE_ROCM
-  #include <hip/hip_bf16.h>
-  typedef __hip_bfloat16 __nv_bfloat16;
-#endif
+// #ifdef USE_ROCM
+//   #include <hip/hip_bf16.h>
+//   typedef __hip_bfloat16 __nv_bfloat16;
+// #endif

 void swap_blocks(
  torch::Tensor& src,