Add condition to check instance type

Author: Jyothirmaikottu

dlc_developer_config.toml

@@ -37,7 +37,7 @@ deep_canary_mode = false
 [build]
 # Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image.
 # available frameworks - ["base", "vllm", "autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
-build_frameworks = []
+build_frameworks = ["vllm"]
 
 
 # By default we build both training and inference containers. Set true/false values to determine which to build.

test/vllm/trigger_test.py

@@ -16,8 +16,9 @@ def run_platform_tests(platform: str, images: List[str], commit_id: str, ipv6_en
     """
     Run tests for a specific platform
     """
-    LOGGER.info(f"Running {platform} tests")
-    if platform == "ec2":
+    instance_type = os.getenv("EC2_GPU_INSTANCE_TYPE")
+    print(f"Running {platform} tests on {instance_type}")
+    if platform == "ec2" and instance_type == "p4d.24xlarge":
         try:
             ec2_resources = setup()
             print(ec2_resources)