Skip to content

MegEngine
MegEngine

MegEngine 天元 / MegEngine
1 年多 前同步成功

通知 403
Star 4705
Fork 582
MegEngine
MegEngine
提交 f8b0f2cb 编写于 作者: M Megvii Engine Team
浏览文件
操作

build(dnn/cutlass): fix build for cutlass 

GitOrigin-RevId: 9aa095fe84e7f01646280b49bc268e52e4fb6909
上级 0fb4e9a9
显示空白变更内容
内联 并排
Showing with 0 addition and 2 deletion
dnn/scripts/cutlass_generator/BUILD
浏览文件 @ f8b0f2cb
...@@ -4,8 +4,6 @@ genrule( ...@@ -4,8 +4,6 @@ genrule(
name = "cutlass_kimpls", name = "cutlass_kimpls",
outs = cutlass_gen_list, outs = cutlass_gen_list,
cmd = """GEN=$(location //brain/megbrain/dnn/scripts/cutlass_generator:generator.py) cmd = """GEN=$(location //brain/megbrain/dnn/scripts/cutlass_generator:generator.py)
pwd > /tmp/a
echo $(@D) > /tmp/b
python3 $$GEN --operations gemm --type simt $(@D) python3 $$GEN --operations gemm --type simt $(@D)
python3 $$GEN --operations gemv --type simt $(@D) python3 $$GEN --operations gemv --type simt $(@D)
python3 $$GEN --operations deconv --type simt $(@D) python3 $$GEN --operations deconv --type simt $(@D)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或