relpath

Signed-off-by: Mayank Mishra <[email protected]>
mayank31398 · Dec 19, 2024 · 7f468d6 · 7f468d6
1 parent 8f70e94
commit 7f468d6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cute_kernels/kernels/swiglu/cuda_implementation/kernels_backward.cu b/cute_kernels/kernels/swiglu/cuda_implementation/kernels_backward.cu
@@ -36,7 +36,7 @@ __global__ void _swiglu_backward_cuda_kernel(const scalar_t *gate,
     static_assert(vector_instruction_width == 1 || vector_instruction_width == 2 || vector_instruction_width == 4 ||
                   vector_instruction_width == 8);
 
-    const int64_t thread_id = get_global_thread_id();
+    const uint64 thread_id = get_global_thread_id();
     using dtype = DType<scalar_t>;
 
     if constexpr (vector_instruction_width == 1) {

diff --git a/cute_kernels/kernels/swiglu/cuda_implementation/kernels_forward.cu b/cute_kernels/kernels/swiglu/cuda_implementation/kernels_forward.cu
@@ -16,7 +16,7 @@ __global__ void _swiglu_forward_cuda_kernel(const scalar_t *gate,
     static_assert(vector_instruction_width == 1 || vector_instruction_width == 2 || vector_instruction_width == 4 ||
                   vector_instruction_width == 8);
 
-    const int64_t thread_id = get_global_thread_id();
+    const uint64 thread_id = get_global_thread_id();
     using dtype = DType<scalar_t>;
 
     if constexpr (vector_instruction_width == 1) {