mirror of
https://github.com/compiler-explorer/compiler-explorer.git
synced 2026-07-16 17:26:58 -04:00
1e820f0511
Add [CuTe DSL](https://docs.nvidia.com/cutlass/latest/media/docs/pythonDSL/overview.html) support to compiler-explorer. Tried to follow the existing Triton code, wrapper has similar flags. <img width="2289" height="894" alt="image" src="https://github.com/user-attachments/assets/f4e89fcb-ca66-4d67-aace-e905693f0889" /> Infra: https://github.com/compiler-explorer/infra/pull/2123 --------- Co-authored-by: Matt Godbolt <matt@godbolt.org> Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
21 lines
339 B
Python
21 lines
339 B
Python
import cutlass
|
|
import cutlass.cute as cute
|
|
|
|
|
|
@cute.kernel
|
|
def hello_world_kernel():
|
|
tidx, _, _ = cute.arch.thread_idx()
|
|
if tidx == 0:
|
|
cute.printf("Hello world from CuTe DSL")
|
|
|
|
|
|
@cute.jit
|
|
def hello_world():
|
|
hello_world_kernel().launch(
|
|
grid=(1, 1, 1),
|
|
block=(32, 1, 1),
|
|
)
|
|
|
|
|
|
cute.compile(hello_world)
|