github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[FRONTEND] switch absolute imports to relative imports in Triton (#1773)

Browse Source
This commit is contained in:
Izzy Putterman
2023-06-14 16:59:24 -07:00
committed by GitHub
parent 700b3cfd6e
commit 71e21f5797
18 changed files with 209 additions and 173 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/triton/testing.py
View File

@@ -4,7 +4,7 @@ import subprocess
import sys
from contextlib import contextmanager
import triton._C.libtriton.triton as _triton
from ._C.libtriton.triton import runtime
def nvsmi(attrs):
@@ -281,7 +281,7 @@ def get_dram_gbps(backend=None, device=None):
from .runtime import driver
if not backend:
backend = _triton.runtime.backend.CUDA
backend = runtime.backend.CUDA
if not device:
device = torch.cuda.current_device()
mem_clock_khz = driver.utils.get_device_properties(device)["mem_clock_rate"] # in kHz
@@ -295,7 +295,7 @@ def get_max_tensorcore_tflops(dtype, backend=None, device=None, clock_rate=None)
from .runtime import driver
if not backend:
backend = _triton.runtime.backend.CUDA
backend = runtime.backend.CUDA
if not device:
device = torch.cuda.current_device()
@@ -398,7 +398,7 @@ def get_max_simd_tflops(dtype, backend=None, device=None):
from .runtime import driver
if not backend:
backend = _triton.runtime.backend.CUDA
backend = runtime.backend.CUDA
if not device:
device = torch.cuda.current_device()