#!/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