Movatterモバイル変換


[0]ホーム

URL:


Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

Commit30f2cf9

Browse files
test
1 parent3e103b0 commit30f2cf9

File tree

3 files changed

+8
-5
lines changed

3 files changed

+8
-5
lines changed

‎pyproject.toml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ before-all = [
3232
[tool.cibuildwheel.linux.environment]
3333
cuda ="12.4"
3434
temp ="/tmp/cuda-temp"
35+
TENSOR_ARRAY_INSTALL_PATH ="/tmp/tensor-array-temp"
3536
CUDA_PATH ="/usr/local/cuda-${cuda}"
3637
PATH ="${PATH}:${CUDA_PATH}/bin"
3738
LD_LIBRARY_PATH ="${LD_LIBRARY_PATH}:${CUDA_PATH}/lib:${CUDA_PATH}/lib64"

‎scripts/build-env/manylinux.sh‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ fi
6262
cd build
6363
# Configure the build with CMake
6464
echo"Configuring the build with CMake..."
65-
cmake .. -DCMAKE_CUDA_HOST_COMPILER="/usr/bin/gcc"
65+
cmake .. -DCMAKE_CUDA_HOST_COMPILER="/usr/bin/gcc" -DCMAKE_INSTALL_PREFIX=${TENSOR_ARRAY_INSTALL_PATH}
6666
cmake --build.
6767
cmake --install.
6868

‎setup.py‎

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,22 @@
77
fromsetuptoolsimportsetup,find_packages
88
frompybind11.setup_helpersimportPybind11Extension,build_ext
99

10-
__version__="0.0.7"
10+
__version__="0.0.8"
1111

1212
defmain():
1313
cwd=os.path.dirname(os.path.abspath(__file__))
1414
withopen(os.path.join(cwd,"README.md"),encoding="utf-8")asf:
1515
long_description=f.read()
1616

17+
tensor_array_lib_path=os.environ['TENSOR_ARRAY_INSTALL_PATH']
18+
1719
ext_modules= [
1820
Pybind11Extension(
1921
"tensor_array._ext",
2022
sources=glob.glob(os.path.join("cpp","*.cc")),
21-
include_dirs=["/usr/local/include"],
22-
library_dirs=["/usr/local/lib","/usr/local/lib64"],
23-
#libraries=["tensorarray_core", "tensorarray_layers"],
23+
include_dirs=[tensor_array_lib_path+"/include"],
24+
library_dirs=[tensor_array_lib_path+"/lib/tensor-array",tensor_array_lib_path+"/lib64/tensor-array"],
25+
libraries=["tensorarray_core","tensorarray_layers"],
2426
define_macros=[("VERSION_INFO",__version__)],
2527
),
2628
]

0 commit comments

Comments
 (0)

[8]ページ先頭

©2009-2025 Movatter.jp