blob: df241132c64405bdc03113da1cedc1dd266d4e46 [file] [log] [blame]
import pytest
def pytest_addoption(parser):
parser.addoption(
"--goldentime-rocm-e2e-ms",
action="store",
type=float,
help="Golden time to test benchmark",
)
parser.addoption(
"--goldentime-rocm-unet-ms",
action="store",
type=float,
help="Golden time to test benchmark",
)
parser.addoption(
"--goldentime-rocm-clip-ms",
action="store",
type=float,
help="Golden time to test benchmark",
)
parser.addoption(
"--goldentime-rocm-vae-ms",
action="store",
type=float,
help="Golden time to test benchmark",
)
parser.addoption(
"--goldendispatch-rocm-unet",
action="store",
default=1718,
type=int,
help="Golden dispatch count to test benchmark",
)
parser.addoption(
"--goldendispatch-rocm-clip",
action="store",
default=1571,
type=int,
help="Golden dispatch count to test benchmark",
)
parser.addoption(
"--goldendispatch-rocm-vae",
action="store",
default=250,
type=int,
help="Golden dispatch count to test benchmark",
)
parser.addoption(
"--goldensize-rocm-unet-bytes",
action="store",
default=2088217,
type=int,
help="Golden vmfb size to test benchmark",
)
parser.addoption(
"--goldensize-rocm-clip-bytes",
action="store",
default=785493,
type=int,
help="Golden vmfb size to test benchmark",
)
parser.addoption(
"--goldensize-rocm-vae-bytes",
action="store",
default=762067,
type=int,
help="Golden vmfb size to test benchmark",
)
parser.addoption(
"--rocm-chip",
action="store",
default="gfx90a",
type=str,
help="ROCm target chip configuration of GPU",
)
@pytest.fixture
def goldentime_rocm_e2e(request):
return request.config.getoption("--goldentime-rocm-e2e-ms")
@pytest.fixture
def goldentime_rocm_unet(request):
return request.config.getoption("--goldentime-rocm-unet-ms")
@pytest.fixture
def goldentime_rocm_clip(request):
return request.config.getoption("--goldentime-rocm-clip-ms")
@pytest.fixture
def goldentime_rocm_vae(request):
return request.config.getoption("--goldentime-rocm-vae-ms")
@pytest.fixture
def goldendispatch_rocm_unet(request):
return request.config.getoption("--goldendispatch-rocm-unet")
@pytest.fixture
def goldendispatch_rocm_clip(request):
return request.config.getoption("--goldendispatch-rocm-clip")
@pytest.fixture
def goldendispatch_rocm_vae(request):
return request.config.getoption("--goldendispatch-rocm-vae")
@pytest.fixture
def goldensize_rocm_unet(request):
return request.config.getoption("--goldensize-rocm-unet-bytes")
@pytest.fixture
def goldensize_rocm_clip(request):
return request.config.getoption("--goldensize-rocm-clip-bytes")
@pytest.fixture
def goldensize_rocm_vae(request):
return request.config.getoption("--goldensize-rocm-vae-bytes")
@pytest.fixture
def rocm_chip(request):
return request.config.getoption("--rocm-chip")