├── .commit-msg-config.yaml ├── .coveragerc ├── .github ├── CODEOWNERS ├── ISSUE_TEMPLATE │ ├── bug-report.yml │ ├── chores.yml │ ├── config.yml │ └── feature-request.yml ├── PULL_REQUEST_TEMPLATE.md ├── codecov.yml ├── release.yml └── workflows │ ├── auto_release.yml │ ├── ci-for-mlu.yml │ ├── ci-for-npu.yml │ ├── lint_check.yml │ └── scripts │ ├── rest_api.py │ └── upload_to_byted_luban.sh ├── .gitignore ├── .lintrunner.toml ├── .pre-commit-config.yaml ├── LICENSE.txt ├── MANIFEST.in ├── README.md ├── docs ├── RELEASE.md ├── RELEASE.md.template ├── RELEASE_CURRENT.md ├── compilation_pipeline.png ├── update_release_note.sh └── xpu_graph_arch.png ├── pyproject.toml ├── requirements.txt ├── scripts ├── build.py ├── commit_msg_check.sh ├── install_githooks.sh └── lint_check.sh ├── src └── xpu_graph │ ├── __init__.py │ ├── accuracy_utils.py │ ├── backends │ ├── __init__.py │ ├── mlu.py │ └── npu.py │ ├── cache.py │ ├── compiler.py │ ├── config.py │ ├── constant_manager.py │ ├── fx_utils.py │ ├── guard_filters.py │ ├── interceptor.py │ ├── passes │ ├── __init__.py │ ├── constant_folding.py │ ├── cse.py │ ├── dce.py │ ├── fake_tensor_prop.py │ ├── optimizer.py │ ├── pass_manager.py │ ├── patterns │ │ ├── __init__.py │ │ ├── common │ │ │ ├── __init__.py │ │ │ ├── change_tensor_like.py │ │ │ ├── combine_pointwise_sink.py │ │ │ ├── combine_pointwise_source.py │ │ │ ├── fold_add.py │ │ │ ├── fold_cat.py │ │ │ ├── fold_clone.py │ │ │ ├── fold_detach.py │ │ │ ├── fold_div.py │ │ │ ├── fold_expand.py │ │ │ ├── fold_mul.py │ │ │ ├── fold_reduce.py │ │ │ ├── fold_slice_like.py │ │ │ ├── fold_squeeze.py │ │ │ ├── fold_stack.py │ │ │ ├── fold_sub.py │ │ │ ├── fold_to_copy.py │ │ │ ├── fold_view.py │ │ │ ├── fold_where.py │ │ │ ├── fuse_addmm.py │ │ │ ├── fuse_addn.py │ │ │ ├── fuse_baddbmm.py │ │ │ ├── fuse_layernorm.py │ │ │ ├── fuse_rmsnorm.py │ │ │ ├── fuse_scaled_dot_product_attention.py │ │ │ ├── gelu.md │ │ │ ├── gelu.py │ │ │ └── sink_view.py │ │ ├── pattern.py │ │ ├── pattern_manager.py │ │ ├── plugin_pattern.py │ │ ├── structure │ │ │ ├── __init__.py │ │ │ ├── custom_denselayer.py │ │ │ ├── custom_layernorm.py │ │ │ ├── custom_rmsnorm.py │ │ │ ├── fuse_combosum.py │ │ │ ├── fuse_multiple_slice_cat.py │ │ │ ├── fuse_slice.py │ │ │ ├── fuse_slice_cat.py │ │ │ └── fuse_slice_stack.py │ │ ├── targets │ │ │ ├── __init__.py │ │ │ ├── mlu │ │ │ │ ├── __init__.py │ │ │ │ ├── combo_matmul.py │ │ │ │ ├── combo_matmul_utils.py │ │ │ │ ├── fuse_add_norm.py │ │ │ │ ├── fuse_densetower.py │ │ │ │ ├── fuse_linear_attention.py │ │ │ │ ├── fuse_mul_sum_cat.py │ │ │ │ ├── fuse_scale_layernorm.py │ │ │ │ ├── fuse_slice_where_cat.py │ │ │ │ ├── fuse_sum_cat.py │ │ │ │ ├── linear_to_matmul.py │ │ │ │ ├── repeat_to_expand.py │ │ │ │ ├── structure_replacements │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── dense_layer_modules.py │ │ │ │ │ └── norm_modules.py │ │ │ │ └── triton_kernel │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── fused_linear_attn.py │ │ │ │ │ ├── fused_mul_sum_cat.py │ │ │ │ │ ├── fused_serial_mm_2dot.py │ │ │ │ │ ├── fused_serial_mm_3dot.py │ │ │ │ │ ├── fused_slice.py │ │ │ │ │ ├── fused_slice_cat.py │ │ │ │ │ ├── fused_slice_sum_cat.py │ │ │ │ │ ├── fused_slice_v2.py │ │ │ │ │ ├── fused_slice_where_cat.py │ │ │ │ │ ├── fused_sum_3d.py │ │ │ │ │ ├── fused_sum_cat.py │ │ │ │ │ ├── get_mlu_devinfo.py │ │ │ │ │ ├── libentry.py │ │ │ │ │ └── linear_attention_kernel.py │ │ │ └── npu │ │ │ │ ├── __init__.py │ │ │ │ ├── check_npu_ops.py │ │ │ │ ├── convert_gemm_like_weights_ND2NZ.py │ │ │ │ ├── fuse_add_rmsnorm.py │ │ │ │ ├── fuse_brc_permute_sum.py │ │ │ │ ├── fuse_div_mul_sum.py │ │ │ │ ├── fuse_silu_mul.py │ │ │ │ ├── remove_layernorm_cast_inference.py │ │ │ │ ├── remove_rmsnorm_cast_inference.py │ │ │ │ ├── scoped_super_kernel.py │ │ │ │ ├── structure_replacements │ │ │ │ ├── __init__.py │ │ │ │ └── rms_norm_module.py │ │ │ │ └── triton_kernel │ │ │ │ ├── __init__.py │ │ │ │ ├── fused_add_rmsnorm.py │ │ │ │ ├── fused_brc_permute_sum.py │ │ │ │ ├── fused_div_mul_sum.py │ │ │ │ └── fused_silu_mul.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── check_ops.py │ │ │ ├── combo_utils.py │ │ │ ├── default_replacements.py │ │ │ ├── get_binary_fold_result.py │ │ │ ├── match_sub_list.py │ │ │ ├── shape_utils.py │ │ │ └── submodule_manager.py │ ├── remove_runtime_assertions.py │ └── view_to_reshape.py │ ├── runtime │ └── __init__.py │ ├── test_utils.py │ ├── utils.py │ └── version.py ├── tests ├── __init__.py ├── common │ ├── __init__.py │ ├── test_cache.py │ ├── test_change_tensor_like.py │ ├── test_combine_pointwise_sink.py │ ├── test_combine_pointwise_source.py │ ├── test_compile_time.py │ ├── test_constant_fold.py │ ├── test_convert_gelu.py │ ├── test_cse.py │ ├── test_custom_op.py │ ├── test_dynamic_shape.py │ ├── test_filter_pattern.py │ ├── test_fold_add.py │ ├── test_fold_cat.py │ ├── test_fold_clone.py │ ├── test_fold_div.py │ ├── test_fold_expand.py │ ├── test_fold_mul.py │ ├── test_fold_reduce.py │ ├── test_fold_slice_like.py │ ├── test_fold_squeeze.py │ ├── test_fold_stack.py │ ├── test_fold_sub.py │ ├── test_fold_to_copy.py │ ├── test_fold_view.py │ ├── test_fold_where.py │ ├── test_fuse_addmm.py │ ├── test_fuse_addn.py │ ├── test_fuse_baddbmm.py │ ├── test_fuse_layernorm.py │ ├── test_fuse_rmsnorm.py │ ├── test_fuse_sdpa.py │ ├── test_guard_filter.py │ ├── test_inference.py │ ├── test_inplace.py │ ├── test_interceptor.py │ ├── test_loop.py │ ├── test_models.py │ ├── test_pattern_manager.py │ ├── test_plugin_pattern.py │ ├── test_sink_view.py │ ├── test_symshape.py │ ├── test_training.py │ └── test_xpu_graph_utils.py ├── conftest.py ├── mlu │ ├── __init__.py │ ├── test_cache.py │ ├── test_combo_matmul.py │ ├── test_constant_fold_inductor.py │ ├── test_cppwrapper.py │ ├── test_custom_batchdenselayer.py │ ├── test_custom_denselayer.py │ ├── test_custom_layernorm.py │ ├── test_custom_rmsnorm.py │ ├── test_ddp_matmul.py │ ├── test_dist_utils.py │ ├── test_flash_attention.py │ ├── test_fsdp_matmul.py │ ├── test_fuse_addlayernorm.py │ ├── test_fuse_densetower.py │ ├── test_fuse_mul_sum_cat.py │ ├── test_fuse_scale_layernorm.py │ ├── test_fuse_slice.py │ ├── test_fuse_slice_cat.py │ ├── test_fuse_slice_stack_sum.py │ ├── test_fuse_slice_where_cat.py │ ├── test_inference.py │ ├── test_interceptor.py │ ├── test_linear_attention.py │ ├── test_linear_to_matmul.py │ ├── test_mixedtype_layernorm.py │ ├── test_numpy_float.py │ ├── test_repeat_to_expand.py │ ├── test_slice_sum_cat.py │ ├── test_symshape.py │ ├── test_tp_matmul.py │ └── test_training.py └── npu │ ├── __init__.py │ ├── test_acl_graph_mempool_reuse.py │ ├── test_add_rmsnorm.py │ ├── test_base.py │ ├── test_convert_gemm_like_weights_ND2NZ.py │ ├── test_custom_rmsnorm.py │ ├── test_ddp_matmul.py │ ├── test_dist_utils.py │ ├── test_fsdp_matmul.py │ ├── test_fuse_div_mul_sum.py │ ├── test_fuse_permute_sum.py │ ├── test_npu_compilation_cache.py │ ├── test_silu_mul.py │ ├── test_super_kernel_scope.py │ └── test_tp_matmul.py └── tools └── linter ├── black_linter.py └── pip_init.py /.commit-msg-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.commit-msg-config.yaml -------------------------------------------------------------------------------- /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.coveragerc -------------------------------------------------------------------------------- /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug-report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/ISSUE_TEMPLATE/bug-report.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/chores.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/ISSUE_TEMPLATE/chores.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- 1 | blank_issues_enabled: false 2 | -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/ISSUE_TEMPLATE/feature-request.yml -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/codecov.yml -------------------------------------------------------------------------------- /.github/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/release.yml -------------------------------------------------------------------------------- /.github/workflows/auto_release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/workflows/auto_release.yml -------------------------------------------------------------------------------- /.github/workflows/ci-for-mlu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/workflows/ci-for-mlu.yml -------------------------------------------------------------------------------- /.github/workflows/ci-for-npu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/workflows/ci-for-npu.yml -------------------------------------------------------------------------------- /.github/workflows/lint_check.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/workflows/lint_check.yml -------------------------------------------------------------------------------- /.github/workflows/scripts/rest_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/workflows/scripts/rest_api.py -------------------------------------------------------------------------------- /.github/workflows/scripts/upload_to_byted_luban.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.github/workflows/scripts/upload_to_byted_luban.sh -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.gitignore -------------------------------------------------------------------------------- /.lintrunner.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.lintrunner.toml -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/README.md -------------------------------------------------------------------------------- /docs/RELEASE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/docs/RELEASE.md -------------------------------------------------------------------------------- /docs/RELEASE.md.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/docs/RELEASE.md.template -------------------------------------------------------------------------------- /docs/RELEASE_CURRENT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/docs/RELEASE_CURRENT.md -------------------------------------------------------------------------------- /docs/compilation_pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/docs/compilation_pipeline.png -------------------------------------------------------------------------------- /docs/update_release_note.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/docs/update_release_note.sh -------------------------------------------------------------------------------- /docs/xpu_graph_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/docs/xpu_graph_arch.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/scripts/build.py -------------------------------------------------------------------------------- /scripts/commit_msg_check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/scripts/commit_msg_check.sh -------------------------------------------------------------------------------- /scripts/install_githooks.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/scripts/install_githooks.sh -------------------------------------------------------------------------------- /scripts/lint_check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/scripts/lint_check.sh -------------------------------------------------------------------------------- /src/xpu_graph/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/accuracy_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/accuracy_utils.py -------------------------------------------------------------------------------- /src/xpu_graph/backends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/backends/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/backends/mlu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/backends/mlu.py -------------------------------------------------------------------------------- /src/xpu_graph/backends/npu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/backends/npu.py -------------------------------------------------------------------------------- /src/xpu_graph/cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/cache.py -------------------------------------------------------------------------------- /src/xpu_graph/compiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/compiler.py -------------------------------------------------------------------------------- /src/xpu_graph/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/config.py -------------------------------------------------------------------------------- /src/xpu_graph/constant_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/constant_manager.py -------------------------------------------------------------------------------- /src/xpu_graph/fx_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/fx_utils.py -------------------------------------------------------------------------------- /src/xpu_graph/guard_filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/guard_filters.py -------------------------------------------------------------------------------- /src/xpu_graph/interceptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/interceptor.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/xpu_graph/passes/constant_folding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/constant_folding.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/cse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/cse.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/dce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/dce.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/fake_tensor_prop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/fake_tensor_prop.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/optimizer.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/pass_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/pass_manager.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/change_tensor_like.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/change_tensor_like.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/combine_pointwise_sink.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/combine_pointwise_sink.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/combine_pointwise_source.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/combine_pointwise_source.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_add.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_clone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_clone.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_detach.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_detach.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_div.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_div.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_expand.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_mul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_mul.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_reduce.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_slice_like.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_slice_like.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_squeeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_squeeze.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_stack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_stack.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_sub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_sub.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_to_copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_to_copy.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_view.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fold_where.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fold_where.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fuse_addmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fuse_addmm.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fuse_addn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fuse_addn.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fuse_baddbmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fuse_baddbmm.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fuse_layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fuse_layernorm.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fuse_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fuse_rmsnorm.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/fuse_scaled_dot_product_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/fuse_scaled_dot_product_attention.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/gelu.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/gelu.md -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/gelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/gelu.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/common/sink_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/common/sink_view.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/pattern.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/pattern_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/pattern_manager.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/plugin_pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/plugin_pattern.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/structure/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/structure/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/structure/custom_denselayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/structure/custom_denselayer.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/structure/custom_layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/structure/custom_layernorm.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/structure/custom_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/structure/custom_rmsnorm.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/structure/fuse_combosum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/structure/fuse_combosum.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/structure/fuse_multiple_slice_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/structure/fuse_multiple_slice_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/structure/fuse_slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/structure/fuse_slice.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/structure/fuse_slice_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/structure/fuse_slice_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/structure/fuse_slice_stack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/structure/fuse_slice_stack.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/combo_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/combo_matmul.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/combo_matmul_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/combo_matmul_utils.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/fuse_add_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/fuse_add_norm.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/fuse_densetower.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/fuse_densetower.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/fuse_linear_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/fuse_linear_attention.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/fuse_mul_sum_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/fuse_mul_sum_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/fuse_scale_layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/fuse_scale_layernorm.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/fuse_slice_where_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/fuse_slice_where_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/fuse_sum_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/fuse_sum_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/linear_to_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/linear_to_matmul.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/repeat_to_expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/repeat_to_expand.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/structure_replacements/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/structure_replacements/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/structure_replacements/dense_layer_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/structure_replacements/dense_layer_modules.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/structure_replacements/norm_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/structure_replacements/norm_modules.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_linear_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_linear_attn.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_mul_sum_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_mul_sum_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_serial_mm_2dot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_serial_mm_2dot.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_serial_mm_3dot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_serial_mm_3dot.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_slice.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_slice_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_slice_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_slice_sum_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_slice_sum_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_slice_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_slice_v2.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_slice_where_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_slice_where_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_sum_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_sum_3d.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_sum_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/fused_sum_cat.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/get_mlu_devinfo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/get_mlu_devinfo.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/libentry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/libentry.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/linear_attention_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/mlu/triton_kernel/linear_attention_kernel.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/check_npu_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/check_npu_ops.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/convert_gemm_like_weights_ND2NZ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/convert_gemm_like_weights_ND2NZ.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/fuse_add_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/fuse_add_rmsnorm.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/fuse_brc_permute_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/fuse_brc_permute_sum.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/fuse_div_mul_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/fuse_div_mul_sum.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/fuse_silu_mul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/fuse_silu_mul.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/remove_layernorm_cast_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/remove_layernorm_cast_inference.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/remove_rmsnorm_cast_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/remove_rmsnorm_cast_inference.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/scoped_super_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/scoped_super_kernel.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/structure_replacements/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/structure_replacements/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/structure_replacements/rms_norm_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/structure_replacements/rms_norm_module.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/triton_kernel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/triton_kernel/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/triton_kernel/fused_add_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/triton_kernel/fused_add_rmsnorm.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/triton_kernel/fused_brc_permute_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/triton_kernel/fused_brc_permute_sum.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/triton_kernel/fused_div_mul_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/triton_kernel/fused_div_mul_sum.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/targets/npu/triton_kernel/fused_silu_mul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/targets/npu/triton_kernel/fused_silu_mul.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/utils/check_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/utils/check_ops.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/utils/combo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/utils/combo_utils.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/utils/default_replacements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/utils/default_replacements.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/utils/get_binary_fold_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/utils/get_binary_fold_result.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/utils/match_sub_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/utils/match_sub_list.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/utils/shape_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/utils/shape_utils.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/patterns/utils/submodule_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/patterns/utils/submodule_manager.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/remove_runtime_assertions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/remove_runtime_assertions.py -------------------------------------------------------------------------------- /src/xpu_graph/passes/view_to_reshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/passes/view_to_reshape.py -------------------------------------------------------------------------------- /src/xpu_graph/runtime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/runtime/__init__.py -------------------------------------------------------------------------------- /src/xpu_graph/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/test_utils.py -------------------------------------------------------------------------------- /src/xpu_graph/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/src/xpu_graph/utils.py -------------------------------------------------------------------------------- /src/xpu_graph/version.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.10.0" 2 | -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/common/test_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_cache.py -------------------------------------------------------------------------------- /tests/common/test_change_tensor_like.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_change_tensor_like.py -------------------------------------------------------------------------------- /tests/common/test_combine_pointwise_sink.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_combine_pointwise_sink.py -------------------------------------------------------------------------------- /tests/common/test_combine_pointwise_source.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_combine_pointwise_source.py -------------------------------------------------------------------------------- /tests/common/test_compile_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_compile_time.py -------------------------------------------------------------------------------- /tests/common/test_constant_fold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_constant_fold.py -------------------------------------------------------------------------------- /tests/common/test_convert_gelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_convert_gelu.py -------------------------------------------------------------------------------- /tests/common/test_cse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_cse.py -------------------------------------------------------------------------------- /tests/common/test_custom_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_custom_op.py -------------------------------------------------------------------------------- /tests/common/test_dynamic_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_dynamic_shape.py -------------------------------------------------------------------------------- /tests/common/test_filter_pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_filter_pattern.py -------------------------------------------------------------------------------- /tests/common/test_fold_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_add.py -------------------------------------------------------------------------------- /tests/common/test_fold_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_cat.py -------------------------------------------------------------------------------- /tests/common/test_fold_clone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_clone.py -------------------------------------------------------------------------------- /tests/common/test_fold_div.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_div.py -------------------------------------------------------------------------------- /tests/common/test_fold_expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_expand.py -------------------------------------------------------------------------------- /tests/common/test_fold_mul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_mul.py -------------------------------------------------------------------------------- /tests/common/test_fold_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_reduce.py -------------------------------------------------------------------------------- /tests/common/test_fold_slice_like.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_slice_like.py -------------------------------------------------------------------------------- /tests/common/test_fold_squeeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_squeeze.py -------------------------------------------------------------------------------- /tests/common/test_fold_stack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_stack.py -------------------------------------------------------------------------------- /tests/common/test_fold_sub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_sub.py -------------------------------------------------------------------------------- /tests/common/test_fold_to_copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_to_copy.py -------------------------------------------------------------------------------- /tests/common/test_fold_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_view.py -------------------------------------------------------------------------------- /tests/common/test_fold_where.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fold_where.py -------------------------------------------------------------------------------- /tests/common/test_fuse_addmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fuse_addmm.py -------------------------------------------------------------------------------- /tests/common/test_fuse_addn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fuse_addn.py -------------------------------------------------------------------------------- /tests/common/test_fuse_baddbmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fuse_baddbmm.py -------------------------------------------------------------------------------- /tests/common/test_fuse_layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fuse_layernorm.py -------------------------------------------------------------------------------- /tests/common/test_fuse_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fuse_rmsnorm.py -------------------------------------------------------------------------------- /tests/common/test_fuse_sdpa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_fuse_sdpa.py -------------------------------------------------------------------------------- /tests/common/test_guard_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_guard_filter.py -------------------------------------------------------------------------------- /tests/common/test_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_inference.py -------------------------------------------------------------------------------- /tests/common/test_inplace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_inplace.py -------------------------------------------------------------------------------- /tests/common/test_interceptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_interceptor.py -------------------------------------------------------------------------------- /tests/common/test_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_loop.py -------------------------------------------------------------------------------- /tests/common/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_models.py -------------------------------------------------------------------------------- /tests/common/test_pattern_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_pattern_manager.py -------------------------------------------------------------------------------- /tests/common/test_plugin_pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_plugin_pattern.py -------------------------------------------------------------------------------- /tests/common/test_sink_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_sink_view.py -------------------------------------------------------------------------------- /tests/common/test_symshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_symshape.py -------------------------------------------------------------------------------- /tests/common/test_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_training.py -------------------------------------------------------------------------------- /tests/common/test_xpu_graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/common/test_xpu_graph_utils.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/mlu/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/mlu/test_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_cache.py -------------------------------------------------------------------------------- /tests/mlu/test_combo_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_combo_matmul.py -------------------------------------------------------------------------------- /tests/mlu/test_constant_fold_inductor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_constant_fold_inductor.py -------------------------------------------------------------------------------- /tests/mlu/test_cppwrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_cppwrapper.py -------------------------------------------------------------------------------- /tests/mlu/test_custom_batchdenselayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_custom_batchdenselayer.py -------------------------------------------------------------------------------- /tests/mlu/test_custom_denselayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_custom_denselayer.py -------------------------------------------------------------------------------- /tests/mlu/test_custom_layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_custom_layernorm.py -------------------------------------------------------------------------------- /tests/mlu/test_custom_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_custom_rmsnorm.py -------------------------------------------------------------------------------- /tests/mlu/test_ddp_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_ddp_matmul.py -------------------------------------------------------------------------------- /tests/mlu/test_dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_dist_utils.py -------------------------------------------------------------------------------- /tests/mlu/test_flash_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_flash_attention.py -------------------------------------------------------------------------------- /tests/mlu/test_fsdp_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_fsdp_matmul.py -------------------------------------------------------------------------------- /tests/mlu/test_fuse_addlayernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_fuse_addlayernorm.py -------------------------------------------------------------------------------- /tests/mlu/test_fuse_densetower.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_fuse_densetower.py -------------------------------------------------------------------------------- /tests/mlu/test_fuse_mul_sum_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_fuse_mul_sum_cat.py -------------------------------------------------------------------------------- /tests/mlu/test_fuse_scale_layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_fuse_scale_layernorm.py -------------------------------------------------------------------------------- /tests/mlu/test_fuse_slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_fuse_slice.py -------------------------------------------------------------------------------- /tests/mlu/test_fuse_slice_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_fuse_slice_cat.py -------------------------------------------------------------------------------- /tests/mlu/test_fuse_slice_stack_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_fuse_slice_stack_sum.py -------------------------------------------------------------------------------- /tests/mlu/test_fuse_slice_where_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_fuse_slice_where_cat.py -------------------------------------------------------------------------------- /tests/mlu/test_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_inference.py -------------------------------------------------------------------------------- /tests/mlu/test_interceptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_interceptor.py -------------------------------------------------------------------------------- /tests/mlu/test_linear_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_linear_attention.py -------------------------------------------------------------------------------- /tests/mlu/test_linear_to_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_linear_to_matmul.py -------------------------------------------------------------------------------- /tests/mlu/test_mixedtype_layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_mixedtype_layernorm.py -------------------------------------------------------------------------------- /tests/mlu/test_numpy_float.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_numpy_float.py -------------------------------------------------------------------------------- /tests/mlu/test_repeat_to_expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_repeat_to_expand.py -------------------------------------------------------------------------------- /tests/mlu/test_slice_sum_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_slice_sum_cat.py -------------------------------------------------------------------------------- /tests/mlu/test_symshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_symshape.py -------------------------------------------------------------------------------- /tests/mlu/test_tp_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_tp_matmul.py -------------------------------------------------------------------------------- /tests/mlu/test_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/mlu/test_training.py -------------------------------------------------------------------------------- /tests/npu/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/npu/test_acl_graph_mempool_reuse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_acl_graph_mempool_reuse.py -------------------------------------------------------------------------------- /tests/npu/test_add_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_add_rmsnorm.py -------------------------------------------------------------------------------- /tests/npu/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_base.py -------------------------------------------------------------------------------- /tests/npu/test_convert_gemm_like_weights_ND2NZ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_convert_gemm_like_weights_ND2NZ.py -------------------------------------------------------------------------------- /tests/npu/test_custom_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_custom_rmsnorm.py -------------------------------------------------------------------------------- /tests/npu/test_ddp_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_ddp_matmul.py -------------------------------------------------------------------------------- /tests/npu/test_dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_dist_utils.py -------------------------------------------------------------------------------- /tests/npu/test_fsdp_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_fsdp_matmul.py -------------------------------------------------------------------------------- /tests/npu/test_fuse_div_mul_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_fuse_div_mul_sum.py -------------------------------------------------------------------------------- /tests/npu/test_fuse_permute_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_fuse_permute_sum.py -------------------------------------------------------------------------------- /tests/npu/test_npu_compilation_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_npu_compilation_cache.py -------------------------------------------------------------------------------- /tests/npu/test_silu_mul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_silu_mul.py -------------------------------------------------------------------------------- /tests/npu/test_super_kernel_scope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_super_kernel_scope.py -------------------------------------------------------------------------------- /tests/npu/test_tp_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tests/npu/test_tp_matmul.py -------------------------------------------------------------------------------- /tools/linter/black_linter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tools/linter/black_linter.py -------------------------------------------------------------------------------- /tools/linter/pip_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XPU-Forces/xpu_graph/HEAD/tools/linter/pip_init.py --------------------------------------------------------------------------------