|
Up
|
|
|
|
|
__pycache__/
|
— |
|
|
|
serialized_patterns/
|
— |
|
|
|
__init__.py
|
|
|
|
|
b2b_gemm.py
|
|
|
|
|
binary_folding.py
|
|
|
|
|
bucketing.py
|
|
|
|
|
control_dependencies.py
|
|
|
|
|
ddp_fusion.py
|
|
|
|
|
decompose_mem_bound_mm.py
|
|
|
|
|
dedupe_symint_uses.py
|
|
|
|
|
efficient_conv_bn_eval.py
|
|
|
|
|
freezing_patterns.py
|
|
|
|
|
fsdp.py
|
|
|
|
|
fuse_attention.py
|
|
|
|
|
graph_view.py
|
|
|
|
|
group_batch_fusion.py
|
|
|
|
|
joint_graph.py
|
|
|
|
|
memory_estimator.py
|
|
|
|
|
micro_pipeline_tp.py
|
|
|
|
|
misc_patterns.py
|
|
|
|
|
mkldnn_fusion.py
|
|
|
|
|
node_runtime_estimation.py
|
|
|
|
|
numeric_utils.py
|
|
|
|
|
overlap_manual_scheduling.py
|
|
|
|
|
overlap_preserving_bucketer.py
|
|
|
|
|
overlap_scheduling.py
|
|
|
|
|
pad_mm.py
|
|
|
|
|
post_grad.py
|
|
|
|
|
pre_grad.py
|
|
|
|
|
quantization.py
|
|
|
|
|
reinplace.py
|
|
|
|
|
replace_random.py
|
|
|
|
|
split_cat.py
|
|
|
|