fix: use setup.py install for CUTLASS extension build

This commit is contained in:
2026-05-16 02:21:17 +00:00
parent 882bff8fb7
commit de9b50cbe7

View File

@@ -35,7 +35,7 @@ pip install -e . --no-build-isolation -q
# Build the CUTLASS NVFP4 GEMM C++ extension
echo " Building CUTLASS C++ extension..."
cd "$REPO_DIR/src/nvfp4_megamoe_kernel/cutlass_nvfp4_gemm"
pip install -e . --no-build-isolation -q
python setup.py install
# 4. Run the test
echo "[4/4] Running layer test..."