|
@@ -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) |
|
|