From 94ef6c36dfbadd97c0e8a6a2d9a2abd6f5ac78af Mon Sep 17 00:00:00 2001 From: Prince Datta Date: Mon, 11 Nov 2024 16:09:39 +0530 Subject: [PATCH 1/4] Running mlvm test --- mlvm/test_mlvm.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mlvm/test_mlvm.py b/mlvm/test_mlvm.py index 121bec6d0..0df3e5752 100644 --- a/mlvm/test_mlvm.py +++ b/mlvm/test_mlvm.py @@ -22,6 +22,7 @@ class MLVMTestCase(DataprocTestCase): RAPIDS_DASK_SCRIPT = "verify_rapids_dask.py" DASK_YARN_SCRIPT = "verify_dask_yarn.py" DASK_STANDALONE_SCRIPT = "verify_dask_standalone.py" + #dummy def verify_python(self): self.assert_dataproc_job( From 735e72b022dad44ed9a025e29aabca73df09e42d Mon Sep 17 00:00:00 2001 From: Prince Datta Date: Tue, 10 Dec 2024 10:49:46 +0530 Subject: [PATCH 2/4] dummy test --- horovod/test_horovod.py | 1 + 1 file changed, 1 insertion(+) diff --git a/horovod/test_horovod.py b/horovod/test_horovod.py index 8037a38a8..cd1d434ce 100644 --- a/horovod/test_horovod.py +++ b/horovod/test_horovod.py @@ -16,6 +16,7 @@ class HorovodTestCase(DataprocTestCase): TENSORFLOW_TEST_SCRIPT = "scripts/verify_tensorflow.py" PYTORCH_TEST_SCRIPT = "scripts/verify_pytorch.py" MXNET_TEST_SCRIPT = "scripts/verify_mxnet.py" + # dummy def _submit_spark_job(self, script): self.assert_dataproc_job( From 182b3a110a6e274e77e9c47026cc5b9046cbf67e Mon Sep 17 00:00:00 2001 From: Prince Datta Date: Tue, 10 Dec 2024 10:51:06 +0530 Subject: [PATCH 3/4] dummy test --- horovod/test_horovod.py | 1 - 1 file changed, 1 deletion(-) diff --git a/horovod/test_horovod.py b/horovod/test_horovod.py index cd1d434ce..8037a38a8 100644 --- a/horovod/test_horovod.py +++ b/horovod/test_horovod.py @@ -16,7 +16,6 @@ class HorovodTestCase(DataprocTestCase): TENSORFLOW_TEST_SCRIPT = "scripts/verify_tensorflow.py" PYTORCH_TEST_SCRIPT = "scripts/verify_pytorch.py" MXNET_TEST_SCRIPT = "scripts/verify_mxnet.py" - # dummy def _submit_spark_job(self, script): self.assert_dataproc_job( From d7ca8df470bc8782ea18032dadb6bb73cf7fae56 Mon Sep 17 00:00:00 2001 From: Prince Datta Date: Tue, 10 Dec 2024 10:55:04 +0530 Subject: [PATCH 4/4] dummy test --- mlvm/test_mlvm.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/mlvm/test_mlvm.py b/mlvm/test_mlvm.py index 0df3e5752..004545fc0 100644 --- a/mlvm/test_mlvm.py +++ b/mlvm/test_mlvm.py @@ -124,11 +124,8 @@ def test_mlvm_gpu(self, configuration, dask_runtime, rapids_runtime): if self.getImageVersion() < pkg_resources.parse_version("2.0"): self.skipTest("Not supported in pre 2.0 images") - metadata = ("init-actions-repo={},include-gpus=true" - ",gpu-driver-provider=NVIDIA").format(self.INIT_ACTIONS_REPO) - - cudnn_version = "8.1.1.33" - cuda_version = "11.2" + cudnn_version = "9.1.0.70" + cuda_version = "12.4" metadata = ("init-actions-repo={},include-gpus=true" ",gpu-driver-provider=NVIDIA," @@ -144,7 +141,7 @@ def test_mlvm_gpu(self, configuration, dask_runtime, rapids_runtime): configuration, self.INIT_ACTIONS, optional_components=self.OPTIONAL_COMPONENTS, - machine_type="n1-standard-4", + machine_type="n1-highmem-8", master_accelerator="type=nvidia-tesla-t4", worker_accelerator="type=nvidia-tesla-t4", timeout_in_minutes=60,