diff --git a/deep_ep/include/deep_ep/common/ptx.cuh b/deep_ep/include/deep_ep/common/ptx.cuh index ac8b1489..dc29546b 100644 --- a/deep_ep/include/deep_ep/common/ptx.cuh +++ b/deep_ep/include/deep_ep/common/ptx.cuh @@ -107,7 +107,7 @@ __forceinline__ __device__ void st_bulk(void* smem_ptr) { if (elect_one_sync()) { asm volatile("st.bulk.weak.shared::cta [%0], %1, 0;\n" :: "r"(static_cast(__cvta_generic_to_shared(smem_ptr))), - "r"(kNumBytes) + "l"(static_cast(kNumBytes)) : "memory"); } #else