Browse Source

build(dnn/cutlass): fix build for cutlass

GitOrigin-RevId: 9aa095fe84
release-1.5
Megvii Engine Team 3 years ago
parent
commit
f8b0f2cb91
1 changed files with 0 additions and 2 deletions
  1. +0
    -2
      dnn/scripts/cutlass_generator/BUILD

+ 0
- 2
dnn/scripts/cutlass_generator/BUILD View File

@@ -4,8 +4,6 @@ genrule(
name = "cutlass_kimpls",
outs = cutlass_gen_list,
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 gemv --type simt $(@D)
python3 $$GEN --operations deconv --type simt $(@D)


Loading…
Cancel
Save