Skip to content

GitLab

Commit 5e5b497d authored by zhuwenwen's avatar zhuwenwen
Browse files

skip bf16

parent 7e1d5e53
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
csrc/cache_kernels.cu
View file @ 5e5b497d
...@@ -13,10 +13,10 @@ ...@@ -13,10 +13,10 @@
#include <map> #include <map>
#include <vector> #include <vector>
#ifdef USE_ROCM // #ifdef USE_ROCM
#include <hip/hip_bf16.h> // #include <hip/hip_bf16.h>
typedef __hip_bfloat16 __nv_bfloat16; // typedef __hip_bfloat16 __nv_bfloat16;
#endif // #endif
void swap_blocks( void swap_blocks(
torch::Tensor& src, torch::Tensor& src,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment