path # lines of code ci/scripts/github/github_skipped_tests_comment.py 200 ci/scripts/github/github_cc_reviewers.py 77 ci/scripts/github/__init__.py 1 ci/scripts/github/update_branch.py 100 ci/scripts/github/ping_reviewers.py 151 ci/scripts/github/github_docs_comment.py 23 ci/scripts/github/github_tag_teams.py 191 ci/scripts/github/github_pr_comment.py 65 ci/scripts/github/github_tvmbot.py 489 ci/scripts/github/github_commenter.py 81 ci/scripts/jenkins/git_skip_ci.py 36 ci/scripts/jenkins/http_utils.py 13 ci/scripts/jenkins/check_pr.py 88 ci/scripts/jenkins/determine_docker_images.py 95 ci/scripts/jenkins/pytest_ids.py 19 ci/scripts/jenkins/git_utils.py 150 ci/scripts/jenkins/s3.py 93 ci/scripts/jenkins/pytest_wrapper.py 84 ci/scripts/jenkins/git_skip_ci_globs.py 48 ci/scripts/jenkins/should_rebuild_docker.py 87 ci/scripts/jenkins/open_docker_update_pr.py 132 ci/scripts/jenkins/cmd_utils.py 65 ci/jenkins/data.py 59 ci/jenkins/generate.py 129 rust/tvm-rt/src/to_function.rs 249 rust/tvm-rt/src/errors.rs 84 rust/tvm-rt/src/module.rs 98 rust/tvm-rt/src/object/object_ptr.rs 441 rust/tvm-rt/src/object/mod.rs 62 rust/tvm-rt/src/ndarray.rs 425 rust/tvm-rt/src/string.rs 121 rust/tvm-rt/src/device.rs 81 rust/tvm-rt/src/lib.rs 114 rust/tvm-rt/src/function.rs 244 rust/tvm-rt/Cargo.toml 90 rust/Cargo.toml 22 rust/tvm-macros/src/import_module.rs 120 rust/tvm-macros/src/lib.rs 38 rust/tvm-macros/src/external.rs 172 rust/tvm-macros/src/object.rs 142 rust/tvm-macros/src/util.rs 32 rust/tvm-macros/Cargo.toml 34 rust/tvm-sys/src/byte_array.rs 80 rust/tvm-sys/src/errors.rs 24 rust/tvm-sys/src/array.rs 30 rust/tvm-sys/src/value.rs 84 rust/tvm-sys/src/packed_func.rs 350 rust/tvm-sys/src/device.rs 228 rust/tvm-sys/src/lib.rs 54 rust/tvm-sys/src/datatype.rs 180 rust/tvm-sys/Cargo.toml 77 rust/tvm-sys/build.rs 248 include/tvm/tir/utils.h 61 include/tvm/tir/buffer.h 129 include/tvm/tir/stmt_functor.h 276 include/tvm/tir/analysis.h 76 include/tvm/tir/var.h 150 include/tvm/tir/op.h 378 include/tvm/tir/expr_functor.h 192 include/tvm/tir/builtin.h 128 include/tvm/tir/data_layout.h 160 include/tvm/tir/data_type_rewriter.h 89 include/tvm/tir/op_attr_types.h 53 include/tvm/tir/index_map.h 62 include/tvm/tir/block_dependence_info.h 31 include/tvm/tir/stmt.h 875 include/tvm/tir/function.h 82 include/tvm/tir/block_scope.h 114 include/tvm/tir/schedule/trace.h 41 include/tvm/tir/schedule/state.h 67 include/tvm/tir/schedule/schedule.h 174 include/tvm/tir/schedule/instruction.h 104 include/tvm/tir/transform.h 98 include/tvm/tir/expr.h 688 include/tvm/runtime/serializer.h 40 include/tvm/runtime/registry.h 33 include/tvm/runtime/relax_vm/executable.h 75 include/tvm/runtime/relax_vm/ndarray_cache_support.h 41 include/tvm/runtime/relax_vm/builtin.h 19 include/tvm/runtime/relax_vm/vm.h 79 include/tvm/runtime/relax_vm/bytecode.h 99 include/tvm/runtime/builtin_fp16.h 12 include/tvm/runtime/memory.h 11 include/tvm/runtime/memory/memory_manager.h 89 include/tvm/runtime/profiling.h 142 include/tvm/runtime/object.h 70 include/tvm/runtime/data_type.h 206 include/tvm/runtime/nvtx.h 24 include/tvm/runtime/c_backend_api.h 27 include/tvm/runtime/contrib/papi.h 13 include/tvm/runtime/container/shape_tuple.h 21 include/tvm/runtime/container/base.h 146 include/tvm/runtime/container/map.h 11 include/tvm/runtime/container/variant.h 10 include/tvm/runtime/container/string.h 12 include/tvm/runtime/container/array.h 12 include/tvm/runtime/container/optional.h 11 include/tvm/runtime/module.h 98 include/tvm/runtime/disco/session.h 129 include/tvm/runtime/disco/cuda_ipc_memory.h 32 include/tvm/runtime/disco/disco_worker.h 44 include/tvm/runtime/disco/builtin.h 47 include/tvm/runtime/logging.h 379 include/tvm/runtime/threading_backend.h 93 include/tvm/runtime/c_runtime_api.h 164 include/tvm/runtime/ndarray.h 187 include/tvm/runtime/device_api.h 132 include/tvm/runtime/packed_func.h 332 include/tvm/script/ir_builder/tir/ir.h 139 include/tvm/script/ir_builder/tir/frame.h 356 include/tvm/script/ir_builder/base.h 111 include/tvm/script/ir_builder/ir/ir.h 19 include/tvm/script/ir_builder/ir/frame.h 40 include/tvm/script/ir_builder/relax/ir.h 33 include/tvm/script/ir_builder/relax/frame.h 147 include/tvm/script/printer/doc.h 531 include/tvm/script/printer/ir_docsifier.h 176 include/tvm/script/printer/ir_docsifier_functor.h 112 include/tvm/arith/int_set.h 77 include/tvm/arith/iter_affine_map.h 147 include/tvm/arith/analyzer.h 213 include/tvm/arith/bound.h 23 include/tvm/arith/pattern.h 11 include/tvm/arith/int_solver.h 124 include/tvm/support/with.h 30 include/tvm/support/parallel_for.h 16 include/tvm/support/span.h 59 include/tvm/support/random_engine.h 51 include/tvm/te/operation.h 195 include/tvm/te/tensor.h 139 include/tvm/meta_schedule/measure_callback.h 61 include/tvm/meta_schedule/database.h 208 include/tvm/meta_schedule/cost_model.h 65 include/tvm/meta_schedule/schedule_rule.h 106 include/tvm/meta_schedule/tune_context.h 59 include/tvm/meta_schedule/extracted_task.h 45 include/tvm/meta_schedule/search_strategy.h 93 include/tvm/meta_schedule/builder.h 77 include/tvm/meta_schedule/feature_extractor.h 49 include/tvm/meta_schedule/space_generator.h 80 include/tvm/meta_schedule/postproc.h 68 include/tvm/meta_schedule/runner.h 102 include/tvm/meta_schedule/mutator.h 66 include/tvm/meta_schedule/arg_info.h 50 include/tvm/meta_schedule/measure_candidate.h 28 include/tvm/meta_schedule/schedule/cuda/thread_bind.h 19 include/tvm/meta_schedule/schedule/generic/winograd.h 9 include/tvm/meta_schedule/profiler.h 52 include/tvm/meta_schedule/task_scheduler.h 125 include/tvm/ir/env_func.h 69 include/tvm/ir/type_functor.h 69 include/tvm/ir/global_var_supply.h 36 include/tvm/ir/name_supply.h 59 include/tvm/ir/analysis.h 22 include/tvm/ir/type.h 115 include/tvm/ir/source_map.h 130 include/tvm/ir/instrument.h 32 include/tvm/ir/op.h 164 include/tvm/ir/attrs.h 581 include/tvm/ir/replace_global_vars.h 16 include/tvm/ir/module.h 94 include/tvm/ir/diagnostic.h 126 include/tvm/ir/function.h 53 include/tvm/ir/transform.h 201 include/tvm/ir/global_info.h 60 include/tvm/ir/expr.h 340 include/tvm/node/repr_printer.h 60 include/tvm/node/attr_registry_map.h 64 include/tvm/node/functor.h 70 include/tvm/node/object_path.h 144 include/tvm/node/serialization.h 10 include/tvm/node/script_printer.h 78 include/tvm/node/structural_hash.h 135 include/tvm/node/reflection.h 201 include/tvm/node/node.h 28 include/tvm/node/structural_equal.h 197 include/tvm/relax/utils.h 24 include/tvm/relax/distributed/axis_group_graph.h 357 include/tvm/relax/distributed/struct_info.h 98 include/tvm/relax/distributed/transform.h 27 include/tvm/relax/distributed/global_info.h 47 include/tvm/relax/exec_builder.h 56 include/tvm/relax/nested_msg.h 295 include/tvm/relax/dataflow_pattern.h 477 include/tvm/relax/dataflow_matcher.h 23 include/tvm/relax/analysis.h 77 include/tvm/relax/type.h 88 include/tvm/relax/expr_functor.h 276 include/tvm/relax/dataflow_pattern_functor.h 118 include/tvm/relax/struct_info_functor.h 94 include/tvm/relax/tir_pattern.h 29 include/tvm/relax/binding_rewrite.h 57 include/tvm/relax/op_attr_types.h 28 include/tvm/relax/tuning_api.h 189 include/tvm/relax/block_builder.h 50 include/tvm/relax/struct_info.h 200 include/tvm/relax/attrs/distributed.h 18 include/tvm/relax/attrs/datatype.h 20 include/tvm/relax/attrs/nn.h 519 include/tvm/relax/attrs/create.h 21 include/tvm/relax/attrs/search.h 18 include/tvm/relax/attrs/index.h 24 include/tvm/relax/attrs/op.h 56 include/tvm/relax/attrs/linear_algebra.h 20 include/tvm/relax/attrs/qdq.h 20 include/tvm/relax/attrs/manipulate.h 157 include/tvm/relax/attrs/sampling.h 16 include/tvm/relax/attrs/sorting.h 63 include/tvm/relax/attrs/statistical.h 34 include/tvm/relax/attrs/image.h 51 include/tvm/relax/attrs/ccl.h 44 include/tvm/relax/backend.h 12 include/tvm/relax/transform.h 125 include/tvm/relax/expr.h 612 include/tvm/topi/utils.h 22 include/tvm/topi/elemwise.h 252 include/tvm/topi/nn.h 461 include/tvm/topi/tags.h 29 include/tvm/topi/broadcast.h 125 include/tvm/topi/contrib/rocblas.h 35 include/tvm/topi/contrib/cublas.h 36 include/tvm/topi/einsum.h 37 include/tvm/topi/nn/group_norm.h 124 include/tvm/topi/nn/local_response_norm.h 54 include/tvm/topi/nn/rms_norm.h 64 include/tvm/topi/nn/mapping.h 25 include/tvm/topi/nn/softmax.h 87 include/tvm/topi/nn/instance_norm.h 19 include/tvm/topi/nn/layer_norm.h 89 include/tvm/topi/nn/pooling.h 449 include/tvm/topi/nn/bnn.h 71 include/tvm/topi/nn/bias_add.h 23 include/tvm/topi/nn/dense.h 38 include/tvm/topi/nn/dilate.h 54 include/tvm/topi/nn/flatten.h 42 include/tvm/topi/detail/constant_utils.h 61 include/tvm/topi/detail/tensor_utils.h 82 include/tvm/topi/detail/extern.h 60 include/tvm/topi/detail/pad_utils.h 21 include/tvm/topi/detail/array_utils.h 20 include/tvm/topi/detail/strided_slice.h 118 include/tvm/topi/detail/fuse.h 16 include/tvm/topi/detail/broadcast.h 115 include/tvm/topi/detail/ravel_unravel.h 36 include/tvm/topi/vision/reorg.h 39 include/tvm/topi/transform.h 1493 include/tvm/topi/reduction.h 332 cmake/modules/LLVM.cmake 52 cmake/modules/Git.cmake 54 cmake/modules/Vulkan.cmake 36 cmake/modules/OpenCL.cmake 80 cmake/modules/Logging.cmake 26 cmake/modules/OpenMP.cmake 47 cmake/modules/ClangFlags.cmake 86 cmake/modules/contrib/ArmComputeLib.cmake 70 cmake/modules/contrib/CUTLASS.cmake 71 cmake/modules/contrib/Random.cmake 21 cmake/modules/contrib/vllm.cmake 23 cmake/modules/contrib/BLAS.cmake 68 cmake/modules/contrib/Posit.cmake 24 cmake/modules/contrib/NNAPI.cmake 34 cmake/modules/contrib/DNNL.cmake 57 cmake/modules/contrib/PAPI.cmake 26 cmake/modules/contrib/Sort.cmake 21 cmake/modules/contrib/CoreML.cmake 24 cmake/modules/contrib/TensorRT.cmake 53 cmake/modules/contrib/TFLite.cmake 52 cmake/modules/contrib/AMX.cmake 22 cmake/modules/contrib/MSCCLPP.cmake 46 cmake/modules/contrib/CLML.cmake 78 cmake/modules/contrib/MSC.cmake 27 cmake/modules/contrib/BNNS.cmake 25 cmake/modules/contrib/Mrvl.cmake 27 cmake/modules/ROCM.cmake 76 cmake/modules/Hexagon.cmake 303 cmake/modules/LibInfo.cmake 133 cmake/modules/Metal.cmake 32 cmake/modules/HexagonSDK.cmake 191 cmake/modules/CUDA.cmake 130 cmake/utils/FindRCCL.cmake 51 cmake/utils/Linker.cmake 58 cmake/utils/FindNCCL.cmake 56 cmake/utils/Utils.cmake 97 cmake/utils/CCache.cmake 64 cmake/utils/CRTConfig.cmake 33 cmake/utils/FindCUDA.cmake 148 cmake/utils/Summary.cmake 68 cmake/utils/FindLLVM.cmake 232 cmake/utils/FindROCM.cmake 70 cmake/utils/FindOpenCL.cmake 67 cmake/utils/FindVulkan.cmake 89 cmake/utils/FindNVSHMEM.cmake 48 cmake/config.cmake 355 cmake/libs/Libbacktrace.cmake 60 src/tir/analysis/is_pure_function.cc 60 src/tir/analysis/block_access_region_detector.cc 306 src/tir/analysis/calculate_allocated_memory.cc 154 src/tir/analysis/check_contains.h 21 src/tir/analysis/buffer_access_lca_detector.cc 236 src/tir/analysis/verify_gpu_code.cc 282 src/tir/analysis/var_touch.cc 44 src/tir/analysis/stmt_finding.cc 113 src/tir/analysis/collect_call_map.cc 25 src/tir/analysis/control_flow_graph.cc 1238 src/tir/analysis/verify_well_formed.cc 253 src/tir/analysis/identify_memcpy.cc 222 src/tir/analysis/control_flow_graph.h 129 src/tir/analysis/var_use_def_analysis.h 40 src/tir/analysis/verify_ssa.cc 110 src/tir/analysis/deep_equal.cc 43 src/tir/analysis/var_use_def_analysis.cc 155 src/tir/analysis/estimate_flops.cc 209 src/tir/analysis/side_effect.cc 43 src/tir/analysis/expr_complexity.cc 22 src/tir/analysis/verify_memory.cc 129 src/tir/analysis/check_contains.cc 32 src/tir/analysis/oob_checker.cc 84 src/tir/transforms/merge_shared_memory_allocations.cc 494 src/tir/transforms/ir_utils.h 145 src/tir/transforms/unify_thread_binding.cc 122 src/tir/transforms/lower_custom_datatypes.cc 182 src/tir/transforms/remove_assume.cc 37 src/tir/transforms/inject_virtual_thread.cc 403 src/tir/transforms/lower_match_buffer.cc 203 src/tir/transforms/memhammer_coalesce.cc 174 src/tir/transforms/plan_update_buffer_allocation_location.cc 196 src/tir/transforms/dtype_conversion.cc 67 src/tir/transforms/replace_global_vars.cc 45 src/tir/transforms/inject_ptx_async_copy.cc 144 src/tir/transforms/common_subexpr_elim_tools.h 104 src/tir/transforms/force_narrow_index_to_i32.cc 54 src/tir/transforms/rewrite_unsafe_select.cc 108 src/tir/transforms/default_gpu_schedule.cc 117 src/tir/transforms/thread_storage_sync.cc 370 src/tir/transforms/dtype_conversion.h 70 src/tir/transforms/simplify.h 10 src/tir/transforms/lower_opaque_block.cc 149 src/tir/transforms/unsupported_dtype_legalize.cc 638 src/tir/transforms/lower_tvm_builtin.cc 490 src/tir/transforms/lower_vtcm_alloc.cc 49 src/tir/transforms/remove_store_undef.cc 125 src/tir/transforms/loop_partition.cc 590 src/tir/transforms/inject_permuted_layout.cc 189 src/tir/transforms/renew_defs.cc 219 src/tir/transforms/hoist_expression.cc 459 src/tir/transforms/lower_async_dma.cc 102 src/tir/transforms/convert_blocks_to_opaque.cc 76 src/tir/transforms/memhammer_intermediate_stage.cc 360 src/tir/transforms/skip_assert.cc 28 src/tir/transforms/transform_mma_buffer_layout.cc 136 src/tir/transforms/profile_instrumentation.cc 220 src/tir/transforms/common_subexpr_elim_tools.cc 356 src/tir/transforms/lower_thread_allreduce.cc 585 src/tir/transforms/lower_device_storage_access_info.cc 95 src/tir/transforms/memhammer_rewrite_rule.h 156 src/tir/transforms/inline_private_functions.cc 188 src/tir/transforms/decorate_device_scope.cc 23 src/tir/transforms/storage_rewrite.cc 1225 src/tir/transforms/vectorize_loop.cc 780 src/tir/transforms/unroll_loop.cc 219 src/tir/transforms/lower_intrin.cc 246 src/tir/transforms/split_host_device.cc 119 src/tir/transforms/arg_binder.cc 263 src/tir/transforms/manifest_shared_memory_local_stage.cc 195 src/tir/transforms/renormalize_split_pattern.cc 149 src/tir/transforms/bind_params.cc 97 src/tir/transforms/replace_selected_expr.cc 41 src/tir/transforms/update_pointer_storage_scope.cc 74 src/tir/transforms/memhammer_lower_auto_copy.cc 596 src/tir/transforms/narrow_datatype.cc 217 src/tir/transforms/extract_constants.cc 72 src/tir/transforms/tensorcore_infer_fragment.cc 158 src/tir/transforms/remap_thread_axis.cc 70 src/tir/transforms/lower_warp_memory.cc 294 src/tir/transforms/replace_selected_expr.h 30 src/tir/transforms/primfunc_utils.cc 78 src/tir/transforms/remove_weight_layout_rewrite_block.cc 189 src/tir/transforms/inject_rolling_buffer.cc 219 src/tir/transforms/reduce_branching_through_overcompute.cc 122 src/tir/transforms/ir_utils.cc 707 src/tir/transforms/common_subexpr_elim.cc 244 src/tir/transforms/combine_context_call.cc 78 src/tir/transforms/make_packed_api.cc 318 src/tir/transforms/common_subexpr_elim.h 40 src/tir/transforms/using_assume_to_reduce_branches.cc 235 src/tir/transforms/arg_binder.h 38 src/tir/transforms/lower_cross_thread_reduction.cc 682 src/tir/transforms/lower_init_block.cc 52 src/tir/transforms/compact_buffer_region.cc 568 src/tir/transforms/make_unpacked_api.cc 142 src/tir/transforms/bound_checker.cc 185 src/tir/transforms/lift_thread_binding.cc 148 src/tir/transforms/memhammer_tensorcore_rewrite.cc 448 src/tir/transforms/remove_no_op.h 14 src/tir/transforms/annotate_device_regions.cc 44 src/tir/transforms/inject_software_pipeline.cc 875 src/tir/transforms/inject_ptx_ldg32.cc 96 src/tir/transforms/lower_device_kernel_launch.cc 261 src/tir/transforms/remove_no_op.cc 263 src/tir/transforms/flatten_buffer.cc 190 src/tir/transforms/storage_access.h 64 src/tir/transforms/convert_for_loops_serial.cc 41 src/tir/transforms/storage_access.cc 244 src/tir/transforms/update_pointer_storage_scope.h 27 src/tir/transforms/inject_double_buffer.cc 243 src/tir/transforms/simplify.cc 266 src/tir/ir/block_dependence_info.cc 60 src/tir/ir/function.cc 126 src/tir/ir/utils.h 9 src/tir/ir/expr.cc 633 src/tir/ir/specialize.cc 295 src/tir/ir/stmt_functor.cc 731 src/tir/ir/block_scope.cc 145 src/tir/ir/script/script_complete.cc 114 src/tir/ir/script/script_complete.h 11 src/tir/ir/transform.cc 94 src/tir/ir/data_layout.cc 385 src/tir/ir/utils.cc 39 src/tir/ir/expr_functor.cc 226 src/tir/ir/data_type_rewriter.cc 593 src/tir/ir/tir_visitor_with_path.h 176 src/tir/ir/buffer.cc 515 src/tir/ir/tir_visitor_with_path.cc 315 src/tir/ir/buffer_common.h 20 src/tir/ir/functor_common.h 18 src/tir/ir/stmt.cc 575 src/tir/ir/index_map.cc 347 src/tir/schedule/analysis/analysis.cc 1724 src/tir/schedule/analysis/verify.cc 205 src/tir/schedule/analysis/reducer.cc 563 src/tir/schedule/analysis/layout.cc 179 src/tir/schedule/utils.h 224 src/tir/schedule/traced_schedule.h 103 src/tir/schedule/schedule.cc 251 src/tir/schedule/error.cc 31 src/tir/schedule/analysis.h 187 src/tir/schedule/transform.cc 456 src/tir/schedule/primitive/block_annotate.cc 336 src/tir/schedule/primitive/loop_transformation.cc 1084 src/tir/schedule/primitive/rolling_buffer.cc 353 src/tir/schedule/primitive/decompose_padding.cc 413 src/tir/schedule/primitive/annotate.cc 118 src/tir/schedule/primitive/compute_inline.cc 761 src/tir/schedule/primitive/reorder_block_iter_var.cc 108 src/tir/schedule/primitive/sampling.cc 452 src/tir/schedule/primitive/pad_einsum.cc 430 src/tir/schedule/primitive/layout_transformation.cc 1299 src/tir/schedule/primitive/cache_read_write.cc 1793 src/tir/schedule/primitive/cache_index.cc 355 src/tir/schedule/primitive/read_write_at.cc 354 src/tir/schedule/primitive/annotate_buffer_access.cc 124 src/tir/schedule/primitive/blockize_tensorize.cc 721 src/tir/schedule/primitive/for_kind.cc 186 src/tir/schedule/primitive/hide_buffer_access.cc 115 src/tir/schedule/primitive/reduction.cc 978 src/tir/schedule/primitive/get_block_loop.cc 194 src/tir/schedule/primitive/compute_at.cc 631 src/tir/schedule/concrete_schedule.h 277 src/tir/schedule/ir_comparator.h 97 src/tir/schedule/ir_comparator.cc 646 src/tir/schedule/instruction_traits.h 343 src/tir/schedule/traced_schedule.cc 645 src/tir/schedule/trace.cc 491 src/tir/schedule/error.h 42 src/tir/schedule/primitive.h 122 src/tir/schedule/state.cc 651 src/tir/schedule/instruction.cc 78 src/tir/schedule/concrete_schedule.cc 903 src/tir/schedule/transform.h 73 src/tir/op/runtime.cc 14 src/tir/op/op.cc 951 src/tir/op/builtin.cc 305 src/runtime/cuda/cuda_module.h 17 src/runtime/cuda/cuda_device_api.cc 278 src/runtime/cuda/cuda_module.cc 218 src/runtime/cuda/cuda_common.h 34 src/runtime/cuda/l2_cache_flush.cc 16 src/runtime/runtime_base.h 26 src/runtime/regex.cc 14 src/runtime/pack_args.h 263 src/runtime/meta_data.h 39 src/runtime/regex.h 9 src/runtime/thread_pool.cc 366 src/runtime/nvtx.cc 19 src/runtime/relax_vm/cuda/cuda_graph_builtin.cc 166 src/runtime/relax_vm/lm_support.cc 423 src/runtime/relax_vm/attn_backend.cc 97 src/runtime/relax_vm/rnn_state.cc 316 src/runtime/relax_vm/paged_kv_cache.cc 1839 src/runtime/relax_vm/attn_backend.h 399 src/runtime/relax_vm/vm.cc 740 src/runtime/relax_vm/hexagon/builtin.cc 44 src/runtime/relax_vm/executable.cc 470 src/runtime/relax_vm/kv_state.cc 87 src/runtime/relax_vm/ndarray_cache_support.cc 294 src/runtime/relax_vm/builtin.cc 398 src/runtime/relax_vm/attn_utils.h 690 src/runtime/relax_vm/kv_state.h 81 src/runtime/relax_vm/bytecode.cc 38 src/runtime/memory/pooled_allocator.h 92 src/runtime/memory/naive_allocator.h 59 src/runtime/memory/memory_manager.cc 204 src/runtime/library_module.cc 145 src/runtime/module.cc 167 src/runtime/texture.h 63 src/runtime/opencl/opencl_module_spirv.cc 97 src/runtime/opencl/opencl_common.h 417 src/runtime/opencl/opencl_device_api.cc 763 src/runtime/opencl/opencl_module.h 19 src/runtime/opencl/opencl_module.cc 323 src/runtime/opencl/opencl_wrapper/opencl_wrapper.cc 566 src/runtime/ndarray.cc 227 src/runtime/vulkan/vulkan_stream.h 52 src/runtime/vulkan/vulkan_wrapped_func.cc 371 src/runtime/vulkan/vulkan_device.cc 466 src/runtime/vulkan/vulkan_stream.cc 119 src/runtime/vulkan/vulkan_amdrgp.cc 27 src/runtime/vulkan/vulkan_buffer.h 52 src/runtime/vulkan/vulkan_module.cc 44 src/runtime/vulkan/vulkan_common.cc 31 src/runtime/vulkan/vulkan_amdrgp.h 32 src/runtime/vulkan/vulkan_common.h 78 src/runtime/vulkan/vulkan_buffer.cc 95 src/runtime/vulkan/vulkan_wrapped_func.h 74 src/runtime/vulkan/vulkan_instance.h 27 src/runtime/vulkan/vulkan_device_api.cc 378 src/runtime/vulkan/vulkan_device_api.h 52 src/runtime/vulkan/vulkan_device.h 122 src/runtime/vulkan/vulkan_instance.cc 107 src/runtime/vulkan/vulkan_module.h 16 src/runtime/const_loader_module.cc 154 src/runtime/hexagon/hexagon_user_dma_instructions.h 47 src/runtime/hexagon/hexagon_user_dma.h 40 src/runtime/hexagon/hexagon_common.cc 63 src/runtime/hexagon/hexagon_power_manager.h 26 src/runtime/hexagon/hexagon_user_dma_descriptors.h 197 src/runtime/hexagon/hexagon_buffer_manager.h 52 src/runtime/hexagon/hexagon_module.cc 63 src/runtime/hexagon/hexagon_buffer.cc 202 src/runtime/hexagon/hexagon_user_dma.cc 86 src/runtime/hexagon/ring_buffer.h 116 src/runtime/hexagon/qhl/qhl_wrapper.cc 57 src/runtime/hexagon/hexagon_htp.h 24 src/runtime/hexagon/rpc/simulator/rpc_server.cc 253 src/runtime/hexagon/rpc/simulator/session.cc 1169 src/runtime/hexagon/rpc/simulator/hexagon_sim_proto.h 22 src/runtime/hexagon/rpc/android/session.cc 89 src/runtime/hexagon/rpc/hexagon/rpc_server.cc 241 src/runtime/hexagon/hexagon_user_dma_registers.h 165 src/runtime/hexagon/hexagon_device_api.cc 226 src/runtime/hexagon/hexagon_power_manager.cc 71 src/runtime/hexagon/profiler/prof_utils.h 5 src/runtime/hexagon/profiler/prof_utils.cc 48 src/runtime/hexagon/hexagon_htp.cc 47 src/runtime/hexagon/hexagon_hvx.h 25 src/runtime/hexagon/hexagon_buffer.h 67 src/runtime/hexagon/hexagon_common.h 24 src/runtime/hexagon/hexagon_hvx.cc 33 src/runtime/hexagon/hexagon_vtcm_pool.cc 117 src/runtime/hexagon/ops/conv2d.h 185 src/runtime/hexagon/ops/conv2d_quant_hvx.cc 223 src/runtime/hexagon/ops/conv2d_fp16_hvx.cc 270 src/runtime/hexagon/ops/conv_utils.cc 121 src/runtime/hexagon/hexagon_device_api.h 92 src/runtime/hexagon/hexagon_module.h 41 src/runtime/hexagon/hexagon_vtcm_pool.h 48 src/runtime/hexagon/hexagon_thread_manager.cc 252 src/runtime/hexagon/hexagon_thread_manager.h 97 src/runtime/static_library.h 16 src/runtime/thread_map.h 58 src/runtime/thread_storage_scope.h 206 src/runtime/registry.cc 162 src/runtime/rpc/rpc_socket_impl.cc 126 src/runtime/rpc/rpc_channel.cc 24 src/runtime/rpc/rpc_local_session.h 26 src/runtime/rpc/rpc_endpoint.h 67 src/runtime/rpc/rpc_local_session.cc 117 src/runtime/rpc/rpc_channel.h 26 src/runtime/rpc/rpc_module.cc 350 src/runtime/rpc/rpc_endpoint.cc 954 src/runtime/rpc/rpc_channel_logger.h 109 src/runtime/rpc/rpc_pipe_impl.cc 83 src/runtime/rpc/rpc_event_impl.cc 19 src/runtime/rpc/rpc_server_env.cc 27 src/runtime/rpc/rpc_session.cc 85 src/runtime/rpc/rpc_session.h 79 src/runtime/rpc/rpc_device_api.cc 119 src/runtime/rpc/rpc_socket_impl.h 8 src/runtime/object_internal.h 33 src/runtime/system_library.cc 84 src/runtime/static_library.cc 80 src/runtime/file_utils.h 62 src/runtime/packed_func.cc 7 src/runtime/builtin_fp16.cc 25 src/runtime/object.cc 59 src/runtime/contrib/cudnn/cudnn_frontend/attention.cc 85 src/runtime/contrib/cudnn/cudnn_frontend/attention.h 44 src/runtime/contrib/cudnn/cudnn_json_runtime.cc 190 src/runtime/contrib/cudnn/cudnn_utils.cc 197 src/runtime/contrib/cudnn/cudnn_utils.h 91 src/runtime/contrib/cudnn/conv_forward.cc 184 src/runtime/contrib/cudnn/softmax.cc 57 src/runtime/contrib/cudnn/conv_backward.cc 213 src/runtime/contrib/rocblas/rocblas.cc 101 src/runtime/contrib/msc/tensorrt_runtime.cc 275 src/runtime/contrib/random/random.cc 106 src/runtime/contrib/random/mt_random_engine.cc 157 src/runtime/contrib/papi/papi.cc 200 src/runtime/contrib/tensorrt/tensorrt_utils.h 28 src/runtime/contrib/tensorrt/tensorrt_ops.h 71 src/runtime/contrib/tensorrt/tensorrt_calibrator.h 75 src/runtime/contrib/tensorrt/tensorrt_logger.h 44 src/runtime/contrib/tensorrt/tensorrt_ops.cc 1235 src/runtime/contrib/tensorrt/tensorrt_builder.h 57 src/runtime/contrib/tensorrt/tensorrt_builder.cc 222 src/runtime/contrib/tensorrt/tensorrt_runtime.cc 376 src/runtime/contrib/mscclpp/allreduce.cu 164 src/runtime/contrib/arm_compute_lib/acl_allocator.h 45 src/runtime/contrib/arm_compute_lib/acl_runtime.cc 385 src/runtime/contrib/arm_compute_lib/acl_allocator.cc 41 src/runtime/contrib/arm_compute_lib/acl_utils.cc 122 src/runtime/contrib/arm_compute_lib/acl_utils.h 35 src/runtime/contrib/tflite/tflite_runtime.h 30 src/runtime/contrib/tflite/tflite_runtime.cc 149 src/runtime/contrib/nnapi/nnapi_ops.h 115 src/runtime/contrib/nnapi/nnapi_runtime.cc 177 src/runtime/contrib/nnapi/nnapi_builder.h 90 src/runtime/contrib/nnapi/nnapi_builder.cc 208 src/runtime/contrib/nnapi/nnapi_ops.cc 463 src/runtime/contrib/sort/sort.cc 463 src/runtime/contrib/cblas/dnnl_blas.cc 26 src/runtime/contrib/cblas/gemm_common.h 170 src/runtime/contrib/cblas/mkl.cc 158 src/runtime/contrib/cblas/cblas.cc 120 src/runtime/contrib/cublas/cublas_utils.cc 43 src/runtime/contrib/cublas/cublas.cc 463 src/runtime/contrib/cublas/cublas_utils.h 100 src/runtime/contrib/cublas/cublas_json_runtime.cc 109 src/runtime/contrib/cutlass/fp8_gemm.cu 83 src/runtime/contrib/cutlass/blockwise_scaled_gemm_runner.cuh 201 src/runtime/contrib/cutlass/gemm_runner.cuh 136 src/runtime/contrib/cutlass/fp8_group_gemm.cu 72 src/runtime/contrib/cutlass/weight_preprocess.cc 26 src/runtime/contrib/cutlass/group_gemm_runner.cuh 184 src/runtime/contrib/cutlass/fp8_blockwise_scaled_gemm.cu 145 src/runtime/contrib/cutlass/fp16_group_gemm.cu 60 src/runtime/contrib/edgetpu/edgetpu_runtime.h 20 src/runtime/contrib/edgetpu/edgetpu_runtime.cc 37 src/runtime/contrib/json/json_node.h 238 src/runtime/contrib/json/json_runtime.h 201 src/runtime/contrib/thrust/thrust.cu 429 src/runtime/contrib/hipblas/hipblas_json_runtime.cc 101 src/runtime/contrib/hipblas/hipblas_utils.cc 43 src/runtime/contrib/hipblas/hipblas_utils.h 116 src/runtime/contrib/hipblas/hipblas.cc 357 src/runtime/contrib/bnns/bnns_wrp.h 314 src/runtime/contrib/bnns/bnns_json_runtime.cc 415 src/runtime/contrib/mps/gemm.mm 62 src/runtime/contrib/mps/conv.mm 114 src/runtime/contrib/mps/mps_utils.h 27 src/runtime/contrib/mps/mps_utils.mm 60 src/runtime/contrib/amx/amx_config.cc 90 src/runtime/contrib/miopen/miopen_utils.cc 58 src/runtime/contrib/miopen/conv_forward.cc 160 src/runtime/contrib/miopen/miopen_utils.h 51 src/runtime/contrib/miopen/softmax.cc 60 src/runtime/contrib/mrvl/mrvl_sw_runtime_lib.h 16 src/runtime/contrib/mrvl/mrvl_base64.h 46 src/runtime/contrib/mrvl/mrvl_runtime.cc 93 src/runtime/contrib/mrvl/mrvl_sw_runtime_lib.cc 136 src/runtime/contrib/mrvl/mrvl_hw_runtime.cc 360 src/runtime/contrib/nvshmem/kv_transfer.cu 308 src/runtime/contrib/nvshmem/memory_allocator.cc 64 src/runtime/contrib/nvshmem/init.cc 82 src/runtime/contrib/curand/helper_cuda_kernels.h 12 src/runtime/contrib/curand/helper_cuda_kernels.cu 37 src/runtime/contrib/curand/curand.cc 86 src/runtime/contrib/vllm/dtype_float16.h 543 src/runtime/contrib/vllm/attention_utils.cuh 48 src/runtime/contrib/vllm/attention_kernels.cu 692 src/runtime/contrib/vllm/cache_alloc.cc 29 src/runtime/contrib/vllm/cache_kernels.cu 198 src/runtime/contrib/dnnl/dnnl_utils.h 74 src/runtime/contrib/dnnl/dnnl.cc 295 src/runtime/contrib/dnnl/dnnl_utils.cc 45 src/runtime/contrib/dnnl/dnnl_json_runtime.cc 659 src/runtime/contrib/dnnl/dnnl_kernel.h 36 src/runtime/contrib/dnnl/dnnl_tensor_requisite.h 455 src/runtime/contrib/clml/clml_memory_planner.h 14 src/runtime/contrib/clml/clml_runtime.h 199 src/runtime/contrib/clml/clml_utils.cc 121 src/runtime/contrib/clml/clml_runtime.cc 1392 src/runtime/contrib/clml/clml_utils.h 37 src/runtime/contrib/clml/clml_memory_planner.cc 179 src/runtime/contrib/coreml/coreml_runtime.mm 187 src/runtime/contrib/coreml/coreml_runtime.h 45 src/runtime/cpu_device_api.cc 115 src/runtime/profiling.cc 790 src/runtime/logging.cc 309 src/runtime/debug_compile.cc 13 src/runtime/disco/utils.h 20 src/runtime/disco/bcast_session.h 42 src/runtime/disco/distributed/socket_session.cc 260 src/runtime/disco/disco_worker_thread.h 31 src/runtime/disco/bcast_session.cc 87 src/runtime/disco/loader.cc 353 src/runtime/disco/cuda_ipc/cuda_ipc_memory.cc 144 src/runtime/disco/cuda_ipc/custom_allreduce.cc 72 src/runtime/disco/builtin.cc 131 src/runtime/disco/disco_worker.cc 160 src/runtime/disco/nccl/nccl_context.h 120 src/runtime/disco/nccl/nccl.cc 343 src/runtime/disco/process_session.cc 162 src/runtime/disco/protocol.h 205 src/runtime/disco/message_queue.h 89 src/runtime/disco/session.cc 36 src/runtime/disco/threaded_session.cc 157 src/runtime/workspace_pool.cc 120 src/runtime/const_loader_module.h 14 src/runtime/workspace_pool.h 22 src/runtime/spirv/spirv_shader.h 31 src/runtime/c_runtime_api.cc 613 src/runtime/minrpc/minrpc_server.h 646 src/runtime/minrpc/minrpc_logger.h 115 src/runtime/minrpc/minrpc_server_logging.h 121 src/runtime/minrpc/rpc_reference.h 404 src/runtime/minrpc/minrpc_logger.cc 242 src/runtime/minrpc/posix_popen_server/posix_popen_server.cc 36 src/runtime/minrpc/minrpc_interfaces.h 32 src/runtime/container.cc 68 src/runtime/source_utils.h 11 src/runtime/library_module.h 29 src/runtime/file_utils.cc 205 src/runtime/source_utils.cc 25 src/runtime/metal/metal_module.h 17 src/runtime/metal/metal_common.h 124 src/runtime/metal/metal_device_api.mm 325 src/runtime/metal/metal_module.mm 231 src/runtime/dso_library.cc 82 src/runtime/rocm/rocm_device_api.cc 249 src/runtime/rocm/rocm_module.cc 166 src/runtime/rocm/rocm_module.h 17 src/runtime/rocm/rocm_common.h 31 src/runtime/threading_backend.cc 348 src/script/ir_builder/tir/utils.h 79 src/script/ir_builder/tir/frame.cc 183 src/script/ir_builder/tir/ir.cc 682 src/script/ir_builder/base.cc 87 src/script/ir_builder/ir/utils.h 36 src/script/ir_builder/ir/frame.cc 28 src/script/ir_builder/ir/ir.cc 145 src/script/ir_builder/relax/utils.h 91 src/script/ir_builder/relax/distributed.cc 36 src/script/ir_builder/relax/frame.cc 196 src/script/ir_builder/relax/ir.cc 204 src/script/printer/tir/function.cc 193 src/script/printer/tir/utils.h 160 src/script/printer/tir/expr.cc 366 src/script/printer/tir/ir.cc 76 src/script/printer/tir/for_loop.cc 102 src/script/printer/tir/block.cc 195 src/script/printer/tir/buffer.cc 318 src/script/printer/tir/stmt.cc 418 src/script/printer/utils.h 141 src/script/printer/doc.cc 312 src/script/printer/doc_printer/python_doc_printer.cc 582 src/script/printer/doc_printer/base_doc_printer.cc 293 src/script/printer/doc_printer/base_doc_printer.h 73 src/script/printer/ir/utils.h 43 src/script/printer/ir/distributed.cc 21 src/script/printer/ir/ir.cc 138 src/script/printer/ir/misc.cc 53 src/script/printer/ir_docsifier.cc 166 src/script/printer/legacy_repr.cc 827 src/script/printer/relax/function.cc 92 src/script/printer/relax/call.cc 307 src/script/printer/relax/utils.h 114 src/script/printer/relax/distributed.cc 106 src/script/printer/relax/expr.cc 73 src/script/printer/relax/region.cc 74 src/script/printer/relax/type.cc 62 src/script/printer/relax/struct_info.cc 157 src/script/printer/relax/tir.cc 95 src/script/printer/relax/binding.cc 70 src/arith/conjunctive_normal_form.cc 267 src/arith/constraint_extract.cc 42 src/arith/domain_touched.cc 116 src/arith/iter_affine_map.cc 1814 src/arith/scalable_expression.cc 63 src/arith/interval_set.h 42 src/arith/rewrite_simplify.cc 1748 src/arith/ir_mutator_with_analyzer.cc 195 src/arith/int_constraints.cc 233 src/arith/solve_linear_equation.cc 303 src/arith/ir_mutator_with_analyzer.h 46 src/arith/const_int_bound.cc 535 src/arith/ir_visitor_with_analyzer.cc 86 src/arith/scalable_expression.h 20 src/arith/solve_linear_inequality.cc 419 src/arith/narrow_predicate_expression.cc 145 src/arith/unwrap_vector_expr.cc 47 src/arith/canonical_simplify.cc 971 src/arith/presburger_set.cc 233 src/arith/analyzer.cc 281 src/arith/pattern_match.h 513 src/arith/rewrite_simplify.h 129 src/arith/narrow_predicate_expression.h 10 src/arith/transitive_comparison_analyzer.cc 458 src/arith/detect_linear_equation.cc 247 src/arith/int_set.cc 1003 src/arith/modular_set.cc 271 src/arith/constraint_extract.h 12 src/arith/presburger_set.h 74 src/arith/ir_visitor_with_analyzer.h 28 src/arith/product_normal_form.h 47 src/arith/detect_common_subexpr.cc 37 src/arith/const_fold.h 353 src/arith/unwrap_vector_expr.h 10 src/arith/bound_deducer.cc 301 src/arith/conjunctive_normal_form.h 10 src/arith/int_operator.h 90 src/support/utils.h 118 src/support/str_escape.h 63 src/support/scalars.cc 149 src/support/arena.h 46 src/support/errno_handling.h 25 src/support/scalars.h 23 src/support/socket.h 432 src/support/pipe.h 102 src/support/libinfo.cc 286 src/support/hexdump.h 16 src/support/base64.h 209 src/support/ordered_set.h 64 src/support/ring_buffer.h 115 src/support/ffi_testing.cc 199 src/support/nd_int_set.h 79 src/support/process_id.h 35 src/support/parallel_for.cc 107 src/support/generic_arena.h 91 src/support/hexdump.cc 31 src/support/ssize.h 10 src/support/table_printer.h 97 src/support/array.h 181 src/te/operation/create_primfunc.h 21 src/te/operation/create_primfunc.cc 659 src/te/operation/compute_op.cc 171 src/te/operation/placeholder_op.cc 47 src/te/operation/scan_op.cc 97 src/te/operation/extern_op.cc 51 src/te/operation/graph.cc 55 src/te/operation/graph.h 12 src/te/tensor.cc 82 src/meta_schedule/utils.h 423 src/meta_schedule/module_equality.h 34 src/meta_schedule/trace_apply.cc 182 src/meta_schedule/mutator/mutate_compute_location.cc 83 src/meta_schedule/mutator/mutate_tile_size.cc 208 src/meta_schedule/mutator/mutator.cc 76 src/meta_schedule/mutator/mutate_unroll.cc 92 src/meta_schedule/mutator/mutate_thread_binding.cc 107 src/meta_schedule/mutator/mutate_parallel.cc 217 src/meta_schedule/space_generator/space_generator_union.cc 52 src/meta_schedule/space_generator/post_order_apply.cc 81 src/meta_schedule/space_generator/space_generator.cc 170 src/meta_schedule/space_generator/schedule_fn.cc 72 src/meta_schedule/measure_callback/remove_build_artifact.cc 29 src/meta_schedule/measure_callback/update_cost_model.cc 46 src/meta_schedule/measure_callback/measure_callback.cc 44 src/meta_schedule/measure_callback/add_to_database.cc 48 src/meta_schedule/profiler.cc 102 src/meta_schedule/schedule_rule/add_rfactor.cc 73 src/meta_schedule/schedule_rule/multi_level_tiling_with_intrin.cc 71 src/meta_schedule/schedule_rule/random_compute_location.cc 72 src/meta_schedule/schedule_rule/multi_level_tiling.h 144 src/meta_schedule/schedule_rule/cross_thread_reduction.cc 170 src/meta_schedule/schedule_rule/multi_level_tiling_wide_vector.cc 74 src/meta_schedule/schedule_rule/schedule_rule.cc 369 src/meta_schedule/schedule_rule/multi_level_tiling_tensor_core.cc 636 src/meta_schedule/schedule_rule/apply_custom_rule.cc 61 src/meta_schedule/schedule_rule/multi_level_tiling.cc 334 src/meta_schedule/schedule_rule/parallel_vectorize_unroll.cc 85 src/meta_schedule/schedule_rule/auto_bind.cc 48 src/meta_schedule/schedule_rule/auto_inline.cc 171 src/meta_schedule/search_strategy/search_strategy.cc 79 src/meta_schedule/search_strategy/evolutionary_search.cc 585 src/meta_schedule/search_strategy/replay_func.cc 117 src/meta_schedule/search_strategy/replay_trace.cc 139 src/meta_schedule/task_scheduler/gradient_based.cc 114 src/meta_schedule/task_scheduler/round_robin.cc 42 src/meta_schedule/task_scheduler/task_scheduler.cc 344 src/meta_schedule/database/ordered_union_database.cc 54 src/meta_schedule/database/database.cc 264 src/meta_schedule/database/schedule_fn_database.cc 71 src/meta_schedule/database/json_database.cc 160 src/meta_schedule/database/memory_database.cc 72 src/meta_schedule/database/union_database.cc 56 src/meta_schedule/database/database_utils.cc 345 src/meta_schedule/postproc/disallow_async_strided_mem_copy.cc 137 src/meta_schedule/postproc/verify_gpu_code.cc 179 src/meta_schedule/postproc/rewrite_cooperative_fetch.cc 166 src/meta_schedule/postproc/disallow_dynamic_loop.cc 55 src/meta_schedule/postproc/rewrite_unbound_block.cc 103 src/meta_schedule/postproc/rewrite_reduction_block.cc 125 src/meta_schedule/postproc/rewrite_layout.cc 190 src/meta_schedule/postproc/postproc.cc 94 src/meta_schedule/postproc/rewrite_parallel_vectorize_unroll.cc 378 src/meta_schedule/postproc/verify_vtcm_limit.cc 46 src/meta_schedule/postproc/rewrite_tensorize.cc 80 src/meta_schedule/runner/runner.cc 52 src/meta_schedule/trace_apply.h 14 src/meta_schedule/builder/builder.cc 40 src/meta_schedule/tune_context.cc 55 src/meta_schedule/feature_extractor/feature_extractor.cc 32 src/meta_schedule/feature_extractor/per_store_feature.cc 1140 src/meta_schedule/arg_info.cc 123 src/meta_schedule/cost_model/cost_model.cc 60 src/meta_schedule/module_equality.cc 69 src/meta_schedule/extracted_task.cc 26 src/meta_schedule/schedule/cuda/thread_bind.cc 155 src/meta_schedule/schedule/cuda/winograd.cc 131 src/meta_schedule/schedule/cpu/winograd.cc 71 src/meta_schedule/schedule/generic/winograd.cc 19 src/contrib/msc/framework/tensorflow/tf_v1_opcode.h 34 src/contrib/msc/framework/tensorflow/config.h 26 src/contrib/msc/framework/tensorflow/codegen.cc 131 src/contrib/msc/framework/tensorflow/codegen_utils.h 23 src/contrib/msc/framework/tensorflow/codegen.h 26 src/contrib/msc/framework/tensorflow/tf_v1_opcode.cc 518 src/contrib/msc/framework/tensorrt/codegen.cc 531 src/contrib/msc/framework/tensorrt/codegen_utils.h 91 src/contrib/msc/framework/tensorrt/transform_tensorrt.cc 775 src/contrib/msc/framework/tensorrt/codegen.h 38 src/contrib/msc/framework/tensorrt/tensorrt_opcode.cc 706 src/contrib/msc/framework/tensorrt/tensorrt_opcode.h 50 src/contrib/msc/framework/torch/torch_opcode.cc 729 src/contrib/msc/framework/torch/config.h 26 src/contrib/msc/framework/torch/codegen.cc 131 src/contrib/msc/framework/torch/codegen_utils.h 33 src/contrib/msc/framework/torch/torch_opcode.h 59 src/contrib/msc/framework/torch/codegen.h 27 src/contrib/msc/framework/tvm/relax_opcode.h 30 src/contrib/msc/framework/tvm/config.h 29 src/contrib/msc/framework/tvm/codegen.cc 188 src/contrib/msc/framework/tvm/codegen_utils.h 31 src/contrib/msc/framework/tvm/codegen.h 26 src/contrib/msc/framework/tvm/relax_opcode.cc 737 src/contrib/msc/core/utils.h 172 src/contrib/msc/core/codegen/codegen_json.cc 37 src/contrib/msc/core/codegen/codegen_utils.cc 63 src/contrib/msc/core/codegen/code_stack.cc 386 src/contrib/msc/core/codegen/cpp_codegen.h 144 src/contrib/msc/core/codegen/codegen_utils.h 146 src/contrib/msc/core/codegen/codegen_json.h 65 src/contrib/msc/core/codegen/py_codegen.h 168 src/contrib/msc/core/codegen/base_codegen.h 165 src/contrib/msc/core/codegen/code_stack.h 446 src/contrib/msc/core/transform/fuse_tuple.cc 191 src/contrib/msc/core/transform/rewrite_utils.cc 30 src/contrib/msc/core/transform/rewrite_utils.h 25 src/contrib/msc/core/transform/layout_utils.h 41 src/contrib/msc/core/transform/layout_utils.cc 198 src/contrib/msc/core/transform/inline_params.cc 149 src/contrib/msc/core/transform/bind_named_params.cc 108 src/contrib/msc/core/transform/set_byoc_attrs.cc 68 src/contrib/msc/core/transform/bind_shape.cc 99 src/contrib/msc/core/transform/set_expr_name.cc 272 src/contrib/msc/core/transform/set_expr_layout.cc 1250 src/contrib/msc/core/utils.cc 488 src/contrib/msc/core/ir/graph.cc 1467 src/contrib/msc/core/ir/graph_builder.cc 760 src/contrib/msc/core/ir/plugin.h 412 src/contrib/msc/core/ir/plugin.cc 268 src/contrib/msc/core/ir/graph.h 637 src/contrib/msc/core/ir/graph_builder.h 233 src/contrib/msc/core/printer/msc_doc.h 145 src/contrib/msc/core/printer/prototxt_printer.h 26 src/contrib/msc/core/printer/msc_base_printer.cc 138 src/contrib/msc/core/printer/print_utils.h 143 src/contrib/msc/core/printer/print_utils.cc 64 src/contrib/msc/core/printer/cpp_printer.cc 314 src/contrib/msc/core/printer/prototxt_printer.cc 79 src/contrib/msc/core/printer/msc_doc.cc 58 src/contrib/msc/core/printer/cpp_printer.h 64 src/contrib/msc/core/printer/python_printer.cc 218 src/contrib/msc/core/printer/msc_base_printer.h 123 src/contrib/msc/core/printer/python_printer.h 34 src/contrib/msc/plugin/tensorrt_codegen.cc 806 src/contrib/msc/plugin/tvm_codegen.h 64 src/contrib/msc/plugin/torch_codegen.cc 450 src/contrib/msc/plugin/tensorrt_codegen.h 67 src/contrib/msc/plugin/codegen_utils.h 47 src/contrib/msc/plugin/torch_codegen.h 73 src/contrib/msc/plugin/tvm_codegen.cc 361 src/contrib/msc/plugin/base_codegen.h 551 src/contrib/rust_extension.cc 6 src/ir/function.cc 47 src/ir/expr.cc 117 src/ir/replace_global_vars.cc 60 src/ir/global_info.cc 19 src/ir/env_func.cc 36 src/ir/apply_pass_to_function.cc 89 src/ir/analysis.cc 20 src/ir/type.cc 47 src/ir/module.cc 263 src/ir/type_functor.cc 49 src/ir/transform.cc 512 src/ir/op.cc 112 src/ir/source_map.cc 160 src/ir/attr_functor.h 97 src/ir/global_var_supply.cc 72 src/ir/instrument.cc 196 src/ir/diagnostic.cc 220 src/ir/name_supply.cc 65 src/ir/attrs.cc 46 src/node/ndarray_hash_equal.h 16 src/node/attr_registry.h 85 src/node/structural_hash.cc 566 src/node/script_printer.cc 106 src/node/repr_printer.cc 106 src/node/serialization.cc 591 src/node/structural_equal.cc 473 src/node/container_printing.cc 39 src/node/reflection.cc 270 src/node/object_path.cc 182 src/relax/analysis/analysis.cc 145 src/relax/analysis/udchain.cc 80 src/relax/analysis/struct_info_analysis.cc 1015 src/relax/analysis/var2value.cc 56 src/relax/analysis/collect_call_map.cc 23 src/relax/analysis/graph_partitioner.h 110 src/relax/analysis/detect_recursion.cc 264 src/relax/analysis/layout_transformation.cc 409 src/relax/analysis/graph_partitioner.cc 359 src/relax/analysis/shape_analysis.cc 25 src/relax/analysis/computable_at_compile_time.cc 60 src/relax/analysis/tir_op_pattern_kind.cc 384 src/relax/analysis/well_formed.cc 457 src/relax/distributed/global_info.cc 46 src/relax/distributed/transform/utils.h 25 src/relax/distributed/transform/utils.cc 59 src/relax/distributed/transform/propagate_sharding.cc 534 src/relax/distributed/transform/lower_global_view_to_local_view.cc 373 src/relax/distributed/transform/lower_distir.cc 221 src/relax/distributed/transform/legalize_redistribute.cc 83 src/relax/distributed/axis_group_graph.cc 355 src/relax/distributed/struct_info.cc 103 src/relax/training/utils.h 13 src/relax/training/utils.cc 136 src/relax/testing/transform.cc 19 src/relax/transform/utils.h 242 src/relax/transform/expand_tuple_arguments.cc 131 src/relax/transform/canonicalize_bindings.cc 416 src/relax/transform/static_plan_block_memory.cc 599 src/relax/transform/eliminate_common_subexpr.cc 150 src/relax/transform/replace_global_vars.cc 44 src/relax/transform/allocate_workspace.cc 143 src/relax/transform/legalize_ops.cc 229 src/relax/transform/bind_symbolic_vars.cc 130 src/relax/transform/rewrite_cuda_graph.cc 636 src/relax/transform/fuse_tir.cc 944 src/relax/transform/annotate_tir_op_pattern.cc 24 src/relax/transform/remove_unused_parameters.cc 168 src/relax/transform/reorder_take_after_matmul.cc 85 src/relax/transform/bundle_model_params.cc 78 src/relax/transform/alter_op_impl.cc 319 src/relax/transform/gradient_simplifier.cc 114 src/relax/transform/split_layout_rewrite_preproc.cc 266 src/relax/transform/gradient_simplifier.h 9 src/relax/transform/dataflow_inplace.cc 743 src/relax/transform/rewrite_dataflow_reshape.cc 111 src/relax/transform/decompose_ops.cc 181 src/relax/transform/lambda_lift.cc 346 src/relax/transform/lazy_transform_params.cc 222 src/relax/transform/attach_global_symbol.cc 49 src/relax/transform/merge_composite_functions.cc 274 src/relax/transform/fuse_ops.cc 922 src/relax/transform/kill_after_last_use.cc 183 src/relax/transform/infer_layout_utils.cc 122 src/relax/transform/utils.cc 58 src/relax/transform/infer_layout_utils.h 106 src/relax/transform/to_non_dataflow.cc 38 src/relax/transform/update_vdevice.cc 76 src/relax/transform/reorder_permute_dims_after_concat.cc 117 src/relax/transform/combine_parallel_matmul.cc 291 src/relax/transform/bind_params.cc 149 src/relax/transform/remove_unused_outputs.cc 248 src/relax/transform/lift_transform_params.cc 636 src/relax/transform/compute_prim_value.cc 56 src/relax/transform/dead_code_elimination.cc 85 src/relax/transform/update_param_struct_info.cc 70 src/relax/transform/to_mixed_precision.cc 433 src/relax/transform/adjust_matmul_order.cc 150 src/relax/transform/tuning_api/database.cc 261 src/relax/transform/tuning_api/primitives.cc 214 src/relax/transform/attach_attr_layout_free_buffers.cc 71 src/relax/transform/expand_matmul_of_sum.cc 65 src/relax/transform/lower_alloc_tensor.cc 69 src/relax/transform/convert_layout.cc 244 src/relax/transform/infer_amp_utils.h 35 src/relax/transform/fold_constant.cc 199 src/relax/transform/run_codegen.cc 153 src/relax/transform/split_call_tir_by_pattern.cc 644 src/relax/transform/infer_amp_utils.cc 38 src/relax/transform/convert_dataflow.cc 83 src/relax/transform/meta_schedule.cc 148 src/relax/transform/few_shot_tuning.cc 144 src/relax/transform/gradient.cc 498 src/relax/transform/normalize.cc 218 src/relax/transform/realize_vdevice.cc 277 src/relax/transform/call_tir_rewrite.cc 134 src/relax/transform/inline_functions.cc 137 src/relax/transform/remove_purity_checking.cc 60 src/relax/transform/topological_sort.cc 275 src/relax/utils.cc 175 src/relax/ir/binding_rewrite.cc 248 src/relax/ir/expr.cc 587 src/relax/ir/tir_pattern.cc 14 src/relax/ir/emit_te.cc 43 src/relax/ir/py_expr_functor.cc 444 src/relax/ir/type.cc 50 src/relax/ir/struct_info_functor.cc 105 src/relax/ir/dataflow_matcher.cc 472 src/relax/ir/dataflow_matcher.h 57 src/relax/ir/dataflow_pattern.cc 566 src/relax/ir/transform.cc 252 src/relax/ir/emit_te.h 25 src/relax/ir/dataflow_expr_rewriter.cc 802 src/relax/ir/dataflow_pattern_functor.cc 70 src/relax/ir/expr_functor.cc 628 src/relax/ir/dataflow_block_rewriter.cc 331 src/relax/ir/struct_info.cc 170 src/relax/ir/dataflow_rewriter.h 120 src/relax/ir/block_builder.cc 739 src/relax/backend/task_extraction.cc 90 src/relax/backend/pattern_registry.cc 49 src/relax/backend/pattern_registry.h 19 src/relax/backend/contrib/cudnn/codegen.cc 109 src/relax/backend/contrib/utils.h 53 src/relax/backend/contrib/tensorrt/codegen.cc 168 src/relax/backend/contrib/nnapi/codegen.cc 202 src/relax/backend/contrib/codegen_json/codegen_json.h 325 src/relax/backend/contrib/cublas/codegen.cc 87 src/relax/backend/contrib/cutlass/codegen.cc 293 src/relax/backend/contrib/utils.cc 46 src/relax/backend/contrib/hipblas/codegen.cc 65 src/relax/backend/contrib/dnnl/codegen.cc 62 src/relax/backend/contrib/clml/codegen.cc 228 src/relax/backend/contrib/codegen_c/codegen_c.h 244 src/relax/backend/vm/codegen_vm_tir.cc 395 src/relax/backend/vm/lower_runtime_builtin.cc 177 src/relax/backend/vm/exec_builder.cc 323 src/relax/backend/vm/codegen_vm.cc 343 src/relax/backend/vm/vm_shape_lower.cc 484 src/relax/op/distributed/binary.cc 40 src/relax/op/distributed/utils.h 18 src/relax/op/distributed/distributed.cc 162 src/relax/op/distributed/distributed.h 15 src/relax/op/distributed/linear_algebra.cc 74 src/relax/op/distributed/nn.h 12 src/relax/op/distributed/op.cc 27 src/relax/op/distributed/unary.cc 41 src/relax/op/distributed/utils.cc 80 src/relax/op/distributed/statistical.cc 52 src/relax/op/distributed/ccl.cc 19 src/relax/op/distributed/unary.h 44 src/relax/op/distributed/linear_algebra.h 11 src/relax/op/distributed/manipulate.cc 102 src/relax/op/distributed/manipulate.h 12 src/relax/op/distributed/binary.h 48 src/relax/op/distributed/nn.cc 30 src/relax/op/distributed/statistical.h 14 src/relax/op/op_common.h 276 src/relax/op/memory/view.cc 267 src/relax/op/memory/view.h 10 src/relax/op/ccl/ccl.cc 104 src/relax/op/ccl/ccl.h 13 src/relax/op/op.cc 996 src/relax/op/image/resize.h 12 src/relax/op/image/resize.cc 99 src/relax/op/op_common.cc 168 src/relax/op/tensor/binary.cc 156 src/relax/op/tensor/set.cc 114 src/relax/op/tensor/inspect.cc 326 src/relax/op/tensor/qdq.cc 140 src/relax/op/tensor/datatype.h 11 src/relax/op/tensor/linear_algebra.cc 193 src/relax/op/tensor/index.cc 381 src/relax/op/tensor/set.h 11 src/relax/op/tensor/create.h 23 src/relax/op/tensor/unary.cc 59 src/relax/op/tensor/grad.h 22 src/relax/op/tensor/search.h 12 src/relax/op/tensor/search.cc 153 src/relax/op/tensor/statistical.cc 165 src/relax/op/tensor/index.h 12 src/relax/op/tensor/datatype.cc 50 src/relax/op/tensor/ternary.h 9 src/relax/op/tensor/sorting.cc 106 src/relax/op/tensor/inspect.h 18 src/relax/op/tensor/ternary.cc 112 src/relax/op/tensor/unary.h 49 src/relax/op/tensor/linear_algebra.h 11 src/relax/op/tensor/qdq.h 10 src/relax/op/tensor/sampling.cc 100 src/relax/op/tensor/manipulate.cc 2055 src/relax/op/tensor/manipulate.h 36 src/relax/op/tensor/sampling.h 10 src/relax/op/tensor/sorting.h 14 src/relax/op/tensor/binary.h 52 src/relax/op/tensor/grad.cc 155 src/relax/op/tensor/create.cc 319 src/relax/op/tensor/statistical.h 37 src/relax/op/nn/attention.cc 146 src/relax/op/nn/nn.h 35 src/relax/op/nn/convolution.cc 674 src/relax/op/nn/attention.h 11 src/relax/op/nn/convolution.h 44 src/relax/op/nn/pooling.cc 587 src/relax/op/nn/pooling.h 17 src/relax/op/nn/nn.cc 806 src/topi/einsum.cc 260 src/topi/transform.cc 192 src/topi/broadcast.cc 52 src/topi/vision.cc 12 src/topi/utils.cc 22 src/topi/nn.cc 150 src/topi/elemwise.cc 111 src/topi/reduction.cc 39 python/pyproject.toml 22 python/setup.py 195 python/gen_requirements.py 373 python/tvm/tir/analysis/__init__.py 1 python/tvm/tir/analysis/_ffi_api.py 2 python/tvm/tir/analysis/analysis.py 51 python/tvm/tir/tensor_intrin/dot_product_common.py 45 python/tvm/tir/tensor_intrin/metal.py 253 python/tvm/tir/tensor_intrin/__init__.py 1 python/tvm/tir/tensor_intrin/x86.py 74 python/tvm/tir/tensor_intrin/hexagon.py 170 python/tvm/tir/tensor_intrin/rocm.py 371 python/tvm/tir/tensor_intrin/arm_cpu.py 469 python/tvm/tir/tensor_intrin/cuda.py 1374 python/tvm/tir/pipeline.py 126 python/tvm/tir/op.py 772 python/tvm/tir/buffer.py 122 python/tvm/tir/expr.py 473 python/tvm/tir/__init__.py 86 python/tvm/tir/stmt_functor.py 12 python/tvm/tir/transform/transform.py 172 python/tvm/tir/transform/__init__.py 2 python/tvm/tir/transform/_ffi_api.py 2 python/tvm/tir/transform/function_pass.py 48 python/tvm/tir/ir_builder.py 168 python/tvm/tir/build.py 95 python/tvm/tir/block_scope.py 37 python/tvm/tir/stmt.py 369 python/tvm/tir/generic.py 14 python/tvm/tir/data_layout.py 38 python/tvm/tir/_ffi_api.py 2 python/tvm/tir/function.py 178 python/tvm/tir/block_dependence_info.py 25 python/tvm/tir/schedule/_type_checker.py 253 python/tvm/tir/schedule/transform.py 7 python/tvm/tir/schedule/__init__.py 7 python/tvm/tir/schedule/schedule.py 696 python/tvm/tir/schedule/_ffi_api.py 2 python/tvm/tir/schedule/trace.py 101 python/tvm/tir/schedule/instruction.py 40 python/tvm/tir/schedule/state.py 81 python/tvm/tir/schedule/analysis.py 37 python/tvm/driver/__init__.py 1 python/tvm/driver/build_module.py 39 python/tvm/driver/_ffi_api.py 2 python/tvm/runtime/device.py 72 python/tvm/runtime/__init__.py 24 python/tvm/runtime/object_path.py 79 python/tvm/runtime/support.py 6 python/tvm/runtime/module.py 279 python/tvm/runtime/params.py 19 python/tvm/runtime/ndarray.py 187 python/tvm/runtime/container.py 2 python/tvm/runtime/_ffi_api.py 2 python/tvm/runtime/executable.py 48 python/tvm/runtime/profiling/__init__.py 59 python/tvm/runtime/profiling/_ffi_api.py 2 python/tvm/runtime/object.py 14 python/tvm/runtime/disco/__init__.py 9 python/tvm/runtime/disco/session.py 230 python/tvm/runtime/disco/_ffi_api.py 2 python/tvm/runtime/disco/process_pool.py 105 python/tvm/runtime/_ffi_node_api.py 13 python/tvm/runtime/script_printer.py 240 python/tvm/runtime/object_generic.py 25 python/tvm/runtime/relax_vm.py 181 python/tvm/runtime/packed_func.py 2 python/tvm/error.py 21 python/tvm/__init__.py 46 python/tvm/testing/tir.py 175 python/tvm/testing/usmp.py 12 python/tvm/testing/__init__.py 26 python/tvm/testing/plugin.py 207 python/tvm/testing/_ffi_api.py 2 python/tvm/testing/popen_pool.py 36 python/tvm/testing/runner.py 121 python/tvm/testing/utils.py 949 python/tvm/script/relax.py 1 python/tvm/script/tir.py 1 python/tvm/script/__init__.py 2 python/tvm/script/ir_builder/base.py 45 python/tvm/script/ir_builder/tir/frame.py 67 python/tvm/script/ir_builder/tir/__init__.py 3 python/tvm/script/ir_builder/tir/triton.py 72 python/tvm/script/ir_builder/tir/external_kernel.py 106 python/tvm/script/ir_builder/tir/_ffi_api.py 2 python/tvm/script/ir_builder/tir/ir.py 1159 python/tvm/script/ir_builder/__init__.py 1 python/tvm/script/ir_builder/_ffi_api.py 2 python/tvm/script/ir_builder/ir/frame.py 5 python/tvm/script/ir_builder/ir/__init__.py 13 python/tvm/script/ir_builder/ir/_ffi_api.py 2 python/tvm/script/ir_builder/ir/ir.py 38 python/tvm/script/ir_builder/relax/distributed/__init__.py 1 python/tvm/script/ir_builder/relax/distributed/_ffi_api.py 4 python/tvm/script/ir_builder/relax/distributed/ir.py 82 python/tvm/script/ir_builder/relax/frame.py 20 python/tvm/script/ir_builder/relax/__init__.py 2 python/tvm/script/ir_builder/relax/_ffi_api.py 2 python/tvm/script/ir_builder/relax/ir.py 488 python/tvm/script/highlight.py 176 python/tvm/script/parser/tir/__init__.py 11 python/tvm/script/parser/tir/operation.py 98 python/tvm/script/parser/tir/parser.py 307 python/tvm/script/parser/tir/entry.py 89 python/tvm/script/parser/__init__.py 3 python/tvm/script/parser/_core.py 4 python/tvm/script/parser/core/doc_core.py 800 python/tvm/script/parser/core/evaluator.py 299 python/tvm/script/parser/core/diagnostics.py 138 python/tvm/script/parser/core/error.py 5 python/tvm/script/parser/core/__init__.py 1 python/tvm/script/parser/core/doc.py 294 python/tvm/script/parser/core/parser.py 312 python/tvm/script/parser/core/dispatch.py 29 python/tvm/script/parser/core/entry.py 72 python/tvm/script/parser/core/utils.py 51 python/tvm/script/parser/ir/__init__.py 13 python/tvm/script/parser/ir/parser.py 49 python/tvm/script/parser/ir/entry.py 17 python/tvm/script/parser/relax/dist.py 73 python/tvm/script/parser/relax/__init__.py 23 python/tvm/script/parser/relax/parser.py 327 python/tvm/script/parser/relax/entry.py 306 python/tvm/script/_ffi_api.py 2 python/tvm/script/printer/__init__.py 1 python/tvm/script/printer/doc.py 289 python/tvm/script/printer/doc_printer.py 19 python/tvm/script/printer/_ffi_api.py 2 python/tvm/support.py 46 python/tvm/arith/iter_affine_map.py 85 python/tvm/arith/__init__.py 21 python/tvm/arith/pattern.py 9 python/tvm/arith/int_set.py 34 python/tvm/arith/bound.py 3 python/tvm/arith/_ffi_api.py 2 python/tvm/arith/analyzer.py 91 python/tvm/arith/int_solver.py 35 python/tvm/exec/rpc_tracker.py 17 python/tvm/exec/rpc_proxy.py 78 python/tvm/exec/__init__.py 1 python/tvm/exec/gpu_memory_bandwidth.py 204 python/tvm/exec/query_rpc_tracker.py 20 python/tvm/exec/rpc_server.py 71 python/tvm/exec/disco_remote_socket_session.py 12 python/tvm/exec/popen_worker.py 66 python/tvm/exec/disco_worker.py 62 python/tvm/generic.py 1 python/tvm/rpc/base.py 71 python/tvm/rpc/client.py 218 python/tvm/rpc/server_ios_launcher.py 274 python/tvm/rpc/tracker.py 297 python/tvm/rpc/__init__.py 4 python/tvm/rpc/tornado_util.py 80 python/tvm/rpc/minrpc.py 26 python/tvm/rpc/proxy.py 534 python/tvm/rpc/_ffi_api.py 2 python/tvm/rpc/server.py 353 python/tvm/te/__init__.py 16 python/tvm/te/operation.py 209 python/tvm/te/tensor.py 93 python/tvm/te/tag.py 32 python/tvm/te/_ffi_api.py 2 python/tvm/dlight/analysis/__init__.py 14 python/tvm/dlight/analysis/common_analysis.py 182 python/tvm/dlight/analysis/gemv.py 99 python/tvm/dlight/__init__.py 13 python/tvm/dlight/cpu/base.py 5 python/tvm/dlight/cpu/__init__.py 1 python/tvm/dlight/cpu/gemv.py 87 python/tvm/dlight/base/transform.py 54 python/tvm/dlight/base/__init__.py 10 python/tvm/dlight/base/schedule_rule.py 37 python/tvm/dlight/base/common_schedules.py 44 python/tvm/dlight/base/utils.py 68 python/tvm/dlight/benchmark/extract.py 171 python/tvm/dlight/benchmark/__init__.py 7 python/tvm/dlight/benchmark/bench.py 177 python/tvm/dlight/benchmark/utils.py 70 python/tvm/dlight/gpu/base.py 5 python/tvm/dlight/gpu/transpose.py 96 python/tvm/dlight/gpu/low_batch_gemv.py 611 python/tvm/dlight/gpu/__init__.py 8 python/tvm/dlight/gpu/reduction.py 244 python/tvm/dlight/gpu/fallback.py 55 python/tvm/dlight/gpu/rmsnorm.py 90 python/tvm/dlight/gpu/matmul.py 918 python/tvm/dlight/gpu/general_reduction.py 110 python/tvm/dlight/gpu/gemv.py 563 python/tvm/meta_schedule/post_optimization/__init__.py 4 python/tvm/meta_schedule/post_optimization/post_opt.py 26 python/tvm/meta_schedule/post_optimization/space.py 177 python/tvm/meta_schedule/post_optimization/utils.py 37 python/tvm/meta_schedule/post_optimization/droplet.py 82 python/tvm/meta_schedule/mutator/mutate_tile_size.py 9 python/tvm/meta_schedule/mutator/mutate_compute_location.py 9 python/tvm/meta_schedule/mutator/__init__.py 6 python/tvm/meta_schedule/mutator/mutator.py 67 python/tvm/meta_schedule/mutator/mutate_thread_binding.py 9 python/tvm/meta_schedule/mutator/mutate_parallel.py 10 python/tvm/meta_schedule/mutator/mutate_unroll.py 9 python/tvm/meta_schedule/space_generator/space_generator_union.py 27 python/tvm/meta_schedule/space_generator/schedule_fn.py 26 python/tvm/meta_schedule/space_generator/__init__.py 5 python/tvm/meta_schedule/space_generator/space_generator.py 143 python/tvm/meta_schedule/space_generator/post_order_apply.py 26 python/tvm/meta_schedule/__init__.py 38 python/tvm/meta_schedule/testing/validate_database.py 437 python/tvm/meta_schedule/testing/dummy_object.py 31 python/tvm/meta_schedule/testing/__init__.py 1 python/tvm/meta_schedule/testing/custom_builder_runner.py 25 python/tvm/meta_schedule/testing/space_generation.py 108 python/tvm/meta_schedule/testing/distributed_measure_candidates.py 151 python/tvm/meta_schedule/testing/tune_te.py 123 python/tvm/meta_schedule/testing/te_workload.py 692 python/tvm/meta_schedule/testing/tune_utils.py 49 python/tvm/meta_schedule/testing/local_rpc.py 37 python/tvm/meta_schedule/extracted_task.py 28 python/tvm/meta_schedule/measure_callback/measure_callback.py 59 python/tvm/meta_schedule/measure_callback/__init__.py 4 python/tvm/meta_schedule/measure_callback/add_to_database.py 9 python/tvm/meta_schedule/measure_callback/update_cost_model.py 9 python/tvm/meta_schedule/measure_callback/remove_build_artifact.py 9 python/tvm/meta_schedule/tune_context.py 157 python/tvm/meta_schedule/relax_integration.py 191 python/tvm/meta_schedule/profiler.py 34 python/tvm/meta_schedule/schedule_rule/__init__.py 15 python/tvm/meta_schedule/schedule_rule/add_rfactor.py 16 python/tvm/meta_schedule/schedule_rule/cross_thread_reduction.py 11 python/tvm/meta_schedule/schedule_rule/auto_inline.py 34 python/tvm/meta_schedule/schedule_rule/apply_custom_rule.py 9 python/tvm/meta_schedule/schedule_rule/schedule_rule.py 63 python/tvm/meta_schedule/schedule_rule/parallel_vectorize_unroll.py 22 python/tvm/meta_schedule/schedule_rule/multi_level_tiling.py 101 python/tvm/meta_schedule/schedule_rule/auto_bind.py 20 python/tvm/meta_schedule/schedule_rule/random_compute_location.py 9 python/tvm/meta_schedule/arg_info.py 38 python/tvm/meta_schedule/search_strategy/search_strategy.py 150 python/tvm/meta_schedule/search_strategy/__init__.py 4 python/tvm/meta_schedule/search_strategy/evolutionary_search.py 35 python/tvm/meta_schedule/search_strategy/replay_func.py 9 python/tvm/meta_schedule/search_strategy/replay_trace.py 11 python/tvm/meta_schedule/task_scheduler/__init__.py 3 python/tvm/meta_schedule/task_scheduler/round_robin.py 12 python/tvm/meta_schedule/task_scheduler/gradient_based.py 21 python/tvm/meta_schedule/task_scheduler/task_scheduler.py 137 python/tvm/meta_schedule/database/memory_database.py 13 python/tvm/meta_schedule/database/__init__.py 6 python/tvm/meta_schedule/database/database.py 226 python/tvm/meta_schedule/database/union_database.py 10 python/tvm/meta_schedule/database/schedule_fn_database.py 17 python/tvm/meta_schedule/database/ordered_union_database.py 10 python/tvm/meta_schedule/database/json_database.py 34 python/tvm/meta_schedule/tune.py 67 python/tvm/meta_schedule/postproc/__init__.py 11 python/tvm/meta_schedule/postproc/rewrite_layout.py 9 python/tvm/meta_schedule/postproc/postproc.py 61 python/tvm/meta_schedule/postproc/rewrite_cooperative_fetch.py 9 python/tvm/meta_schedule/postproc/verify_gpu_code.py 9 python/tvm/meta_schedule/postproc/disallow_dynamic_loop.py 9 python/tvm/meta_schedule/postproc/rewrite_unbound_block.py 10 python/tvm/meta_schedule/postproc/rewrite_parallel_vectorize_unroll.py 9 python/tvm/meta_schedule/postproc/disallow_async_strided_mem_copy.py 9 python/tvm/meta_schedule/postproc/verify_vtcm_limit.py 9 python/tvm/meta_schedule/postproc/rewrite_tensorize.py 10 python/tvm/meta_schedule/postproc/rewrite_reduction_block.py 9 python/tvm/meta_schedule/_ffi_api.py 2 python/tvm/meta_schedule/tir_integration.py 137 python/tvm/meta_schedule/runner/__init__.py 12 python/tvm/meta_schedule/runner/rpc_runner.py 270 python/tvm/meta_schedule/runner/config.py 103 python/tvm/meta_schedule/runner/local_runner.py 206 python/tvm/meta_schedule/runner/runner.py 92 python/tvm/meta_schedule/runner/utils.py 57 python/tvm/meta_schedule/builder/__init__.py 2 python/tvm/meta_schedule/builder/builder.py 65 python/tvm/meta_schedule/builder/local_builder.py 136 python/tvm/meta_schedule/utils.py 161 python/tvm/meta_schedule/trace_apply.py 5 python/tvm/meta_schedule/feature_extractor/__init__.py 3 python/tvm/meta_schedule/feature_extractor/per_store_feature.py 24 python/tvm/meta_schedule/feature_extractor/random_feature_extractor.py 29 python/tvm/meta_schedule/feature_extractor/feature_extractor.py 48 python/tvm/meta_schedule/logging.py 156 python/tvm/meta_schedule/cost_model/cost_model.py 106 python/tvm/meta_schedule/cost_model/metric.py 8 python/tvm/meta_schedule/cost_model/__init__.py 3 python/tvm/meta_schedule/cost_model/random_model.py 46 python/tvm/meta_schedule/cost_model/mlp_model.py 593 python/tvm/meta_schedule/cost_model/xgb_model.py 506 python/tvm/meta_schedule/schedule/cuda/__init__.py 1 python/tvm/meta_schedule/schedule/cuda/layout_transform.py 222 python/tvm/meta_schedule/schedule/__init__.py 1 python/tvm/meta_schedule/schedule/cpu/__init__.py 1 python/tvm/meta_schedule/schedule/generic/__init__.py 1 python/tvm/meta_schedule/schedule/x86/__init__.py 1 python/tvm/contrib/msc/pipeline/manager.py 65 python/tvm/contrib/msc/pipeline/wrapper.py 168 python/tvm/contrib/msc/pipeline/pipeline.py 403 python/tvm/contrib/msc/pipeline/__init__.py 2 python/tvm/contrib/msc/pipeline/dynamic.py 193 python/tvm/contrib/msc/pipeline/utils.py 105 python/tvm/contrib/msc/pipeline/worker.py 427 python/tvm/contrib/msc/framework/tensorflow/codegen/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/codegen/codegen.py 27 python/tvm/contrib/msc/framework/tensorflow/runtime/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/runtime/runner.py 130 python/tvm/contrib/msc/framework/tensorflow/tools/distill/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/tools/distill/distiller.py 16 python/tvm/contrib/msc/framework/tensorflow/tools/__init__.py 4 python/tvm/contrib/msc/framework/tensorflow/tools/prune/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/tools/prune/pruner.py 16 python/tvm/contrib/msc/framework/tensorflow/tools/quantize/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/tools/quantize/quantizer.py 16 python/tvm/contrib/msc/framework/tensorflow/tools/track/tracker.py 16 python/tvm/contrib/msc/framework/tensorflow/tools/track/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/__init__.py 5 python/tvm/contrib/msc/framework/tensorflow/_ffi_api.py 2 python/tvm/contrib/msc/framework/tensorflow/frontend/translate.py 14 python/tvm/contrib/msc/framework/tensorflow/frontend/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/codegen/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/codegen/sources.py 21 python/tvm/contrib/msc/framework/tensorrt/codegen/codegen.py 125 python/tvm/contrib/msc/framework/tensorrt/codegen/utils.py 21 python/tvm/contrib/msc/framework/tensorrt/runtime/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/runtime/runner.py 73 python/tvm/contrib/msc/framework/tensorrt/tools/distill/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/tools/distill/distiller.py 16 python/tvm/contrib/msc/framework/tensorrt/tools/__init__.py 4 python/tvm/contrib/msc/framework/tensorrt/tools/prune/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/tools/prune/pruner.py 16 python/tvm/contrib/msc/framework/tensorrt/tools/quantize/method.py 59 python/tvm/contrib/msc/framework/tensorrt/tools/quantize/__init__.py 2 python/tvm/contrib/msc/framework/tensorrt/tools/quantize/quantizer.py 196 python/tvm/contrib/msc/framework/tensorrt/tools/track/tracker.py 54 python/tvm/contrib/msc/framework/tensorrt/tools/track/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/transform/transform.py 13 python/tvm/contrib/msc/framework/tensorrt/transform/__init__.py 2 python/tvm/contrib/msc/framework/tensorrt/transform/pattern.py 245 python/tvm/contrib/msc/framework/tensorrt/_ffi_api.py 2 python/tvm/contrib/msc/framework/tensorrt/frontend/translate.py 30 python/tvm/contrib/msc/framework/tensorrt/frontend/__init__.py 1 python/tvm/contrib/msc/framework/__init__.py 1 python/tvm/contrib/msc/framework/torch/codegen/__init__.py 1 python/tvm/contrib/msc/framework/torch/codegen/codegen.py 34 python/tvm/contrib/msc/framework/torch/runtime/jit.py 66 python/tvm/contrib/msc/framework/torch/runtime/__init__.py 2 python/tvm/contrib/msc/framework/torch/runtime/runner.py 167 python/tvm/contrib/msc/framework/torch/tools/distill/method.py 46 python/tvm/contrib/msc/framework/torch/tools/distill/__init__.py 2 python/tvm/contrib/msc/framework/torch/tools/distill/distiller.py 61 python/tvm/contrib/msc/framework/torch/tools/__init__.py 4 python/tvm/contrib/msc/framework/torch/tools/prune/__init__.py 1 python/tvm/contrib/msc/framework/torch/tools/prune/pruner.py 16 python/tvm/contrib/msc/framework/torch/tools/quantize/method.py 127 python/tvm/contrib/msc/framework/torch/tools/quantize/__init__.py 2 python/tvm/contrib/msc/framework/torch/tools/quantize/quantizer.py 16 python/tvm/contrib/msc/framework/torch/tools/track/tracker.py 16 python/tvm/contrib/msc/framework/torch/tools/track/__init__.py 1 python/tvm/contrib/msc/framework/torch/__init__.py 1 python/tvm/contrib/msc/framework/torch/_ffi_api.py 2 python/tvm/contrib/msc/framework/torch/frontend/translate.py 40 python/tvm/contrib/msc/framework/torch/frontend/__init__.py 1 python/tvm/contrib/msc/framework/tvm/codegen/__init__.py 1 python/tvm/contrib/msc/framework/tvm/codegen/codegen.py 14 python/tvm/contrib/msc/framework/tvm/runtime/__init__.py 1 python/tvm/contrib/msc/framework/tvm/runtime/runner.py 157 python/tvm/contrib/msc/framework/tvm/tools/distill/__init__.py 1 python/tvm/contrib/msc/framework/tvm/tools/distill/distiller.py 16 python/tvm/contrib/msc/framework/tvm/tools/__init__.py 4 python/tvm/contrib/msc/framework/tvm/tools/prune/__init__.py 1 python/tvm/contrib/msc/framework/tvm/tools/prune/pruner.py 16 python/tvm/contrib/msc/framework/tvm/tools/quantize/method.py 87 python/tvm/contrib/msc/framework/tvm/tools/quantize/__init__.py 2 python/tvm/contrib/msc/framework/tvm/tools/quantize/quantizer.py 67 python/tvm/contrib/msc/framework/tvm/tools/track/tracker.py 58 python/tvm/contrib/msc/framework/tvm/tools/track/__init__.py 1 python/tvm/contrib/msc/framework/tvm/__init__.py 1 python/tvm/contrib/msc/framework/tvm/_ffi_api.py 2 python/tvm/contrib/msc/__init__.py 1 python/tvm/contrib/msc/core/codegen/__init__.py 2 python/tvm/contrib/msc/core/codegen/sources.py 7 python/tvm/contrib/msc/core/codegen/codegen.py 117 python/tvm/contrib/msc/core/utils/expr.py 82 python/tvm/contrib/msc/core/utils/file.py 206 python/tvm/contrib/msc/core/utils/__init__.py 9 python/tvm/contrib/msc/core/utils/log.py 76 python/tvm/contrib/msc/core/utils/info.py 210 python/tvm/contrib/msc/core/utils/arguments.py 120 python/tvm/contrib/msc/core/utils/namespace.py 42 python/tvm/contrib/msc/core/utils/message.py 91 python/tvm/contrib/msc/core/utils/register.py 130 python/tvm/contrib/msc/core/utils/dataset.py 274 python/tvm/contrib/msc/core/runtime/jit.py 129 python/tvm/contrib/msc/core/runtime/hook.py 52 python/tvm/contrib/msc/core/runtime/__init__.py 2 python/tvm/contrib/msc/core/runtime/runner.py 614 python/tvm/contrib/msc/core/tools/distill/method.py 28 python/tvm/contrib/msc/core/tools/distill/__init__.py 3 python/tvm/contrib/msc/core/tools/distill/configer.py 22 python/tvm/contrib/msc/core/tools/distill/distiller.py 95 python/tvm/contrib/msc/core/tools/__init__.py 6 python/tvm/contrib/msc/core/tools/prune/method.py 46 python/tvm/contrib/msc/core/tools/prune/__init__.py 3 python/tvm/contrib/msc/core/tools/prune/configer.py 44 python/tvm/contrib/msc/core/tools/prune/pruner.py 307 python/tvm/contrib/msc/core/tools/tool.py 612 python/tvm/contrib/msc/core/tools/configer.py 26 python/tvm/contrib/msc/core/tools/execute.py 129 python/tvm/contrib/msc/core/tools/quantize/method.py 202 python/tvm/contrib/msc/core/tools/quantize/__init__.py 3 python/tvm/contrib/msc/core/tools/quantize/configer.py 74 python/tvm/contrib/msc/core/tools/quantize/quantizer.py 107 python/tvm/contrib/msc/core/tools/track/method.py 48 python/tvm/contrib/msc/core/tools/track/tracker.py 85 python/tvm/contrib/msc/core/tools/track/__init__.py 3 python/tvm/contrib/msc/core/tools/track/configer.py 31 python/tvm/contrib/msc/core/__init__.py 1 python/tvm/contrib/msc/core/transform/transform.py 25 python/tvm/contrib/msc/core/transform/__init__.py 2 python/tvm/contrib/msc/core/transform/pattern.py 317 python/tvm/contrib/msc/core/_ffi_api.py 2 python/tvm/contrib/msc/core/ir/graph.py 329 python/tvm/contrib/msc/core/ir/__init__.py 1 python/tvm/contrib/msc/core/gym/__init__.py 3 python/tvm/contrib/msc/core/gym/agent/method.py 21 python/tvm/contrib/msc/core/gym/agent/__init__.py 2 python/tvm/contrib/msc/core/gym/agent/search_agent.py 65 python/tvm/contrib/msc/core/gym/agent/base_agent.py 111 python/tvm/contrib/msc/core/gym/namespace.py 15 python/tvm/contrib/msc/core/gym/control/controller.py 43 python/tvm/contrib/msc/core/gym/control/service.py 357 python/tvm/contrib/msc/core/gym/control/__init__.py 2 python/tvm/contrib/msc/core/gym/control/configer.py 34 python/tvm/contrib/msc/core/gym/control/namespace.py 15 python/tvm/contrib/msc/core/gym/control/worker.py 98 python/tvm/contrib/msc/core/gym/environment/method.py 70 python/tvm/contrib/msc/core/gym/environment/quantize_env.py 25 python/tvm/contrib/msc/core/gym/environment/__init__.py 3 python/tvm/contrib/msc/core/gym/environment/base_env.py 159 python/tvm/contrib/msc/core/gym/environment/prune_env.py 28 python/tvm/contrib/msc/core/frontend/translate.py 153 python/tvm/contrib/msc/core/frontend/__init__.py 1 python/tvm/contrib/msc/plugin/codegen/__init__.py 1 python/tvm/contrib/msc/plugin/codegen/sources.py 21 python/tvm/contrib/msc/plugin/codegen/codegen.py 177 python/tvm/contrib/msc/plugin/__init__.py 1 python/tvm/contrib/msc/plugin/build.py 123 python/tvm/contrib/msc/plugin/register.py 42 python/tvm/contrib/msc/plugin/_ffi_api.py 2 python/tvm/contrib/msc/plugin/utils.py 43 python/tvm/contrib/msc/plugin/op/__init__.py 1 python/tvm/contrib/msc/plugin/op/_ffi_api.py 2 python/tvm/contrib/dnnl.py 79 python/tvm/contrib/cudnn.py 494 python/tvm/contrib/dlpack.py 14 python/tvm/contrib/spirv.py 20 python/tvm/contrib/clang.py 44 python/tvm/contrib/__init__.py 1 python/tvm/contrib/nvcc.py 230 python/tvm/contrib/cublas.py 29 python/tvm/contrib/rocblas.py 26 python/tvm/contrib/xcode.py 87 python/tvm/contrib/cutlass/attention_operation.py 29 python/tvm/contrib/cutlass/conv2d_operation.py 309 python/tvm/contrib/cutlass/rms_norm_operation.py 4 python/tvm/contrib/cutlass/library.py 208 python/tvm/contrib/cutlass/__init__.py 1 python/tvm/contrib/cutlass/conv2d_profiler.py 16 python/tvm/contrib/cutlass/layer_norm_operation.py 4 python/tvm/contrib/cutlass/build.py 655 python/tvm/contrib/cutlass/gen_gemm.py 272 python/tvm/contrib/cutlass/gen_conv2d.py 315 python/tvm/contrib/cutlass/_ffi_api.py 2 python/tvm/contrib/cutlass/gemm_operation.py 254 python/tvm/contrib/cutlass/gemm_profiler.py 15 python/tvm/contrib/cutlass/gen_tensor_op.py 724 python/tvm/contrib/cublaslt.py 17 python/tvm/contrib/mkl.py 46 python/tvm/contrib/hexagon/tools.py 345 python/tvm/contrib/hexagon/__init__.py 1 python/tvm/contrib/hexagon/hexagon_unary_ops.py 61 python/tvm/contrib/hexagon/build.py 504 python/tvm/contrib/hexagon/session.py 139 python/tvm/contrib/hexagon/_ci_env_check.py 17 python/tvm/contrib/hexagon/profiling/process_lwp_data.py 270 python/tvm/contrib/hexagon/meta_schedule.py 121 python/tvm/contrib/hexagon/hexagon_profiler.py 65 python/tvm/contrib/hexagon/generate_take_op.py 65 python/tvm/contrib/hexagon/pytest_plugin.py 248 python/tvm/contrib/rpc.py 7 python/tvm/contrib/thrust.py 19 python/tvm/contrib/rocm.py 139 python/tvm/contrib/emcc.py 46 python/tvm/contrib/nnpack.py 111 python/tvm/contrib/download.py 83 python/tvm/contrib/miopen.py 87 python/tvm/contrib/tvmjs.py 269 python/tvm/contrib/random.py 32 python/tvm/contrib/pickle_memoize.py 75 python/tvm/contrib/popen_pool.py 206 python/tvm/contrib/cc.py 206 python/tvm/contrib/mps.py 31 python/tvm/contrib/hipblas.py 29 python/tvm/contrib/tflite_runtime.py 28 python/tvm/contrib/utils.py 125 python/tvm/contrib/tar.py 53 python/tvm/contrib/mrvl.py 325 python/tvm/contrib/coreml_runtime.py 18 python/tvm/contrib/ndk.py 86 python/tvm/contrib/cblas.py 34 python/tvm/ir/base.py 74 python/tvm/ir/_ffi_transform_api.py 2 python/tvm/ir/op.py 45 python/tvm/ir/transform.py 149 python/tvm/ir/expr.py 75 python/tvm/ir/supply.py 31 python/tvm/ir/__init__.py 28 python/tvm/ir/instrument.py 75 python/tvm/ir/type.py 32 python/tvm/ir/module.py 91 python/tvm/ir/attrs.py 39 python/tvm/ir/container.py 2 python/tvm/ir/diagnostics/__init__.py 38 python/tvm/ir/diagnostics/_ffi_api.py 2 python/tvm/ir/json_compact.py 94 python/tvm/ir/_ffi_analysis_api.py 2 python/tvm/ir/_ffi_api.py 2 python/tvm/ir/function.py 30 python/tvm/ir/analysis.py 7 python/tvm/ir/type_relation.py 11 python/tvm/ir/global_info.py 25 python/tvm/ir/_ffi_instrument_api.py 2 python/tvm/_ffi/base.py 31 python/tvm/_ffi/__init__.py 5 python/tvm/_ffi/_pyversion.py 4 python/tvm/_ffi/registry.py 7 python/tvm/_ffi/libinfo.py 160 python/tvm/ffi/dtype.py 73 python/tvm/ffi/error.py 52 python/tvm/ffi/__init__.py 44 python/tvm/ffi/ndarray.py 57 python/tvm/ffi/container.py 102 python/tvm/ffi/registry.py 59 python/tvm/ffi/_ffi_api.py 2 python/tvm/ffi/convert.py 30 python/tvm/ffi/cython/function.pxi 214 python/tvm/ffi/cython/object.pxi 123 python/tvm/ffi/cython/core.pyx 8 python/tvm/ffi/cython/device.pxi 117 python/tvm/ffi/cython/dtype.pxi 60 python/tvm/ffi/cython/string.pxi 31 python/tvm/ffi/cython/error.pxi 84 python/tvm/ffi/cython/ndarray.pxi 172 python/tvm/ffi/cython/base.pxi 159 python/tvm/relax/dpl/__init__.py 9 python/tvm/relax/dpl/pattern.py 312 python/tvm/relax/dpl/rewrite.py 62 python/tvm/relax/dpl/context.py 21 python/tvm/relax/dpl/_ffi.py 2 python/tvm/relax/analysis/estimate_memory_usage.py 99 python/tvm/relax/analysis/__init__.py 29 python/tvm/relax/analysis/_ffi_api.py 2 python/tvm/relax/analysis/analysis.py 87 python/tvm/relax/distributed/struct_info.py 46 python/tvm/relax/distributed/__init__.py 3 python/tvm/relax/distributed/transform/transform.py 10 python/tvm/relax/distributed/transform/__init__.py 6 python/tvm/relax/distributed/transform/_ffi_api.py 2 python/tvm/relax/distributed/_ffi_api.py 2 python/tvm/relax/distributed/global_info.py 22 python/tvm/relax/struct_info.py 116 python/tvm/relax/pipeline.py 164 python/tvm/relax/training/__init__.py 7 python/tvm/relax/training/trainer.py 183 python/tvm/relax/training/optimizer.py 283 python/tvm/relax/training/_ffi_api.py 2 python/tvm/relax/training/loss.py 150 python/tvm/relax/training/utils.py 35 python/tvm/relax/training/setup_trainer.py 92 python/tvm/relax/expr.py 519 python/tvm/relax/__init__.py 76 python/tvm/relax/testing/lib_comparator.py 58 python/tvm/relax/testing/vm.py 46 python/tvm/relax/testing/transform.py 62 python/tvm/relax/testing/runtime_builtin.py 9 python/tvm/relax/testing/__init__.py 4 python/tvm/relax/testing/ast_printer.py 255 python/tvm/relax/testing/attention.py 111 python/tvm/relax/testing/nn.py 163 python/tvm/relax/testing/matmul.py 40 python/tvm/relax/transform/ipc_allreduce_rewrite.py 90 python/tvm/relax/transform/legalize_ops/datatype.py 11 python/tvm/relax/transform/legalize_ops/unary.py 42 python/tvm/relax/transform/legalize_ops/qdq.py 63 python/tvm/relax/transform/legalize_ops/ccl.py 95 python/tvm/relax/transform/legalize_ops/__init__.py 16 python/tvm/relax/transform/legalize_ops/search.py 17 python/tvm/relax/transform/legalize_ops/inspect_op.py 82 python/tvm/relax/transform/legalize_ops/statistical.py 47 python/tvm/relax/transform/legalize_ops/manipulate.py 242 python/tvm/relax/transform/legalize_ops/nn.py 638 python/tvm/relax/transform/legalize_ops/grad.py 171 python/tvm/relax/transform/legalize_ops/create.py 76 python/tvm/relax/transform/legalize_ops/image.py 19 python/tvm/relax/transform/legalize_ops/linear_algebra.py 84 python/tvm/relax/transform/legalize_ops/distributed.py 23 python/tvm/relax/transform/legalize_ops/common.py 28 python/tvm/relax/transform/legalize_ops/binary.py 37 python/tvm/relax/transform/legalize_ops/index.py 89 python/tvm/relax/transform/transform.py 340 python/tvm/relax/transform/lazy_transform_params.py 260 python/tvm/relax/transform/__init__.py 80 python/tvm/relax/transform/fold_batch_norm_to_conv2d_for_inference.py 54 python/tvm/relax/transform/fuse_transpose_matmul.py 121 python/tvm/relax/transform/optimize_layout_transform.py 36 python/tvm/relax/transform/lower_gpu_ipc_alloc_storage.py 51 python/tvm/relax/transform/attach_external_modules.py 22 python/tvm/relax/transform/tuning_api/primitives.py 142 python/tvm/relax/transform/tuning_api/__init__.py 3 python/tvm/relax/transform/tuning_api/database.py 72 python/tvm/relax/transform/tuning_api/_ffi_api.py 2 python/tvm/relax/transform/tuning_api/default_functions.py 155 python/tvm/relax/transform/_ffi_api.py 2 python/tvm/relax/transform/fast_math.py 27 python/tvm/relax/transform/remove_redundant_reshape.py 32 python/tvm/relax/expr_functor.py 469 python/tvm/relax/vm_build.py 124 python/tvm/relax/_ffi_api.py 2 python/tvm/relax/ir/instrument.py 19 python/tvm/relax/ty.py 19 python/tvm/relax/frontend/__init__.py 2 python/tvm/relax/frontend/torch/__init__.py 3 python/tvm/relax/frontend/torch/base_fx_graph_translator.py 1460 python/tvm/relax/frontend/torch/exported_program_translator.py 536 python/tvm/relax/frontend/torch/fx_translator.py 830 python/tvm/relax/frontend/torch/dynamo.py 112 python/tvm/relax/frontend/stablehlo/__init__.py 1 python/tvm/relax/frontend/stablehlo/stablehlo_translator.py 318 python/tvm/relax/frontend/nn/extern.py 211 python/tvm/relax/frontend/nn/llm/kv_cache.py 2487 python/tvm/relax/frontend/nn/llm/__init__.py 4 python/tvm/relax/frontend/nn/llm/tree_attn.py 1058 python/tvm/relax/frontend/nn/llm/position_embedding.py 383 python/tvm/relax/frontend/nn/spec.py 156 python/tvm/relax/frontend/nn/exporter.py 261 python/tvm/relax/frontend/nn/op.py 780 python/tvm/relax/frontend/nn/__init__.py 23 python/tvm/relax/frontend/nn/modules.py 507 python/tvm/relax/frontend/nn/torch.py 97 python/tvm/relax/frontend/nn/visitor.py 38 python/tvm/relax/frontend/nn/core.py 359 python/tvm/relax/frontend/nn/subroutine.py 113 python/tvm/relax/frontend/nn/_tensor_op.py 62 python/tvm/relax/frontend/common.py 69 python/tvm/relax/frontend/onnx/onnx_frontend.py 2434 python/tvm/relax/frontend/onnx/__init__.py 1 python/tvm/relax/type_converter.py 78 python/tvm/relax/utils.py 206 python/tvm/relax/backend/cuda/cudnn.py 128 python/tvm/relax/backend/cuda/pipeline.py 54 python/tvm/relax/backend/cuda/__init__.py 7 python/tvm/relax/backend/cuda/cublas.py 172 python/tvm/relax/backend/cuda/cutlass.py 419 python/tvm/relax/backend/cuda/flashinfer.py 283 python/tvm/relax/backend/dispatch_sampling.py 59 python/tvm/relax/backend/__init__.py 4 python/tvm/relax/backend/adreno/pipeline.py 41 python/tvm/relax/backend/adreno/__init__.py 7 python/tvm/relax/backend/adreno/clml.py 445 python/tvm/relax/backend/pattern_registry.py 38 python/tvm/relax/backend/contrib/__init__.py 1 python/tvm/relax/backend/contrib/nnapi.py 218 python/tvm/relax/backend/patterns.py 336 python/tvm/relax/backend/_ffi_api.py 2 python/tvm/relax/backend/dispatch_sort_scan.py 192 python/tvm/relax/backend/utils.py 51 python/tvm/relax/backend/gpu_generic/pipeline.py 53 python/tvm/relax/backend/gpu_generic/__init__.py 9 python/tvm/relax/backend/gpu_generic/cumsum.py 125 python/tvm/relax/backend/gpu_generic/sampling.py 243 python/tvm/relax/backend/cpu_generic/pipeline.py 42 python/tvm/relax/backend/cpu_generic/__init__.py 6 python/tvm/relax/backend/metal/__init__.py 1 python/tvm/relax/backend/metal/coreml.py 326 python/tvm/relax/backend/rocm/pipeline.py 53 python/tvm/relax/backend/rocm/__init__.py 6 python/tvm/relax/backend/rocm/hipblas.py 120 python/tvm/relax/block_builder.py 223 python/tvm/relax/exec_builder.py 84 python/tvm/relax/binding_rewrite.py 35 python/tvm/relax/op/base.py 255 python/tvm/relax/op/datatype.py 8 python/tvm/relax/op/_op_gradient.py 738 python/tvm/relax/op/unary.py 68 python/tvm/relax/op/qdq.py 10 python/tvm/relax/op/distributed/__init__.py 6 python/tvm/relax/op/distributed/_ffi_api.py 2 python/tvm/relax/op/distributed/distributed.py 28 python/tvm/relax/op/sorting.py 17 python/tvm/relax/op/ternary.py 4 python/tvm/relax/op/__init__.py 137 python/tvm/relax/op/memory/view.py 22 python/tvm/relax/op/memory/__init__.py 2 python/tvm/relax/op/memory/_ffi_api.py 2 python/tvm/relax/op/memory/memory.py 33 python/tvm/relax/op/ccl/ccl.py 15 python/tvm/relax/op/ccl/__init__.py 1 python/tvm/relax/op/ccl/_ffi_api.py 2 python/tvm/relax/op/search.py 9 python/tvm/relax/op/image/__init__.py 1 python/tvm/relax/op/image/_ffi_api.py 2 python/tvm/relax/op/image/image.py 43 python/tvm/relax/op/statistical.py 50 python/tvm/relax/op/builtin/__init__.py 1 python/tvm/relax/op/builtin/_ffi_api.py 2 python/tvm/relax/op/builtin/builtin.py 25 python/tvm/relax/op/set.py 50 python/tvm/relax/op/manipulate.py 111 python/tvm/relax/op/grad/__init__.py 1 python/tvm/relax/op/grad/_ffi_api.py 2 python/tvm/relax/op/grad/grad.py 70 python/tvm/relax/op/mask.py 5 python/tvm/relax/op/_ffi_api.py 2 python/tvm/relax/op/vm/vm.py 34 python/tvm/relax/op/vm/__init__.py 1 python/tvm/relax/op/vm/_ffi_api.py 2 python/tvm/relax/op/sampling.py 16 python/tvm/relax/op/create.py 77 python/tvm/relax/op/nn/__init__.py 37 python/tvm/relax/op/nn/nn.py 535 python/tvm/relax/op/nn/_ffi_api.py 2 python/tvm/relax/op/linear_algebra.py 19 python/tvm/relax/op/op_attrs.py 72 python/tvm/relax/op/binary.py 52 python/tvm/relax/op/index.py 25 python/tvm/topi/unique.py 20 python/tvm/topi/scatter_elements.py 89 python/tvm/topi/generic_op_impl.py 22 python/tvm/topi/transform.py 210 python/tvm/topi/__init__.py 23 python/tvm/topi/testing/batch_to_space_nd.py 39 python/tvm/topi/testing/correlation_nchw_python.py 43 python/tvm/topi/testing/batch_norm.py 40 python/tvm/topi/testing/rms_norm_python.py 8 python/tvm/topi/testing/reorg_python.py 25 python/tvm/topi/testing/matrix_set_diag.py 33 python/tvm/topi/testing/__init__.py 56 python/tvm/topi/testing/nll_loss.py 16 python/tvm/topi/testing/roi_pool_python.py 39 python/tvm/topi/testing/pool_grad_python.py 44 python/tvm/topi/testing/conv3d_transpose_ncdhw_python.py 58 python/tvm/topi/testing/resize_python.py 193 python/tvm/topi/testing/grid_sample_python.py 319 python/tvm/topi/testing/conv2d_backcward_weight_python.py 60 python/tvm/topi/testing/sequence_mask_python.py 13 python/tvm/topi/testing/conv2d_hwcn_python.py 30 python/tvm/topi/testing/poolnd_python.py 135 python/tvm/topi/testing/dense.py 10 python/tvm/topi/testing/conv3d_ndhwc_python.py 49 python/tvm/topi/testing/crop_and_resize_python.py 87 python/tvm/topi/testing/softmax_python.py 13 python/tvm/topi/testing/space_to_depth.py 11 python/tvm/topi/testing/one_hot.py 26 python/tvm/topi/testing/batch_matmul.py 21 python/tvm/topi/testing/conv2d_transpose_python.py 75 python/tvm/topi/testing/conv2d_nchw_python.py 52 python/tvm/topi/testing/attention_python.py 62 python/tvm/topi/testing/space_to_batch_nd.py 33 python/tvm/topi/testing/strided_slice_python.py 46 python/tvm/topi/testing/layer_norm_python.py 12 python/tvm/topi/testing/l2_normalize_python.py 7 python/tvm/topi/testing/depthwise_conv2d_python.py 62 python/tvm/topi/testing/conv1d_transpose_ncw_python.py 37 python/tvm/topi/testing/adaptive_pool_python.py 84 python/tvm/topi/testing/lrn_python.py 22 python/tvm/topi/testing/conv2d_nhwc_python.py 39 python/tvm/topi/testing/instance_norm_python.py 9 python/tvm/topi/testing/lstm_python.py 67 python/tvm/topi/testing/group_norm_python.py 29 python/tvm/topi/testing/depth_to_space.py 15 python/tvm/topi/testing/slice_axis_python.py 11 python/tvm/topi/testing/conv3d_ncdhw_python.py 45 python/tvm/topi/testing/gather_nd_python.py 18 python/tvm/topi/testing/roi_align_python.py 132 python/tvm/topi/testing/dilate_python.py 16 python/tvm/topi/testing/searchsorted.py 12 python/tvm/topi/testing/common.py 17 python/tvm/topi/testing/conv1d_ncw_python.py 38 python/tvm/topi/testing/deformable_conv2d_python.py 72 python/tvm/topi/testing/gather_python.py 9 python/tvm/topi/signal.py 116 python/tvm/topi/reduction.py 40 python/tvm/topi/tensor.py 16 python/tvm/topi/broadcast.py 58 python/tvm/topi/image/resize.py 750 python/tvm/topi/image/__init__.py 4 python/tvm/topi/image/grid_sample.py 257 python/tvm/topi/image/dilation2d.py 87 python/tvm/topi/einsum.py 3 python/tvm/topi/scan.py 102 python/tvm/topi/sort.py 81 python/tvm/topi/tag.py 13 python/tvm/topi/scatter.py 79 python/tvm/topi/index_put.py 55 python/tvm/topi/math.py 188 python/tvm/topi/gpu/__init__.py 2 python/tvm/topi/gpu/scan.py 366 python/tvm/topi/gpu/sort.py 590 python/tvm/topi/utils.py 179 python/tvm/topi/nn/pixel_shuffle.py 27 python/tvm/topi/nn/bnn.py 40 python/tvm/topi/nn/batch_to_space_nd.py 4 python/tvm/topi/nn/batch_norm.py 61 python/tvm/topi/nn/winograd_util.py 99 python/tvm/topi/nn/dilate.py 29 python/tvm/topi/nn/bitserial_conv2d.py 160 python/tvm/topi/nn/__init__.py 37 python/tvm/topi/nn/instance_norm.py 3 python/tvm/topi/nn/mapping.py 21 python/tvm/topi/nn/conv3d_transpose.py 98 python/tvm/topi/nn/conv1d.py 27 python/tvm/topi/nn/pad.py 141 python/tvm/topi/nn/dense.py 131 python/tvm/topi/nn/deformable_conv2d.py 129 python/tvm/topi/nn/conv2d.py 713 python/tvm/topi/nn/elemwise.py 37 python/tvm/topi/nn/upsampling.py 108 python/tvm/topi/nn/space_to_depth.py 43 python/tvm/topi/nn/layer_norm.py 3 python/tvm/topi/nn/batch_matmul.py 85 python/tvm/topi/nn/lstm.py 151 python/tvm/topi/nn/softmax.py 86 python/tvm/topi/nn/pooling.py 107 python/tvm/topi/nn/qnn.py 88 python/tvm/topi/nn/loss.py 4 python/tvm/topi/nn/conv2d_transpose.py 119 python/tvm/topi/nn/space_to_batch_nd.py 4 python/tvm/topi/nn/bitserial_dense.py 43 python/tvm/topi/nn/bitserial_util.py 54 python/tvm/topi/nn/conv1d_transpose.py 78 python/tvm/topi/nn/rms_norm.py 3 python/tvm/topi/nn/flatten.py 20 python/tvm/topi/nn/correlation.py 57 python/tvm/topi/nn/utils.py 120 python/tvm/topi/nn/conv3d.py 61 python/tvm/topi/nn/group_norm.py 3 python/tvm/topi/nn/depthwise_conv2d.py 256 python/tvm/topi/nn/depth_to_space.py 39 python/tvm/topi/nn/local_response_norm.py 4 python/tvm/topi/nn/fifo_buffer.py 131 python/tvm/topi/cpp/__init__.py 8 python/tvm/topi/cpp/x86.py 2 python/tvm/topi/cpp/generic.py 2 python/tvm/topi/cpp/rocm.py 2 python/tvm/topi/cpp/nn.py 2 python/tvm/topi/cpp/utils.py 2 python/tvm/topi/cpp/cuda.py 2 python/tvm/topi/cpp/impl.py 2 python/tvm/topi/cpp/vision/yolo.py 2 python/tvm/topi/cpp/vision/__init__.py 3 python/tvm/topi/searchsorted.py 53 python/tvm/topi/sparse_reshape.py 106 apps/android_rpc/app/src/main/java/org/apache/tvm/tvmrpc/MainActivity.java 125 apps/android_rpc/app/src/main/java/org/apache/tvm/tvmrpc/RPCProcessor.java 58 apps/android_rpc/app/src/main/java/org/apache/tvm/tvmrpc/RPCActivity.java 41 apps/android_rpc/app/src/main/java/org/apache/tvm/tvmrpc/RPCAndroidWatchdog.java 13 apps/android_rpc/app/src/main/jni/tvm_runtime.h 79 apps/android_rpc/app/src/main/res/values/colors.xml 6 apps/android_rpc/app/src/main/res/values/strings.xml 18 apps/android_rpc/app/src/main/res/values/styles.xml 14 apps/android_rpc/app/src/main/res/layout/content_rpc.xml 15 apps/android_rpc/app/src/main/res/layout/activity_rpc.xml 21 apps/android_rpc/app/src/main/res/layout/content_main.xml 82 apps/android_rpc/app/src/main/res/layout/activity_main.xml 21 apps/ios_rpc/init_proj.py 37 apps/ios_rpc/tvmrpc/RPCArgs.h 22 apps/ios_rpc/tvmrpc/TVMRuntime.mm 71 apps/ios_rpc/tvmrpc/AppDelegate.h 4 apps/ios_rpc/tvmrpc/main.m 9 apps/ios_rpc/tvmrpc/ViewController.mm 104 apps/ios_rpc/tvmrpc/RPCServer.mm 589 apps/ios_rpc/tvmrpc/RPCServer.h 31 apps/ios_rpc/tvmrpc/RPCArgs.mm 147 apps/ios_rpc/tvmrpc/AppDelegate.m 18 apps/ios_rpc/tvmrpc/ViewController.h 12 apps/cpp_rpc/rpc_tracker_client.h 146 apps/cpp_rpc/rpc_server.cc 263 apps/cpp_rpc/rpc_env.h 17 apps/cpp_rpc/rpc_env.cc 244 apps/cpp_rpc/rpc_server.h 15 apps/cpp_rpc/main.cc 203 apps/cpp_rpc/win32_process.h 12 apps/cpp_rpc/win32_process.cc 153 apps/hexagon_launcher/launcher_hexagon.cc 171 apps/hexagon_launcher/launcher_core.h 87 apps/hexagon_launcher/launcher_util.h 11 apps/hexagon_launcher/cmake/HexagonLauncher.cmake 66 apps/hexagon_launcher/launcher_main.cc 120 apps/hexagon_launcher/launcher_core.cc 168 apps/hexagon_launcher/launcher_android.cc 133 apps/hexagon_launcher/launcher_util.cc 39 golang/sample/pack_func_convert.go 29 golang/sample/pack_func_register.go 44 golang/sample/pack_func_handle_arg.go 39 golang/sample/complex.go 122 golang/sample/pack_func_closure_return.go 38 golang/sample/pack_func_closure_arg.go 35 golang/sample/simple.go 37 golang/src/gotvm.go 8 golang/src/function.go 241 golang/src/module.go 71 golang/src/bytearray.go 37 golang/src/value.go 234 golang/src/gotvm.cc 88 golang/src/device.go 39 golang/src/ndarray.go 251 golang/src/type.go 48 golang/src/gotvm.h 20 golang/src/error.go 15 golang/src/tvm_runtime_pack.cc 19 golang/src/utils.go 12 conftest.py 46 pyproject.toml 52 ffi/include/tvm/ffi/function_details.h 135 ffi/include/tvm/ffi/rvalue_ref.h 67 ffi/include/tvm/ffi/base_details.h 179 ffi/include/tvm/ffi/reflection/reflection.h 93 ffi/include/tvm/ffi/memory.h 97 ffi/include/tvm/ffi/c_api.h 190 ffi/include/tvm/ffi/object.h 397 ffi/include/tvm/ffi/type_traits.h 461 ffi/include/tvm/ffi/cast.h 79 ffi/include/tvm/ffi/container/container_details.h 177 ffi/include/tvm/ffi/container/map.h 968 ffi/include/tvm/ffi/container/variant.h 123 ffi/include/tvm/ffi/container/tuple.h 189 ffi/include/tvm/ffi/container/shape.h 105 ffi/include/tvm/ffi/container/ndarray.h 176 ffi/include/tvm/ffi/container/array.h 577 ffi/include/tvm/ffi/error.h 168 ffi/include/tvm/ffi/string.h 328 ffi/include/tvm/ffi/endian.h 50 ffi/include/tvm/ffi/any.h 313 ffi/include/tvm/ffi/function.h 448 ffi/include/tvm/ffi/optional.h 186 ffi/include/tvm/ffi/dtype.h 122 ffi/cmake/Utils/Sanitizer.cmake 34 ffi/cmake/Utils/AddLibbacktrace.cmake 61 ffi/cmake/Utils/AddGoogleTest.cmake 57 ffi/cmake/Utils/Library.cmake 65 ffi/cmake/Utils/CxxWarning.cmake 29 ffi/src/ffi/function.cc 171 ffi/src/ffi/traceback.h 103 ffi/src/ffi/traceback_win.cc 90 ffi/src/ffi/error.cc 46 ffi/src/ffi/ndarray.cc 49 ffi/src/ffi/traceback.cc 119 ffi/src/ffi/object.cc 219 ffi/src/ffi/dtype.cc 269 ffi/src/ffi/container.cc 56 version.py 118 jvm/conf/google_checks.xml 178 jvm/core/src/main/java/org/apache/tvm/NDArrayBase.java 36 jvm/core/src/main/java/org/apache/tvm/NDArray.java 261 jvm/core/src/main/java/org/apache/tvm/TVMValueBytes.java 11 jvm/core/src/main/java/org/apache/tvm/TVMType.java 74 jvm/core/src/main/java/org/apache/tvm/TVMValueString.java 11 jvm/core/src/main/java/org/apache/tvm/Device.java 118 jvm/core/src/main/java/org/apache/tvm/TVMValueLong.java 11 jvm/core/src/main/java/org/apache/tvm/NativeLibraryLoader.java 104 jvm/core/src/main/java/org/apache/tvm/TVMValueDouble.java 11 jvm/core/src/main/java/org/apache/tvm/Function.java 189 jvm/core/src/main/java/org/apache/tvm/TVMValueNull.java 6 jvm/core/src/main/java/org/apache/tvm/TVMValue.java 33 jvm/core/src/main/java/org/apache/tvm/rpc/Utils.java 62 jvm/core/src/main/java/org/apache/tvm/rpc/NativeServerLoop.java 60 jvm/core/src/main/java/org/apache/tvm/rpc/ServerProcessor.java 4 jvm/core/src/main/java/org/apache/tvm/rpc/Client.java 16 jvm/core/src/main/java/org/apache/tvm/rpc/RPCSession.java 120 jvm/core/src/main/java/org/apache/tvm/rpc/TVMRemoteDevice.java 9 jvm/core/src/main/java/org/apache/tvm/rpc/ConnectTrackerServerProcessor.java 191 jvm/core/src/main/java/org/apache/tvm/rpc/SocketChannel.java 41 jvm/core/src/main/java/org/apache/tvm/rpc/ConnectProxyServerProcessor.java 58 jvm/core/src/main/java/org/apache/tvm/rpc/RPCWatchdog.java 50 jvm/core/src/main/java/org/apache/tvm/rpc/Server.java 39 jvm/core/src/main/java/org/apache/tvm/rpc/RPC.java 35 jvm/core/src/main/java/org/apache/tvm/rpc/StandaloneServerProcessor.java 48 jvm/core/src/main/java/org/apache/tvm/API.java 22 jvm/core/src/main/java/org/apache/tvm/Module.java 80 jvm/core/src/main/java/org/apache/tvm/Base.java 115 jvm/core/src/main/java/org/apache/tvm/TVMValueHandle.java 11 jvm/core/src/main/java/org/apache/tvm/ArgTypeCode.java 14 jvm/core/src/main/java/org/apache/tvm/LibInfo.java 31 jvm/core/src/main/java/org/apache/tvm/APIInternal.java 8 jvm/native/src/main/native/org_apache_tvm_native_c_api.cc 361 jvm/native/src/main/native/jni_helper_func.h 187 web/emcc/decorate_as_wasi.py 15 web/emcc/preload.js 13 web/emcc/tvmjs_support.cc 204 web/emcc/wasm_runtime.cc 165 web/emcc/webgpu_runtime.cc 175 web/src/index.ts 19 web/src/rpc_server.ts 371 web/src/memory.ts 266 web/src/compact.ts 21 web/src/environment.ts 92 web/src/types.ts 7 web/src/webgpu.ts 793 web/src/asyncify.ts 110 web/src/artifact_cache.ts 278 web/src/runtime.ts 1891 web/src/ctypes.ts 77 web/src/support.ts 58 web/rollup.config.mjs 43 web/apps/browser/rpc_server.html 111 web/apps/browser/rpc_plugin.html 1 web/apps/node/example.js 13 web/apps/node/wasi_rpc_server.js 10 web/apps/node/wasi_example.js 9 web/run_jest.py 5 conda/build-environment.yaml 18 conda/render_cuda_dockerfiles.py 26 conda/recipe/conda_build_config.yaml 4 conda/recipe/meta.yaml 70 conda/recipe/cross-linux.cmake 32