Skip to content

Commit 9f47f37

Browse files
committed
update paddlenlp_ops name
1 parent 4821b63 commit 9f47f37

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

csrc/tools/build_wheel.sh

+10-1
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,7 @@ function create_directories(){
9393
9494
import os
9595
from datetime import datetime
96+
import paddle
9697
9798
from setuptools import find_packages, setup
9899
@@ -109,12 +110,20 @@ def read(file: str):
109110
content = f.read().strip()
110111
return content
111112
113+
def get_sm_version():
114+
prop = paddle.device.cuda.get_device_properties()
115+
cc = prop.major * 10 + prop.minor
116+
return cc
112117
113118
def read_version():
114119
"""
115120
read version and return content
116121
"""
117-
__version__ = "3.0.0b4.post"
122+
cuda_version = float(paddle.version.cuda())
123+
sm_version = get_sm_version()
124+
paddle_commit = paddle.__git_commit__[:7]
125+
paddlenlp_version= "3.0.0b4.post"
126+
__version__ = "cuda{}-sm{}-paddle_{}-paddlenlp_{}".format(cuda_version, sm_version, paddle_commit, paddlenlp_version)
118127
formatted_date = datetime.now().date().strftime("%Y%m%d")
119128
__version__ = __version__.replace(".post", ".post{}".format(formatted_date))
120129

0 commit comments

Comments
 (0)