Skip to content

Commit

Permalink
Added HPTT tests.
Browse files Browse the repository at this point in the history
  • Loading branch information
alexnick83 committed Jul 11, 2023
1 parent f272b05 commit ba77a1f
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 0 deletions.
1 change: 1 addition & 0 deletions pytest.ini
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ markers =
mpi: Test requires MPI. (select with '-m mpi')
scalapack: Test requires ScaLAPACK (Intel MKL and OpenMPI). (select with '-m scalapack')
datainstrument: Test uses data instrumentation (select with '-m datainstrument')
hptt: Test requires the HPTT library (select with '-m "hptt')
python_files =
*_test.py
*_cudatest.py
Expand Down
11 changes: 11 additions & 0 deletions tests/numpy/transpose_test.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
# Copyright 2019-2021 ETH Zurich and the DaCe authors. All rights reserved.
import numpy as np
import dace
import pytest
from common import compare_numpy_output

M, N = 24, 24
Expand Down Expand Up @@ -45,10 +46,20 @@ def test_transpose():
assert rel_error <= 1e-5


@pytest.mark.hptt
def test_hptt():
with dace.config.set_temporary('library', 'ttranspose', 'default_implementation', value='HPTT'):
test_transpose_axes0()
test_transpose_axes1()
test_transpose_axes2()
test_transpose_none()


if __name__ == '__main__':
test_transpose_axes0()
test_transpose_axes1()
test_transpose_axes2()
test_transpose()
test_transpose_none()
test_transpose_no()
test_hptt()

0 comments on commit ba77a1f

Please sign in to comment.