zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix compilation error from integral_constant (#2326)

Browse Source
This commit is contained in:
Cheng
2025-07-02 22:04:38 +09:00
committed by GitHub
parent cfb6a244ea
commit e76e9b87f0
4 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/backend/cuda/rms_norm.cu
View File

@@ -320,7 +320,7 @@ void RMSNormVJP::eval_gpu(
constexpr int N_READS = 4;
auto kernel = cu::rms_norm_vjp<
DataType,
has_w_constant(),
has_w_constant.value,
block_dim(),
N_READS>;
kernel<<<n_rows, block_dim(), 0, stream>>>(