triton-lang / triton

Development repository for the Triton language and compiler
https://triton-lang.org/
MIT License
12.09k stars 1.44k forks source link

Is there a sample program about interpreter mode ? #3596

Open guojilei opened 3 months ago

guojilei commented 3 months ago

Hi I want to run triton on cpu, i saw "That refers to the CPU support, which won't happen shortly, but you can offload triton to pytorch using the interpreter mode. https://github.com/openai/triton/commit/858a2f0a5ed42784bedcaa4bcec98efec420db45", but i cannot find a sample 。Is there a sample program about interpreter mode?

Jokeren commented 3 months ago

https://drive.google.com/drive/folders/1bKpvz1NiBL_fHrGhMoZPvQfXCeetV2iY

Jokeren commented 3 months ago

Will work on a doc page soon

yiliu30 commented 2 months ago

@Jokeren That's really cool! Could you point me to the first PR that supports this? I want to start by understanding the mental model behind it.