Skip to content

Commit 6b18cf8

Browse files
committed
fabtests/pytest/efa: Add ROCr memory type parameters to test fixtures
Signed-off-by: Darryl Abbate <[email protected]>
1 parent 178b48d commit 6b18cf8

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

fabtests/pytest/efa/conftest.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,15 @@
1414
pytest.param("cuda_to_cuda", marks=pytest.mark.cuda_memory),
1515
pytest.param("host_to_neuron", marks=pytest.mark.neuron_memory),
1616
pytest.param("neuron_to_neuron", marks=pytest.mark.neuron_memory),
17+
pytest.param("host_to_rocr", marks=pytest.mark.rocr_memory),
18+
pytest.param("rocr_to_rocr", marks=pytest.mark.rocr_memory),
1719
]
1820

1921
# Add more memory types that are useful for uni-directional tests.
2022
memory_type_list_all = memory_type_list_bi_dir + [
2123
pytest.param("cuda_to_host", marks=pytest.mark.cuda_memory),
2224
pytest.param("neuron_to_host", marks=pytest.mark.neuron_memory),
25+
pytest.param("rocr_to_host", marks=pytest.mark.rocr_memory),
2326
]
2427

2528
@pytest.fixture(scope="module", params=memory_type_list_all)

0 commit comments

Comments
 (0)