Home
last modified time | relevance | path

Searched defs:triton (Results 1 – 10 of 10) sorted by relevance

/external/pytorch/
DMakefile53 triton: target
Dmypy.ini160 ignore_missing_imports = True
/external/pytorch/torch/_inductor/
Dasync_compile.py183 def triton(self, kernel_name: str, source_code: str, device_str: str = "cuda"): member in AsyncCompile
Dconfig.py831 class triton: class
/external/pytorch/torch/_inductor/runtime/
Dcoordinate_descent_tuner.py14 triton = None variable
Dtriton_heuristics.py52 triton = None variable
/external/pytorch/torch/_inductor/codegen/
Dcpp_wrapper_cuda.py352 triton=True, argument
Dcpp_wrapper_cpu.py80 triton=True, argument
Dwrapper.py1647 triton=True, argument
Dcommon.py902 triton: Optional[Callable[..., str]] = None variable in OverridesData