tensorplay.backends.triton
Classes
class TritonBackend [source]
python
TritonBackend()Methods
__init__(self) [source]
Initialize self. See help(type(self)) for accurate signature.
compile(self, graph, example_inputs) [source]
class TritonCodegen [source]
python
TritonCodegen(graph)Generates Triton kernels aligned with PyTorch Inductor's style:
- Explicit variable naming (in_ptr0, tmp0, out_ptr0)
- XBLOCK based tiling
- Autotuning configs
- Robust masking
Methods
__init__(self, graph) [source]
Initialize self. See help(type(self)) for accurate signature.
generate(self) [source]
class TritonScheduler [source]
python
TritonScheduler(graph)