4e9b1c4e创建于 2022年10月8日历史提交

文件	最后提交记录	最后更新时间
BUILD	feat(dnn): add rrconv wgrad, support int32 and uint8 region mask GitOrigin-RevId: 0da9b3bca86ba6ae11289e451b224aca348f647b	3 年前
README.md	feat(dnn/cuda): generate cutlass kimpls using cmake and bazel GitOrigin-RevId: da3bcfb85af3f148116b8b6a51f214c6cef8684e	5 年前
conv2d_operation.py	feat(dnn): add rrconv wgrad, support int32 and uint8 region mask GitOrigin-RevId: 0da9b3bca86ba6ae11289e451b224aca348f647b	3 年前
gemm_operation.py	style(mgb/tools): add format for tools, dnn and ci GitOrigin-RevId: 5684e5ea43385ae7f3802c5c2e779362a030d643	4 年前
gen_list.py	feat(dnn): add rrconv wgrad, support int32 and uint8 region mask GitOrigin-RevId: 0da9b3bca86ba6ae11289e451b224aca348f647b	3 年前
generator.py	feat(dnn): add rrconv wgrad, support int32 and uint8 region mask GitOrigin-RevId: 0da9b3bca86ba6ae11289e451b224aca348f647b	3 年前
library.py	feat(dnn): add rrconv wgrad, support int32 and uint8 region mask GitOrigin-RevId: 0da9b3bca86ba6ae11289e451b224aca348f647b	3 年前
list.bzl	feat(dnn): add rrconv wgrad, support int32 and uint8 region mask GitOrigin-RevId: 0da9b3bca86ba6ae11289e451b224aca348f647b	3 年前
manifest.py	style(mgb/tools): add format for tools, dnn and ci GitOrigin-RevId: 5684e5ea43385ae7f3802c5c2e779362a030d643	4 年前

Generate device kernel registration code for CUTLASS kernels

Usage

python3 generator.py [--operations {gemm, gemv, conv2d, deconv}] [--type {simt, tensorop8816, tensorop8832}]
                     output

We generate list.bzl because the genrule method of bazel requires that the output file list be specified in the analysis phase.

Please call gen_list.py when new operations are added.

python3 gen_list.py