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_ext_elemwise.py
View File

@@ -5,6 +5,7 @@ from torch.testing import assert_close
import triton
import triton.language as tl
from tests.libdevice_testutil import system_libdevice_path
@pytest.mark.parametrize('num_warps, block_size, iter_size', [
@@ -125,7 +126,7 @@ def test_fmad_rn_no_mask(num_warps, block_size, iter_size):
@pytest.mark.parametrize("dtype_str, expr, lib_path",
[('int32', 'libdevice.ffs', '/usr/local/cuda/nvvm/libdevice/libdevice.10.bc'),
[('int32', 'libdevice.ffs', system_libdevice_path()),
('int32', 'libdevice.ffs', '')])
def test_libdevice(dtype_str, expr, lib_path):
src = f"""