Commit abdd8b40 authored by lijian6's avatar lijian6
Browse files

Merge branch 'test_0513' into 'main'

Add python version for whl package.

See merge request dcutoolkit/deeplearing/DeepEP!30
parents 8f9cb191 e3b8804b
...@@ -5,6 +5,7 @@ import subprocess ...@@ -5,6 +5,7 @@ import subprocess
import sysconfig import sysconfig
from typing import Optional from typing import Optional
import subprocess import subprocess
from setuptools import Distribution
import torch import torch
...@@ -105,6 +106,10 @@ def get_deepep_version() -> str: ...@@ -105,6 +106,10 @@ def get_deepep_version() -> str:
version = get_version() version = get_version()
return version return version
class BinaryDistribution(Distribution):
def has_ext_modules(self):
return True
if __name__ == '__main__': if __name__ == '__main__':
setuptools.setup( setuptools.setup(
name='deep_ep', name='deep_ep',
...@@ -113,4 +118,5 @@ if __name__ == '__main__': ...@@ -113,4 +118,5 @@ if __name__ == '__main__':
include_package_data=True, include_package_data=True,
package_data={"deep_ep": [f"deep_ep_cpp{sysconfig.get_config_var('EXT_SUFFIX')}"]}, package_data={"deep_ep": [f"deep_ep_cpp{sysconfig.get_config_var('EXT_SUFFIX')}"]},
zip_safe=False, zip_safe=False,
distclass=BinaryDistribution,
) )
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