From 8d942a85f6a15486f09644340b0588318ed4a87c Mon Sep 17 00:00:00 2001
From: ANANDHU S <71482562+anandhu-eng@users.noreply.github.com>
Date: Mon, 30 Sep 2024 20:11:49 +0530
Subject: [PATCH] skip model and dataset download if env variable is set

---
 script/app-mlperf-inference-nvidia/_cm.yaml | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/script/app-mlperf-inference-nvidia/_cm.yaml b/script/app-mlperf-inference-nvidia/_cm.yaml
index e39fec5a91..1e5ec0d323 100644
--- a/script/app-mlperf-inference-nvidia/_cm.yaml
+++ b/script/app-mlperf-inference-nvidia/_cm.yaml
@@ -165,6 +165,8 @@ deps:
       - on
       CM_RUN_STATE_DOCKER:
       - 'yes'
+      CM_MLPERF_MODEL_DLRM_SKIP_DOWNLOAD:
+      - 'yes'
     names:
       - criteo-preprocessed
     tags: get,dataset,preprocessed,criteo
@@ -180,6 +182,8 @@ deps:
       - on
       CM_RUN_STATE_DOCKER:
       - 'yes'
+      CM_MLPERF_MODEL_DLRM_SKIP_DOWNLOAD:
+      - 'yes'
     names:
       - dlrm-model
     tags: get,ml-model,dlrm,_pytorch