Add dependency on cudatoolkit (#55)

* cudatoolkit version added in meta.yaml build host and run section * Incremented build number and added cudatoolkit version in build host and run section of variants * Limit cudatoolkit change to run section only * Removed cudatoolkit in build section * Cudatoolkit parameter correction in torchtext meta and torchtext variants --------- Co-authored-by: root <[email protected]> Co-authored-by: root <[email protected]>
open-ce · Jul 10, 2024 · 7a07318 · 7a07318
1 parent ae62de2
commit 7a07318
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/text-0.16.2/recipe/meta.yaml b/text-0.16.2/recipe/meta.yaml
@@ -13,7 +13,7 @@ source:
     - 0101-disable-tcmalloc.patch
 
 build:
-  number: 2
+  number: 3
   string: {{ build_type }}_py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }}   #[build_type == 'cpu']
   string: {{ build_type }}{{ cudatoolkit | replace(".*", "") }}_py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }} #[build_type == 'cuda']
   script_env:                                    #[ppc_arch == 'p10']
@@ -29,6 +29,7 @@ requirements:
     - python {{ python }}
     - _pytorch_select  {{ pytorch_select_version }}
     - pytorch-base     {{ pytorch }}
+    - cudatoolkit {{ cudatoolkit }}            #[build_type == 'cuda']
   run:
     - numpy {{ numpy }}
     - python {{ python }}
@@ -38,7 +39,7 @@ requirements:
     - tqdm {{ tqdm }}
     - sentencepiece {{ sentencepiece }}
     - torchdata
-
+    - cudatoolkit {{ cudatoolkit }}            #[build_type == 'cuda']
 test:
   imports:
     - torchtext

diff --git a/text-0.16.2/variants/meta.yaml b/text-0.16.2/variants/meta.yaml
@@ -7,7 +7,7 @@ package:
   version: {{ version }} 
 
 build:
-  number: 1
+  number: 2
   string: py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }}   #[build_type == 'cpu']
   string: {{ build_type }}{{ cudatoolkit | replace(".*", "") }}_py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }} #[build_type == 'cuda']
 
@@ -17,6 +17,7 @@ requirements:
     - _pytorch_select {{ pytorch_select_version }}
     - torchtext-base {{ torchtext }}
     - python {{ python }}
+    - cudatoolkit {{ cudatoolkit }}            #[build_type == 'cuda']
 
 about:
   home: http://pytorch.org/