gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[FRONTEND] Support alternative install locations of system libdevice.10.bc (#951)

Browse Source
This commit is contained in:
Crutcher Dunnavant
2022-12-05 19:41:44 -08:00
committed by GitHub
parent e419781978
commit 9490252261
4 changed files with 38 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/tests/test_core.py
View File

@@ -12,6 +12,7 @@ import triton
import triton._C.libtriton.triton as _triton
import triton.language as tl
from triton.runtime.jit import JITFunction, TensorWrapper, reinterpret
from tests.libdevice_testutil import system_libdevice_path
int_dtypes = ['int8', 'int16', 'int32', 'int64']
uint_dtypes = ['uint8', 'uint16', 'uint32', 'uint64']
@@ -1552,7 +1553,7 @@ def test_num_warps_pow2():
@pytest.mark.parametrize("dtype_str, expr, lib_path",
[('int32', 'libdevice.ffs', ''),
('float32', 'libdevice.pow', '/usr/local/cuda/nvvm/libdevice/libdevice.10.bc'),
('float32', 'libdevice.pow', system_libdevice_path()),
('float64', 'libdevice.norm4d', '')])
def test_libdevice_tensor(dtype_str, expr, lib_path):