build.sh 273 Bytes
Newer Older
wangkx1's avatar
init  
wangkx1 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#!/bin/bash

# 构建目录
BUILD_DIR="./build"

# 创建构建目录
mkdir -p $BUILD_DIR
cd $BUILD_DIR

# 运行 CMake
cmake .. 

# 编译
cmake --build .

# 复制编译后的库到当前目录
cp test_ops.so ..

cd ..

# 运行测试
python test_torch_library_expand.py