Closed zuocebianpingmao closed 11 months ago
Environment
Reproduction Steps 运行以下命令时报错 python triton/python/triton/tools/compile.py \ fmha_triton.py \ -n fused_attention_kernel \ -o aot/fp16/fmha_kernel_d64_fp16 \ --out-name fmha_d64_fp16 \ -w 4 \ -s "*fp16:16, *fp32:16, *fp32:16, *fp16:16, *fp16:16, *fp16:16, fp32, i32, 128, 64, 128"
python triton/python/triton/tools/compile.py \ fmha_triton.py \ -n fused_attention_kernel \ -o aot/fp16/fmha_kernel_d64_fp16 \ --out-name fmha_d64_fp16 \ -w 4 \ -s "*fp16:16, *fp32:16, *fp32:16, *fp16:16, *fp16:16, *fp16:16, fp32, i32, 128, 64, 128"
Expected Behavior 在当前目录生成aot/fp16/fmha_kernel_d64_fp16文件夹
Actual Behavior 出现错误 usage: compile.py [-h] --kernel-name KERNEL_NAME [--num-warps NUM_WARPS] [--num-stages NUM_STAGES] [--out-name OUT_NAME] [--out-path OUT_PATH] --signature SIGNATURE --grid GRID path compile.py: error: the following arguments are required: --grid/-g
usage: compile.py [-h] --kernel-name KERNEL_NAME [--num-warps NUM_WARPS] [--num-stages NUM_STAGES] [--out-name OUT_NAME] [--out-path OUT_PATH] --signature SIGNATURE --grid GRID path compile.py: error: the following arguments are required: --grid/-g
Additional Notes 报错原因是缺少必要参数--grid/-g,triton库更新后compile增加了--grid/-g参数
Environment
Reproduction Steps 运行以下命令时报错
python triton/python/triton/tools/compile.py \ fmha_triton.py \ -n fused_attention_kernel \ -o aot/fp16/fmha_kernel_d64_fp16 \ --out-name fmha_d64_fp16 \ -w 4 \ -s "*fp16:16, *fp32:16, *fp32:16, *fp16:16, *fp16:16, *fp16:16, fp32, i32, 128, 64, 128"
Expected Behavior 在当前目录生成aot/fp16/fmha_kernel_d64_fp16文件夹
Actual Behavior 出现错误
usage: compile.py [-h] --kernel-name KERNEL_NAME [--num-warps NUM_WARPS] [--num-stages NUM_STAGES] [--out-name OUT_NAME] [--out-path OUT_PATH] --signature SIGNATURE --grid GRID path compile.py: error: the following arguments are required: --grid/-g
Additional Notes 报错原因是缺少必要参数--grid/-g,triton库更新后compile增加了--grid/-g参数