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)