biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Misc] AMD Build Improvements (#12923)

Browse Source
This commit is contained in:
Shiyan Deng
2025-02-12 02:36:10 -08:00
committed by GitHub
parent 7c4033acd4
commit f1042e86f0
4 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/moe/moe_align_sum_kernels.cu
View File

@@ -3,7 +3,7 @@
#include <c10/cuda/CUDAGuard.h>
#include <ATen/ATen.h>
#include <THC/THCAtomics.cuh>
#include <ATen/cuda/Atomic.cuh>
#include "../cuda_compat.h"
#include "../dispatch_utils.h"