feature: Update instance types for integ test (#2881)

Author: jeniyat

tests/conftest.py

@@ -363,7 +363,7 @@ def cpu_instance_type(sagemaker_session, request):
 
 @pytest.fixture(scope="module")
 def gpu_instance_type(request):
-    return "ml.p2.xlarge"
+    return "ml.p3.2xlarge"
 
 
 @pytest.fixture(scope="session")
@@ -408,7 +408,7 @@ def pytest_generate_tests(metafunc):
             region in tests.integ.HOSTING_NO_P2_REGIONS
             or region in tests.integ.TRAINING_NO_P2_REGIONS
         ):
-            params.append("ml.p2.xlarge")
+            params.append("ml.p3.2xlarge")
         metafunc.parametrize("instance_type", params, scope="session")
 
     _generate_all_framework_version_fixtures(metafunc)

tests/integ/test_horovod.py

@@ -59,7 +59,7 @@ def test_hvd_gpu(
         sagemaker_session,
         tensorflow_training_latest_version,
         tensorflow_training_latest_py_version,
-        "ml.p2.xlarge",
+        "ml.p3.2xlarge",
         tmpdir,
     )