diff options
Diffstat (limited to 'install-orttraining-files.diff')
-rw-r--r-- | install-orttraining-files.diff | 24 |
1 files changed, 13 insertions, 11 deletions
diff --git a/install-orttraining-files.diff b/install-orttraining-files.diff index 7df2871aa219..8da1337458df 100644 --- a/install-orttraining-files.diff +++ b/install-orttraining-files.diff @@ -1,18 +1,20 @@ ---- a/setup.py 2021-12-29 22:44:09.924917943 +0800 -+++ b/setup.py 2021-12-29 22:49:16.216878004 +0800 -@@ -355,7 +355,7 @@ +diff --git a/setup.py b/setup.py +index 294b975a56..1612a4419c 100644 +--- a/setup.py ++++ b/setup.py +@@ -523,7 +523,7 @@ classifiers = [ if not enable_training: classifiers.extend(["Operating System :: Microsoft :: Windows", "Operating System :: MacOS"]) --if enable_training: +-if enable_training or enable_training_apis: +if True: - packages.extend( - [ - "onnxruntime.training", -@@ -373,6 +373,7 @@ - '*.cuh', - '*.h' - ] + packages.append("onnxruntime.training") + if enable_training: + packages.extend( +@@ -565,6 +565,7 @@ if enable_training or enable_training_apis: + ] + ) + +if enable_training: requirements_file = "requirements-training.txt" # with training, we want to follow this naming convention: |