|
- load("list.bzl", "cutlass_gen_list")
-
- genrule(
- name = "cutlass_kimpls",
- outs = cutlass_gen_list,
- cmd = """GEN=$(location //brain/megbrain/dnn/scripts/cutlass_generator:generator.py)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations gemm --type simt $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations gemm --type tensorop884 $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations gemm --type tensorop1688 $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations gemv --type simt $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations deconv --type simt $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations deconv --type tensorop8816 $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations conv2d --type simt $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations conv2d --type tensorop8816 $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations conv2d --type tensorop8832 $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations dwconv2d_fprop --type simt $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations dwconv2d_fprop --type tensorop884 $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations dwconv2d_dgrad --type simt $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations dwconv2d_dgrad --type tensorop884 $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations dwconv2d_wgrad --type simt $(@D)
- CUTLASS_WITH_LONG_PATH=true python3 $$GEN --operations dwconv2d_wgrad --type tensorop884 $(@D)
- """,
- tools = ["//brain/megbrain/dnn/scripts/cutlass_generator:generator.py"],
- visibility = ["//visibility:public"],
- )
|
