1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
import pytest
@pytest.fixture(scope="module", params=["host_to_host",
pytest.param("host_to_cuda", marks=pytest.mark.cuda_memory),
pytest.param("cuda_to_host", marks=pytest.mark.cuda_memory),
pytest.param("cuda_to_cuda", marks=pytest.mark.cuda_memory)])
def memory_type(request):
return request.param
@pytest.fixture(scope="module", params=["r:0,4,64",
"r:4048,4,4148",
"r:8000,4,9000",
"r:17000,4,18000",
"r:0,1024,1048576"])
def message_size(request):
return request.param
|
