OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branch: release-1.8
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'release-1.8'
${ noResults }
MegEngine/dnn/scripts/cutlass_generator/README.md

646 B
Raw Permalink Blame History

Generate device kernel registration code for CUTLASS kernels

Usage

python3 generator.py [--operations {gemm, gemv, conv2d, deconv}] [--type {simt, tensorop8816, tensorop8832}]
                     output
  • operations: operation kind, including gemm|gemv|conv2d|deconv
  • type: opcode class, simt|tensorop8816|tensorop8832
  • output: the output directory for CUTLASS kernels

Generate file list for bazel

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