DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-05-06 01:54:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix linking error from ODR violation

Browse Source
This commit is contained in:
Lequn Chen 2025-04-05 17:35:04 +00:00
parent c187c23ba8
commit 611e3f659d
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deep_gemm/include/deep_gemm/mma_utils.cuh
View File

@ -888,15 +888,15 @@ struct SM90_U32x4_STSM_N {
} }
}; };
__device__ void warpgroup_arrive() { __forceinline__ __device__ void warpgroup_arrive() {
asm volatile("wgmma.fence.sync.aligned;\n" ::: "memory"); asm volatile("wgmma.fence.sync.aligned;\n" ::: "memory");
} }
__device__ void warpgroup_commit_batch() { __forceinline__ __device__ void warpgroup_commit_batch() {
asm volatile("wgmma.commit_group.sync.aligned;\n" ::: "memory"); asm volatile("wgmma.commit_group.sync.aligned;\n" ::: "memory");
} }
__device__ void warpgroup_fence_operand(float& reg) { __forceinline__ __device__ void warpgroup_fence_operand(float& reg) {
asm volatile("" : "+f"(reg) :: "memory"); asm volatile("" : "+f"(reg) :: "memory");
} }

2
deep_gemm/include/deep_gemm/tma_utils.cuh
View File

@ -40,7 +40,7 @@ constexpr CUtensorMapDataType get_CUtensorMapDataType() {
} }
} }
PFN_cuTensorMapEncodeTiled get_cuTensorMapEncodeTiled() { inline PFN_cuTensorMapEncodeTiled get_cuTensorMapEncodeTiled() {
// Get pointer to `cuTensorMapEncodeTiled` // Get pointer to `cuTensorMapEncodeTiled`
cudaDriverEntryPointQueryResult driver_status; cudaDriverEntryPointQueryResult driver_status;
void* cuTensorMapEncodeTiled_ptr = nullptr; void* cuTensorMapEncodeTiled_ptr = nullptr;