Commit e5d5fc7e authored by gaoqiong's avatar gaoqiong
Browse files

Update setup.py

parent 224ab8d9
...@@ -8,6 +8,7 @@ from setuptools import setup, find_packages ...@@ -8,6 +8,7 @@ from setuptools import setup, find_packages
from torch.utils.cpp_extension import CUDAExtension from torch.utils.cpp_extension import CUDAExtension
from typing import Optional, Union from typing import Optional, Union
pwd = os.path.dirname(__file__)
add_git_version = False add_git_version = False
if int(os.environ.get('ADD_GIT_VERSION', '0')) == 1: if int(os.environ.get('ADD_GIT_VERSION', '0')) == 1:
add_git_version = True add_git_version = True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment