Getting ptx from Triton
Published:
You can get the ptx of a triton kernel like so: my_kernel.cache[DEVICE_KEY][INPUTS_KEY].asm['ptx']
, where DEVICE_KEY
and INPUTS_KEY
are determined like below.
Smarties are little nuggets of technical knowhow I picked up during my work. They’re inspired by Simon Willison’s TILs.
Published:
You can get the ptx of a triton kernel like so: my_kernel.cache[DEVICE_KEY][INPUTS_KEY].asm['ptx']
, where DEVICE_KEY
and INPUTS_KEY
are determined like below.
Published:
As of June 13 2024, to get Triton 3.0 you have to install it from source, like so:
Published:
With the tabulate library by Sergey Astanin you can remove a bit of coginitive load (and therefore be more productive) by making output look more beautiful.
Published:
To run jupyter lab on a remote computer, and keep it alive when my local computer sleeps: