MCPcopy
hub / github.com/NVIDIA/TensorRT-LLM / generate_kernels.py

File generate_kernels.py

cpp/tensorrt_llm/kernels/cutlass_kernels/python/generate_kernels.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import argparse
2import enum
3import os
4from itertools import chain, product

Callers

nothing calls this directly

Calls 14

e2m1_typeClass · 0.85
has_archFunction · 0.85
generate_sm90_operationsFunction · 0.85
generate_sm80_operationsFunction · 0.85
should_skipFunction · 0.85
is_mixed_dtype_groupedFunction · 0.85
write_fileFunction · 0.85
clean_leftover_filesFunction · 0.85
splitMethod · 0.45

Tested by

no test coverage detected