debugging setup
Browse files
setup.py
CHANGED
@@ -1,6 +1,9 @@
|
|
1 |
from setuptools import setup, find_packages
|
2 |
from setuptools.command.install import install
|
3 |
-
import os
|
|
|
|
|
|
|
4 |
|
5 |
|
6 |
with open("README.md", "r") as f:
|
@@ -12,21 +15,23 @@ with open('requirements.txt', 'r', encoding='utf-16') as ff:
|
|
12 |
|
13 |
class InstallCommand(install):
|
14 |
user_option = install.user_options + [
|
15 |
-
('
|
16 |
]
|
17 |
|
18 |
def initialize_options(self):
|
19 |
-
|
20 |
self.cupy = "cupy"
|
21 |
|
22 |
def finalize_options(self):
|
23 |
-
print("CuPy version selected is: ", self.
|
24 |
-
|
25 |
|
26 |
def run(self):
|
|
|
27 |
global cupy
|
28 |
required.append(self.cupy)
|
29 |
-
|
|
|
30 |
|
31 |
|
32 |
setup(
|
|
|
1 |
from setuptools import setup, find_packages
|
2 |
from setuptools.command.install import install
|
3 |
+
import os, sys
|
4 |
+
|
5 |
+
|
6 |
+
print("\n\n\n\n---:", sys.args)
|
7 |
|
8 |
|
9 |
with open("README.md", "r") as f:
|
|
|
15 |
|
16 |
class InstallCommand(install):
|
17 |
user_option = install.user_options + [
|
18 |
+
('cupyyy=', None, 'enable flag to install package with GPU support'),
|
19 |
]
|
20 |
|
21 |
def initialize_options(self):
|
22 |
+
super().initialize_options()
|
23 |
self.cupy = "cupy"
|
24 |
|
25 |
def finalize_options(self):
|
26 |
+
print("CuPy version selected is: ", self.cupyyy)
|
27 |
+
super().finalize_options()
|
28 |
|
29 |
def run(self):
|
30 |
+
# use options
|
31 |
global cupy
|
32 |
required.append(self.cupy)
|
33 |
+
|
34 |
+
super().run()
|
35 |
|
36 |
|
37 |
setup(
|