mirror of
https://github.com/ROCm/ROCm.git
synced 2026-02-21 03:00:39 -05:00
858a2f0a5e
Simple mechanism to run Triton kernels on PyTorch for debugging purpose (upstream from Kernl). Todo: - random grid iteration - support of atomic ops - more unit tests - cover new APIs?
10 lines
150 B
Python
10 lines
150 B
Python
from typing import Tuple
|
|
|
|
import dataclasses
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class ExecutionContext:
|
|
program_id: Tuple[int]
|
|
program_size: Tuple[int]
|