JinjieLiu/triton-tvm-ffi
1
0
Fork 0
mirror of https://github.com/sgjzfzzf/triton-tvm-ffi.git synced 2026-07-01 08:51:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

support GetDeviceProperties

Browse Source 
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
This commit is contained in:
Jinjie Liu
2026-01-28 02:16:10 +08:00
parent ab83dded12
commit 3a485166b4
8 changed files with 95 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/triton_tvm_ffi/driver.py
+4 -3
View File
@@ -1,7 +1,8 @@
from __future__ import annotations
from typing import Type
from typing import Mapping, Type
from triton.backends.nvidia.driver import CudaDriver
from .utils import get_device_properties
class TVMFFIUtils(object):
@@ -19,8 +20,8 @@ class TVMFFIUtils(object):
def load_binary(self, *args, **kwargs):
return self._utils.load_binary(*args, **kwargs)
def get_device_properties(self, *args, **kwargs):
return self._utils.get_device_properties(*args, **kwargs)
def get_device_properties(self, device_id: int) -> Mapping[str, int]:
return get_device_properties(device_id)
def cuOccupancyMaxActiveClusters(self, *args, **kwargs):
return self._utils.cuOccupancyMaxActiveClusters(*args, **kwargs)