Re:CubeCL: GPU Kernels in Rust for CUDA, ROCm, and WGPU
by syl20bnr on Wednesday April 23, 2025 @11:54PM
(#43779118)
It also compiles directly to MSL, it is just missing from the post title.
[ Parent
| on HN ]
Re:CubeCL: GPU Kernels in Rust for CUDA, ROCm, and WGPU
by adastra22 on Thursday April 24, 2025 @01:27AM
(#43779524)
No it compiles indirectly through wgpu, which means it doesn’t have access to any Metal extensions not exposed by the wgpu interface.
Re:CubeCL: GPU Kernels in Rust for CUDA, ROCm, and WGPU
by syl20bnr on Thursday April 24, 2025 @07:28AM
(#43781416)
I am the coder of the MSL dialect for the CubeCL CPP compiler. Since 0.5 release it directly compiles to MSL and support simdgroup matrix functions for instance. It does use wgpu for the runtime but without naga as we added msl pass through to wgpu just for this.
Re:CubeCL: GPU Kernels in Rust for CUDA, ROCm, and WGPU
by adastra22 on Thursday April 24, 2025 @10:37AM
(#43783362)
You should update the README.
Re:CubeCL: GPU Kernels in Rust for CUDA, ROCm, and WGPU
by syl20bnr on Thursday April 24, 2025 @01:40PM
(#43785381)
You are right we just released Burn and updated its readme, we were not thinking that CubeCL would be the one that could be featured. ^_^
Re:CubeCL: GPU Kernels in Rust for CUDA, ROCm, and WGPU
by grovesNL on Thursday April 24, 2025 @08:01AM
(#43781672)
wgpu has some options to access backend-specific types and shader passthrough (i.e., you provide your own shader for a backend directly).
Generally wgpu is open to supporting any Metal extensions you need. There's usually an analogous extension in one of the other backends (e.g., Vulkan, DX12) anyway.