Skip to content

Commit

Permalink
fix manifest and sdist version
Browse files Browse the repository at this point in the history
  • Loading branch information
ver217 committed Jul 19, 2022
1 parent c99ce33 commit e51dc5f
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 5 deletions.
7 changes: 4 additions & 3 deletions MANIFEST.in
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
include *.txt README.md
recursive-include csrc *.cpp *.h *.cu *.tr *.cuh *.cc *.txt
recursive-include include *.cpp *.h *.cu *.tr *.cuh *.cc *.txt
recursive-include 3rd *.cpp *.h *.cu *.tr *.cuh *.cc *.txt
recursive-include csrc *.cpp *.txt
recursive-include include *.h
recursive-include 3rd *.txt
recursive-include tensornvme *.pyi
5 changes: 3 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,8 +85,9 @@ def setup_dependencies():
def get_version():
with open('version.txt') as f:
version = f.read().strip()
torch_version = '.'.join(torch.__version__.split('.')[:2])
version += f'+torch{torch_version}'
if sys.argv[1] != 'sdist':
torch_version = '.'.join(torch.__version__.split('.')[:2])
version += f'+torch{torch_version}'
return version


Expand Down

0 comments on commit e51dc5f

Please sign in to comment.