File tree Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change 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.
2022memory_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 )
You can’t perform that action at this time.
0 commit comments