Skip to content

Commit

Permalink
Fix csrc readme (PaddlePaddle#9515)
Browse files Browse the repository at this point in the history
  • Loading branch information
yuanlehome authored Nov 28, 2024
1 parent 5926c30 commit e23214b
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion csrc/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,11 @@ pip install -r requirements.txt

## 编译 Cuda 算子

生成 FP8的 cutlass 算子
```shell
python setup_cuda.py install
```

## 生成 FP8的 cutlass 算子
```shell
python utils/auto_gen_fp8_fp8_gemm_fused_kernels.py

Expand Down

0 comments on commit e23214b

Please sign in to comment.