From 406402a3f88f0bcbad69606881a154660ab38547 Mon Sep 17 00:00:00 2001
From: Archana-Shinde1 <Archana.Shinde1@ibm.com>
Date: Mon, 24 Jun 2024 11:00:45 +0000
Subject: [PATCH 1/3] pin to use appropriate cuda version

---
 recipe/meta.yaml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/recipe/meta.yaml b/recipe/meta.yaml
index adde92a..cee823e 100644
--- a/recipe/meta.yaml
+++ b/recipe/meta.yaml
@@ -20,7 +20,7 @@ source:
     - 0303-Update-rules_python-to-fix-runtime-python-repo.patch         #[ppc64le]
 
 build:
-  number: 6
+  number: 7
   string: {{ build_type }}_py{{ python | replace(".", "") }}_pb{{ protobuf | replace(".*", "")}}_{{ PKG_BUILDNUM }}   #[build_type == 'cpu']
   string: {{ build_type }}{{ cudatoolkit | replace(".*", "") }}_py{{ python | replace(".", "") }}_pb{{ protobuf | replace(".*", "")}}_{{ PKG_BUILDNUM }} #[build_type == 'cuda']
 
@@ -34,6 +34,7 @@ outputs:
       host:
         - git >=2.2
         - bazel {{ bazel }}
+        - cudatoolkit {{ cudatoolkit }}
         - setuptools {{ setuptools }}
         - python {{ python }}
         - numpy {{ numpy }}
@@ -63,6 +64,7 @@ outputs:
       host:
         - git >=2.2
         - setuptools {{ setuptools }}
+        - cudatoolkit {{ cudatoolkit }}
         - python {{ python }}
         - numpy {{ numpy }}
         - tensorflow-base {{ tensorflow }}

From 2ee659d6c8460af6742745f30d7cf3af2279a12b Mon Sep 17 00:00:00 2001
From: Archana-Shinde1 <161808226+Archana-Shinde1@users.noreply.github.com>
Date: Mon, 24 Jun 2024 18:05:29 +0530
Subject: [PATCH 2/3] Update meta.yaml

---
 recipe/meta.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/recipe/meta.yaml b/recipe/meta.yaml
index cee823e..71501c9 100644
--- a/recipe/meta.yaml
+++ b/recipe/meta.yaml
@@ -48,6 +48,7 @@ outputs:
         - python {{ python }}
         - numpy {{ numpy }}
         - tensorflow-base {{ tensorflow }}
+        - cudatoolkit {{ cudatoolkit }}
         - _tensorflow_select 1.0          #[build_type == 'cpu']
         - _tensorflow_select 2.0          #[build_type == 'cuda']
         - keras {{ keras }}

From ef0fa52e7bd3d0febefd112e810ba5b36eb4f3c1 Mon Sep 17 00:00:00 2001
From: Archana-Shinde1 <Archana.Shinde1@ibm.com>
Date: Mon, 24 Jun 2024 12:49:02 +0000
Subject: [PATCH 3/3] Add filter

---
 recipe/meta.yaml | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/recipe/meta.yaml b/recipe/meta.yaml
index 71501c9..b0e4a74 100644
--- a/recipe/meta.yaml
+++ b/recipe/meta.yaml
@@ -34,7 +34,7 @@ outputs:
       host:
         - git >=2.2
         - bazel {{ bazel }}
-        - cudatoolkit {{ cudatoolkit }}
+        - cudatoolkit {{ cudatoolkit }}   #[build_type == 'cuda']
         - setuptools {{ setuptools }}
         - python {{ python }}
         - numpy {{ numpy }}
@@ -48,7 +48,7 @@ outputs:
         - python {{ python }}
         - numpy {{ numpy }}
         - tensorflow-base {{ tensorflow }}
-        - cudatoolkit {{ cudatoolkit }}
+        - cudatoolkit {{ cudatoolkit }}   #[build_type == 'cuda']
         - _tensorflow_select 1.0          #[build_type == 'cpu']
         - _tensorflow_select 2.0          #[build_type == 'cuda']
         - keras {{ keras }}
@@ -65,7 +65,7 @@ outputs:
       host:
         - git >=2.2
         - setuptools {{ setuptools }}
-        - cudatoolkit {{ cudatoolkit }}
+        - cudatoolkit {{ cudatoolkit }}   #[build_type == 'cuda']
         - python {{ python }}
         - numpy {{ numpy }}
         - tensorflow-base {{ tensorflow }}
@@ -75,6 +75,7 @@ outputs:
         - tensorflow-estimator {{ tensorflow }}
         - tensorboard {{ tensorboard }}
       run:
+        - cudatoolkit {{ cudatoolkit }}   #[build_type == 'cuda']
         - python {{ python }}
         - numpy {{ numpy }}
         - tensorflow-base {{ tensorflow }}