def test_mxnet_inference_telemetry_gpu(mxnet_inference, ec2_connection,
gpu_only, ec2_instance_type):
if test_utils.is_image_incompatible_with_instance_type(
mxnet_inference, ec2_instance_type):
pytest.skip(
f"Image {mxnet_inference} is incompatible with instance type {ec2_instance_type}"
)
execute_ec2_inference_test(ec2_connection, mxnet_inference,
MX_TELEMETRY_CMD)
def test_pytorch_inference_torchaudio_cpu(pytorch_inference, ec2_connection,
cpu_only):
_, image_framework_version = get_framework_and_version_from_tag(
pytorch_inference)
if Version(image_framework_version) not in SpecifierSet(">=1.9,!=1.10.*"):
pytest.skip(
"torchaudio is installed in PT 1.9 and above, except PT 1.10.*")
execute_ec2_inference_test(ec2_connection, pytorch_inference,
PT_TORCHAUDIO_CMD)
def test_pytorch_inference_telemetry_gpu(pytorch_inference, ec2_connection,
gpu_only, ec2_instance_type,
pt15_and_above_only):
if test_utils.is_image_incompatible_with_instance_type(
pytorch_inference, ec2_instance_type):
pytest.skip(
f"Image {pytorch_inference} is incompatible with instance type {ec2_instance_type}"
)
execute_ec2_inference_test(ec2_connection, pytorch_inference,
PT_TELEMETRY_CMD)
def test_pytorch_inference_torchdata_cpu(pytorch_inference, ec2_connection,
cpu_only, pt111_and_above_only):
_, image_framework_version = get_framework_and_version_from_tag(
pytorch_inference)
if Version(image_framework_version) in SpecifierSet("==1.11.*"):
execute_ec2_inference_test(ec2_connection, pytorch_inference,
PT_TORCHDATA_DEV_CMD)
else:
execute_ec2_inference_test(ec2_connection, pytorch_inference,
PT_TORCHDATA_CMD)
def test_pytorch_inference_torchdata_gpu(pytorch_inference, ec2_connection,
gpu_only, ec2_instance_type,
pt111_and_above_only):
_, image_framework_version = get_framework_and_version_from_tag(
pytorch_inference)
if test_utils.is_image_incompatible_with_instance_type(
pytorch_inference, ec2_instance_type):
pytest.skip(
f"Image {pytorch_inference} is incompatible with instance type {ec2_instance_type}"
)
execute_ec2_inference_test(ec2_connection, pytorch_inference,
PT_TORCHDATA_CMD)
def test_pytorch_inference_torchaudio_gpu(pytorch_inference, ec2_connection,
gpu_only, ec2_instance_type):
_, image_framework_version = get_framework_and_version_from_tag(
pytorch_inference)
if Version(image_framework_version) not in SpecifierSet(">=1.9,!=1.10.*"):
pytest.skip(
"torchaudio is installed in PT 1.9 and above, except PT 1.10.*")
if test_utils.is_image_incompatible_with_instance_type(
pytorch_inference, ec2_instance_type):
pytest.skip(
f"Image {pytorch_inference} is incompatible with instance type {ec2_instance_type}"
)
execute_ec2_inference_test(ec2_connection, pytorch_inference,
PT_TORCHAUDIO_CMD)
def test_mxnet_inference_telemetry_cpu(mxnet_inference, ec2_connection,
cpu_only):
execute_ec2_inference_test(ec2_connection, mxnet_inference,
MX_TELEMETRY_CMD)
def test_pytorch_inference_telemetry_gpu(pytorch_inference, ec2_connection,
gpu_only):
execute_ec2_inference_test(ec2_connection, pytorch_inference,
PT_TELEMETRY_CMD)
def test_pytorch_inference_telemetry_graviton_cpu(pytorch_inference_graviton,
ec2_connection, cpu_only):
execute_ec2_inference_test(ec2_connection, pytorch_inference_graviton,
PT_TELEMETRY_CMD)
def test_pytorch_inference_telemetry_cpu(pytorch_inference, ec2_connection,
cpu_only, pt15_and_above_only):
execute_ec2_inference_test(ec2_connection, pytorch_inference,
PT_TELEMETRY_CMD)
def test_pytorch_inference_torchdata_cpu(pytorch_inference, ec2_connection,
cpu_only, pt111_and_above_only):
_, image_framework_version = get_framework_and_version_from_tag(
pytorch_inference)
execute_ec2_inference_test(ec2_connection, pytorch_inference,
PT_TORCHDATA_CMD)
