Skip to content

GitLab

Commit 36bc29ed authored by Tri Dao's avatar Tri Dao
Browse files

Use int64_t instead of uint32_t in kernel_traits.h

parent 000b67f5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
csrc/flash_attn/src/kernel_traits.h
View file @ 36bc29ed
...@@ -24,7 +24,7 @@ struct Flash_kernel_traits { ...@@ -24,7 +24,7 @@ struct Flash_kernel_traits {
#endif #endif
using ElementAccum = float; using ElementAccum = float;
using index_t = uint32_t; using index_t = int64_t;
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 800 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 800
using MMA_Atom_Arch = std::conditional_t< using MMA_Atom_Arch = std::conditional_t<
......
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