Open Machine Learning Compiler Framework
10899 matches across 18 categories. Click a row to expand file-level details.
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | conftest.py | 64 | def pytest_collection_modifyitems(config, items): |
| LOW | python/tvm/topi/scatter.py | 28 | def _verify_scatter_nd_inputs(data, indices, updates): |
| LOW | python/tvm/topi/math.py | 686 | def fixed_point_multiply_per_axis( |
| LOW | python/tvm/topi/gpu/scan.py | 221 | def get_reduction_from_exclusive_scan(data, ex_scan_output, binop=tvm.tirx.generic.add): |
| LOW | python/tvm/topi/nn/winograd_util.py | 164 | def winograd_transform_matrices(tile_size, kernel_size, out_dtype): |
| LOW | python/tvm/topi/vision/nms_util.py | 245 | def collect_selected_indices_and_scores( |
| LOW | python/tvm/topi/vision/nms.py | 852 | def _collect_selected_indices_ir( |
| LOW | python/tvm/topi/vision/nms.py | 898 | def _collect_selected_indices_and_scores_ir( |
| LOW | python/tvm/topi/vision/nms.py | 944 | def all_class_non_max_suppression( |
| LOW | python/tvm/topi/image/resize.py | 28 | def can_convert_multiply_to_intdiv(origin_size, scaled_size): |
| LOW | python/tvm/topi/testing/adaptive_pool_python.py | 77 | def adaptive_pool_channel_first(np_data, out_size, pool_op, np_op): |
| LOW | python/tvm/topi/testing/adaptive_pool_python.py | 91 | def adaptive_pool_channel_last(np_data, out_size, pool_op, np_op): |
| LOW | python/tvm/topi/testing/nms_python.py | 37 | def non_max_suppression_python( |
| LOW | python/tvm/ir/instrument.py | 154 | def _wrap_class_pass_instrument(pi_cls): |
| LOW | python/tvm/ir/base.py | 211 | def get_first_structural_mismatch(lhs, rhs, map_free_vars=False, skip_tensor_content=False): |
| LOW | python/tvm/target/detect_target.py | 109 | def detect_target_from_device(dev: str | Device) -> Target: |
| LOW | python/tvm/runtime/params.py | 98 | def load_param_dict_from_file(path): |
| LOW | python/tvm/runtime/device.py | 87 | def max_shared_memory_per_block(self): |
| LOW | python/tvm/runtime/module.py | 111 | def _collect_from_import_tree(self, filter_func): |
| LOW | python/tvm/runtime/module.py | 234 | def get_source_format_from_module(module): |
| LOW | python/tvm/runtime/disco/session.py | 587 | def _create_socket_session_local_workers(num_workers) -> Session: |
| LOW | python/tvm/s_tir/schedule/analysis.py | 69 | def get_tensorize_loop_mapping( |
| LOW | python/tvm/s_tir/schedule/analysis.py | 97 | def get_auto_tensorize_mapping_info( |
| LOW | python/tvm/s_tir/schedule/schedule.py | 79 | def _parse_error_render_level(error_render_level: str) -> int: |
| LOW | python/tvm/s_tir/schedule/schedule.py | 104 | def _get_sblock_default_dtype(block: SBlock) -> str: |
| LOW | python/tvm/s_tir/schedule/schedule.py | 1050 | def matmul_after_reorder_block_iter_var( |
| LOW | python/tvm/s_tir/schedule/schedule.py | 1588 | def before_reindex_cache_read(a: T.handle, b: T.handle) -> None: |
| LOW | python/tvm/s_tir/schedule/schedule.py | 1688 | def before_reindex_cache_write(a: T.handle, b: T.handle) -> None: |
| LOW | python/tvm/s_tir/schedule/schedule.py | 2174 | def before_reverse_compute_at(a: T.handle, c: T.handle) -> None: |
| LOW | python/tvm/s_tir/schedule/schedule.py | 3428 | def two_elementwise_transformed_intermediate_buffer(a: T.handle, c: T.handle) -> None: |
| LOW | python/tvm/s_tir/schedule/schedule.py | 3513 | def before_transform_block_layout( |
| LOW | python/tvm/s_tir/schedule/schedule.py | 3535 | def after_transform_block_layout( |
| LOW | python/tvm/s_tir/schedule/schedule.py | 3599 | def before_set_axis_separator( |
| LOW | python/tvm/s_tir/schedule/schedule.py | 3627 | def after_set_axis_separators( |
| LOW | python/tvm/s_tir/schedule/schedule.py | 3940 | def unsafe_hide_buffer_access( |
| LOW | python/tvm/s_tir/schedule/schedule.py | 4000 | def before_annotate_buffer_access( |
| LOW | python/tvm/s_tir/schedule/schedule.py | 4029 | def after_annotate_buffer_access( |
| LOW | python/tvm/s_tir/schedule/testing.py | 29 | def assert_structural_equal_ignore_global_symbol( |
| LOW | python/tvm/s_tir/analysis/__init__.py | 58 | def get_sblock_read_write_region( |
| LOW | python/tvm/s_tir/analysis/__init__.py | 145 | def calculate_allocated_bytes( |
| LOW | python/tvm/s_tir/analysis/__init__.py | 197 | def get_vtcm_compaction_passes() -> list[tvm.transform.Pass]: |
| LOW | python/tvm/s_tir/meta_schedule/logging.py | 179 | def _batch_parameterize_config( |
| LOW | python/tvm/s_tir/meta_schedule/logging.py | 239 | def get_loggers_from_work_dir( |
| LOW | python/tvm/s_tir/meta_schedule/trace_apply.py | 25 | def schedule_using_anchor_trace(sch: Schedule, anchor_trace: Trace, target: Target) -> None: |
| LOW | python/tvm/s_tir/meta_schedule/relax_integration.py | 105 | def extracted_tasks_to_tune_contexts( |
| LOW | python/tvm/s_tir/meta_schedule/utils.py | 131 | def get_global_func_with_default_on_worker( |
| LOW | python/tvm/s_tir/meta_schedule/utils.py | 168 | def get_global_func_on_rpc_session( |
| LOW | python/tvm/s_tir/meta_schedule/tune_context.py | 248 | def generate_measure_candidates(self) -> list["MeasureCandidate"] | None: |
| LOW | …/s_tir/meta_schedule/schedule/cuda/layout_transform.py | 135 | def pad_dimension_to_at_least_number(loop: LoopRV, requested_size: int): |
| LOW | …/s_tir/meta_schedule/schedule/cuda/layout_transform.py | 140 | def pad_dimension_to_factor_of_tile_size( |
| LOW | …/s_tir/meta_schedule/schedule/cuda/layout_transform.py | 239 | def get_high_level_loop_structure( |
| LOW | …/s_tir/meta_schedule/schedule/cuda/layout_transform.py | 503 | def cuda_layout_transform_schedule_rule( |
| LOW | …s_tir/meta_schedule/space_generator/space_generator.py | 65 | def _initialize_with_tune_context(self, context: "TuneContext") -> None: |
| LOW | …s_tir/meta_schedule/space_generator/space_generator.py | 231 | def _initialize_with_tune_context(self, context: "TuneContext") -> None: |
| LOW | python/tvm/s_tir/meta_schedule/cost_model/xgb_model.py | 684 | def optional_xgboost_callback(cls): |
| LOW | python/tvm/s_tir/meta_schedule/postproc/postproc.py | 44 | def _initialize_with_tune_context(self, context: "TuneContext") -> None: |
| LOW | python/tvm/s_tir/meta_schedule/postproc/postproc.py | 152 | def _initialize_with_tune_context(self, context: "TuneContext") -> None: |
| LOW | python/tvm/s_tir/meta_schedule/testing/dummy_object.py | 56 | def _initialize_with_tune_context(self, context: "TuneContext") -> None: |
| LOW | …n/tvm/s_tir/meta_schedule/testing/validate_database.py | 421 | def f_with_args_alloc_argument_common( |
| LOW | …n/tvm/s_tir/meta_schedule/testing/validate_database.py | 448 | def f_with_args_run_evaluator_common( |
| 3957 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | .asf.yaml | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | .asf.yaml | 41 | |
| LOW | conftest.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | .markdownlint-cli2.yaml | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | version.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | .pre-commit-config.yaml | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | pyproject.toml | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | .yamllint.yaml | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | docker/dev_common.sh | 1 | #!/bin/bash -e |
| LOW | docker/lint.sh | 1 | #!/bin/bash -e |
| LOW | docker/build.sh | 1 | #!/usr/bin/env bash |
| LOW | docker/build.sh | 21 | # Build a docker container and optionally execute command within a it |
| LOW | docker/build.sh | 41 | # DOCKERFILE_PATH: (Optional) Path to the Dockerfile used for docker build. If |
| LOW | docker/bash.sh | 1 | #!/usr/bin/env bash |
| LOW | docker/bash.sh | 21 | # Start a bash, mount REPO_MOUNT_POINT to be current directory. |
| LOW | docker/clear-stale-images.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_core.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_libtorch.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_dgl.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_setup_tz.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_nodejs.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_tflite.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_googletest.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_vulkan.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_onnx.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_dnnl.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_aprofile_aem.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_xgboost.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_java.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_jax.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_sphinx.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_rocm.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_nnef.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_universal.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_cmake.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_tensorflow_aarch64.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_coreml.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_clang_format.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_emscripten.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_opencl.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_python_package.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_python.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_sccache.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_tensorflow.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_llvm.sh | 1 | #!/bin/bash |
| LOW | docker/install/ubuntu_install_cuda_python.sh | 1 | #!/bin/bash |
| LOW | docker/python/bootstrap/generate.sh | 1 | #!/bin/env bash |
| LOW | docker/utils/apt-install-and-clear.sh | 1 | #!/bin/bash -e |
| LOW | docker/utils/download-and-verify.sh | 1 | #!/bin/bash |
| LOW | python/tvm/error.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | python/tvm/__init__.py | 1 | # isort: skip_file |
| LOW | python/tvm/libinfo.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | python/tvm/ffi.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | python/tvm/base.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | python/tvm/base.py | 41 | # |
| LOW | python/tvm/topi/signal.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | python/tvm/topi/slice_scatter.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | python/tvm/topi/reduction.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | python/tvm/topi/sort.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| LOW | python/tvm/topi/einsum.py | 1 | # Licensed to the Apache Software Foundation (ASF) under one |
| 1887 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | python/tvm/__init__.py | 26 | |
| LOW | python/tvm/__init__.py | 26 | |
| LOW | python/tvm/__init__.py | 26 | |
| LOW | python/tvm/__init__.py | 29 | |
| LOW | python/tvm/__init__.py | 29 | |
| LOW | python/tvm/__init__.py | 33 | |
| LOW | python/tvm/__init__.py | 34 | |
| LOW | python/tvm/__init__.py | 34 | |
| LOW | python/tvm/__init__.py | 34 | |
| LOW | python/tvm/__init__.py | 34 | |
| LOW | python/tvm/__init__.py | 34 | |
| LOW | python/tvm/__init__.py | 34 | |
| LOW | python/tvm/__init__.py | 35 | |
| LOW | python/tvm/__init__.py | 35 | |
| LOW | python/tvm/__init__.py | 35 | |
| LOW | python/tvm/__init__.py | 35 | |
| LOW | python/tvm/__init__.py | 36 | |
| LOW | python/tvm/__init__.py | 36 | |
| LOW | python/tvm/__init__.py | 42 | |
| LOW | python/tvm/__init__.py | 43 | |
| LOW | python/tvm/__init__.py | 44 | |
| LOW | python/tvm/__init__.py | 45 | |
| LOW | python/tvm/__init__.py | 49 | |
| LOW | python/tvm/__init__.py | 52 | |
| LOW | python/tvm/__init__.py | 55 | |
| LOW | python/tvm/__init__.py | 58 | |
| LOW | python/tvm/__init__.py | 61 | |
| LOW | python/tvm/__init__.py | 61 | |
| LOW | python/tvm/__init__.py | 64 | |
| LOW | python/tvm/__init__.py | 67 | |
| LOW | python/tvm/__init__.py | 70 | |
| LOW | python/tvm/__init__.py | 70 | |
| LOW | python/tvm/__init__.py | 78 | |
| LOW | python/tvm/__init__.py | 81 | |
| LOW | python/tvm/libinfo.py | 19 | |
| LOW | python/tvm/ffi.py | 21 | |
| LOW | python/tvm/base.py | 80 | |
| LOW | python/tvm/topi/__init__.py | 29 | |
| LOW | python/tvm/topi/__init__.py | 33 | |
| LOW | python/tvm/topi/__init__.py | 35 | |
| LOW | python/tvm/topi/__init__.py | 36 | |
| LOW | python/tvm/topi/__init__.py | 37 | |
| LOW | python/tvm/topi/__init__.py | 38 | |
| LOW | python/tvm/topi/__init__.py | 39 | |
| LOW | python/tvm/topi/__init__.py | 40 | |
| LOW | python/tvm/topi/__init__.py | 41 | |
| LOW | python/tvm/topi/__init__.py | 42 | |
| LOW | python/tvm/topi/__init__.py | 43 | |
| LOW | python/tvm/topi/__init__.py | 44 | |
| LOW | python/tvm/topi/__init__.py | 45 | |
| LOW | python/tvm/topi/__init__.py | 46 | |
| LOW | python/tvm/topi/__init__.py | 47 | |
| LOW | python/tvm/topi/__init__.py | 48 | |
| LOW | python/tvm/topi/__init__.py | 49 | |
| LOW | python/tvm/topi/__init__.py | 50 | |
| LOW | python/tvm/topi/__init__.py | 51 | |
| LOW | python/tvm/topi/__init__.py | 52 | |
| LOW | python/tvm/topi/__init__.py | 53 | |
| LOW | python/tvm/topi/__init__.py | 54 | |
| LOW | python/tvm/topi/__init__.py | 55 | |
| 2149 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | version.py | 144 | |
| LOW | python/tvm/libinfo.py | 249 | |
| LOW | python/tvm/topi/signal.py | 27 | |
| LOW | python/tvm/topi/signal.py | 131 | |
| LOW | python/tvm/topi/signal.py | 71 | |
| LOW | python/tvm/topi/signal.py | 160 | |
| LOW | python/tvm/topi/unique.py | 25 | |
| LOW | python/tvm/topi/sparse_reshape.py | 26 | |
| LOW | python/tvm/topi/sparse_reshape.py | 74 | |
| LOW | python/tvm/topi/utils.py | 176 | |
| LOW | python/tvm/topi/scatter.py | 57 | |
| LOW | python/tvm/topi/scatter.py | 98 | |
| LOW | python/tvm/topi/scan.py | 32 | |
| LOW | python/tvm/topi/scan.py | 117 | |
| LOW | python/tvm/topi/scatter_elements.py | 27 | |
| LOW | python/tvm/topi/scatter_elements.py | 99 | |
| LOW | python/tvm/topi/index_put.py | 27 | |
| LOW | python/tvm/topi/index_put.py | 91 | |
| LOW | python/tvm/topi/gpu/sort.py | 39 | |
| LOW | python/tvm/topi/gpu/sort.py | 85 | |
| LOW | python/tvm/topi/gpu/sort.py | 183 | |
| LOW | python/tvm/topi/gpu/sort.py | 1084 | |
| LOW | python/tvm/topi/gpu/sort.py | 257 | |
| LOW | python/tvm/topi/gpu/sort.py | 1122 | |
| LOW | python/tvm/topi/gpu/scan.py | 55 | |
| LOW | python/tvm/topi/gpu/scan.py | 221 | |
| LOW | python/tvm/topi/gpu/scan.py | 248 | |
| LOW | python/tvm/topi/gpu/scatter_elements.py | 30 | |
| LOW | python/tvm/topi/gpu/scatter_elements.py | 62 | |
| LOW | python/tvm/topi/gpu/scatter_nd.py | 31 | |
| LOW | python/tvm/topi/gpu/scatter_nd.py | 35 | |
| LOW | python/tvm/topi/nn/fifo_buffer.py | 29 | |
| LOW | python/tvm/topi/nn/bitserial_util.py | 29 | |
| LOW | python/tvm/topi/nn/utils.py | 118 | |
| LOW | python/tvm/topi/nn/utils.py | 167 | |
| LOW | python/tvm/topi/nn/utils.py | 215 | |
| LOW | python/tvm/topi/nn/utils.py | 274 | |
| LOW | python/tvm/topi/vision/nms_util.py | 87 | |
| LOW | python/tvm/topi/vision/nms_util.py | 126 | |
| LOW | python/tvm/topi/vision/nms_util.py | 288 | |
| LOW | python/tvm/topi/vision/nms.py | 39 | |
| LOW | python/tvm/topi/vision/nms.py | 182 | |
| LOW | python/tvm/topi/vision/nms.py | 664 | |
| LOW | python/tvm/topi/vision/nms.py | 706 | |
| LOW | python/tvm/topi/vision/nms.py | 778 | |
| LOW | python/tvm/topi/vision/nms.py | 852 | |
| LOW | python/tvm/topi/vision/nms.py | 898 | |
| LOW | python/tvm/topi/vision/nms.py | 944 | |
| LOW | python/tvm/topi/vision/nms.py | 670 | |
| LOW | python/tvm/topi/vision/nms.py | 724 | |
| LOW | python/tvm/topi/vision/nms.py | 781 | |
| LOW | python/tvm/topi/vision/nms.py | 824 | |
| LOW | python/tvm/topi/image/resize.py | 140 | |
| LOW | python/tvm/topi/image/resize.py | 178 | |
| LOW | python/tvm/topi/image/resize.py | 224 | |
| LOW | python/tvm/topi/image/resize.py | 373 | |
| LOW | python/tvm/topi/image/resize.py | 495 | |
| LOW | python/tvm/topi/image/resize.py | 746 | |
| LOW | python/tvm/topi/image/resize.py | 959 | |
| LOW | python/tvm/topi/testing/adaptive_pool_python.py | 91 | |
| 1751 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | version.py | 37 | # --------------------------------------------------- |
| MEDIUM | version.py | 51 | # --------------------------------------------------- |
| MEDIUM | python/tvm/base.py | 29 | # ---------------------------- |
| MEDIUM | python/tvm/base.py | 31 | # ---------------------------- |
| MEDIUM | python/tvm/base.py | 36 | # ---------------------------- |
| MEDIUM | python/tvm/base.py | 38 | # ---------------------------- |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 122 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 124 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 133 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 135 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 171 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 173 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 182 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 184 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 187 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 189 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 192 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 194 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 203 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 205 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 212 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 214 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 360 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 362 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 368 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 370 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 56 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 58 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 83 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 85 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 147 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 149 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 225 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 227 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 252 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 254 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 266 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 268 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 283 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 285 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 305 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 307 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 338 | # ===================================================================== |
| MEDIUM | python/tvm/target/tag_registry/cuda.py | 340 | # ===================================================================== |
| MEDIUM | python/tvm/tirx/layout.py | 528 | # ------------------------------------------------------------------ |
| MEDIUM | python/tvm/tirx/layout.py | 530 | # ------------------------------------------------------------------ |
| MEDIUM | python/tvm/tirx/layout.py | 557 | # ------------------------------------------------------------------ |
| MEDIUM | python/tvm/tirx/layout.py | 559 | # ------------------------------------------------------------------ |
| MEDIUM | python/tvm/tirx/layout.py | 616 | # ------------------------------------------------------------------ |
| MEDIUM | python/tvm/tirx/layout.py | 618 | # ------------------------------------------------------------------ |
| MEDIUM | python/tvm/tirx/lang/alloc_pool.py | 27 | # --------------------------------------------------------------------------- |
| MEDIUM | python/tvm/tirx/lang/alloc_pool.py | 29 | # --------------------------------------------------------------------------- |
| MEDIUM | python/tvm/tirx/lang/alloc_pool.py | 49 | # --------------------------------------------------------------------------- |
| MEDIUM | python/tvm/tirx/lang/alloc_pool.py | 51 | # --------------------------------------------------------------------------- |
| MEDIUM | python/tvm/tirx/lang/alloc_pool.py | 176 | # --------------------------------------------------------------------------- |
| MEDIUM | python/tvm/tirx/lang/alloc_pool.py | 178 | # --------------------------------------------------------------------------- |
| MEDIUM | python/tvm/tirx/lang/alloc_pool.py | 228 | # --------------------------------------------------------------------------- |
| MEDIUM | python/tvm/tirx/lang/alloc_pool.py | 230 | # --------------------------------------------------------------------------- |
| MEDIUM | python/tvm/tirx/lang/alloc_pool.py | 394 | # --------------------------------------------------------------------------- |
| MEDIUM | python/tvm/tirx/lang/alloc_pool.py | 396 | # --------------------------------------------------------------------------- |
| 405 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | …hon/tvm/s_tir/meta_schedule/cost_model/random_model.py | 0 | load the cost model from given file location. parameters ---------- path : str the file path. |
| HIGH | python/tvm/s_tir/meta_schedule/cost_model/cost_model.py | 0 | load the cost model from given file location. parameters ---------- path : str the file path. |
| HIGH | python/tvm/s_tir/meta_schedule/cost_model/cost_model.py | 0 | load the cost model from given file location. parameters ---------- path : str the file path. |
| HIGH | …hon/tvm/s_tir/meta_schedule/cost_model/random_model.py | 0 | save the cost model to given file location. parameters ---------- path : str the file path. |
| HIGH | python/tvm/s_tir/meta_schedule/cost_model/cost_model.py | 0 | save the cost model to given file location. parameters ---------- path : str the file path. |
| HIGH | python/tvm/s_tir/meta_schedule/cost_model/cost_model.py | 0 | save the cost model to given file location. parameters ---------- path : str the file path. |
| HIGH | …hon/tvm/s_tir/meta_schedule/cost_model/random_model.py | 0 | update the cost model given running results. parameters ---------- context : tunecontext, the tuning context. candidates |
| HIGH | python/tvm/s_tir/meta_schedule/cost_model/cost_model.py | 0 | update the cost model given running results. parameters ---------- context : tunecontext, the tuning context. candidates |
| HIGH | python/tvm/s_tir/meta_schedule/cost_model/cost_model.py | 0 | update the cost model given running results. parameters ---------- context : tunecontext, the tuning context. candidates |
| HIGH | python/tvm/contrib/mkl.py | 0 | create an extern op that compute matrix mult of a and rhs with crhslas this function serves as an example on how to call |
| HIGH | python/tvm/contrib/mkl.py | 0 | create an extern op that compute matrix mult of a and rhs with crhslas this function serves as an example on how to call |
| HIGH | python/tvm/contrib/cblas.py | 0 | create an extern op that compute matrix mult of a and rhs with crhslas this function serves as an example on how to call |
| HIGH | python/tvm/contrib/dnnl.py | 0 | create an extern op that compute matrix mult of a and rhs with crhslas this function serves as an example on how to call |
| HIGH | python/tvm/contrib/hipblas.py | 0 | create an extern op that compute matrix mult of a and rhs with cublas parameters ---------- lhs : tensor the left matrix |
| HIGH | python/tvm/contrib/cublaslt.py | 0 | create an extern op that compute matrix mult of a and rhs with cublas parameters ---------- lhs : tensor the left matrix |
| HIGH | python/tvm/contrib/cublas.py | 0 | create an extern op that compute matrix mult of a and rhs with cublas parameters ---------- lhs : tensor the left matrix |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/contrib/hexagon/build.py | 0 | abstract method implementation. see description in hexagonlauncherrpc. |
| HIGH | python/tvm/relax/pipeline.py | 0 | get the default legalization passes for the given target. |
| HIGH | python/tvm/relax/pipeline.py | 0 | get the default legalization passes for the given target. |
| HIGH | python/tvm/relax/pipeline.py | 0 | get the default legalization passes for the given target. |
| HIGH | python/tvm/relax/frontend/nn/llm/_kernel_common.py | 0 | the rope mode of the paged kv cache. if it is none, the kv cache will not apply rope to q and k. if it is normal, rope w |
| HIGH | …x/test_runtime_builtin_paged_attention_kv_cache_cpu.py | 0 | the rope mode of the paged kv cache. if it is none, the kv cache will not apply rope to q and k. if it is normal, rope w |
| HIGH | docs/how_to/tutorials/optimize_llm.py | 0 | the rope mode of the paged kv cache. if it is none, the kv cache will not apply rope to q and k. if it is normal, rope w |
| HIGH | python/tvm/relax/training/optimizer.py | 0 | use blockbuilder to construct an optimizer function that executes updates of the parameters and the optimizer state. `in |
| HIGH | python/tvm/relax/training/optimizer.py | 0 | use blockbuilder to construct an optimizer function that executes updates of the parameters and the optimizer state. `in |
| HIGH | python/tvm/relax/training/optimizer.py | 0 | use blockbuilder to construct an optimizer function that executes updates of the parameters and the optimizer state. `in |
| HIGH | python/tvm/relax/testing/transform.py | 0 | inner function for the dataflow inplace transformation exposed for testing. |
| HIGH | python/tvm/relax/testing/transform.py | 0 | inner function for the dataflow inplace transformation exposed for testing. |
| HIGH | python/tvm/relax/testing/transform.py | 0 | inner function for the dataflow inplace transformation exposed for testing. |
| HIGH | python/tvm/relax/testing/transform.py | 0 | inner function for the dataflow inplace transformation exposed for testing. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the destination expression (output) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the destination expression (output) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the destination expression (output) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the destination expression (output) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the source expressions (inputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the destination expressions (outputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the destination expressions (outputs) of the operator. |
| HIGH | python/tvm/tirx/operator/tile_primitive/ops.py | 0 | get the destination expressions (outputs) of the operator. |
| 40 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| CRITICAL | …vm/tirx/operator/tile_primitive/cuda/copy_async/tma.py | 1189 | Tx.ptx.cp_async.bulk.tensor.g2c( |
| CRITICAL | …vm/tirx/operator/tile_primitive/cuda/copy_async/tma.py | 1201 | Tx.ptx.cp_async.bulk.tensor.s2g( |
| CRITICAL | …vm/tirx/operator/tile_primitive/cuda/copy_async/tma.py | 1209 | Tx.ptx.cp_async.bulk.tensor.s2g_reduce( |
| CRITICAL | …end/adreno/test_transform_fold_vdevice_scope_change.py | 29 | mod = tvm.relax.backend.adreno.transform.FoldVDeviceScopeChange()(input) |
| CRITICAL | …backend/adreno/test_transform_annotate_custom_scope.py | 95 | mod = tvm.relax.backend.adreno.transform.AnnotateCustomMemoryScope(tgt)(mod) |
| CRITICAL | …backend/adreno/test_transform_annotate_custom_scope.py | 107 | mod = tvm.relax.backend.adreno.transform.FoldVDeviceScopeChange()(mod) |
| CRITICAL | tests/python/tirx/test_printer_tir_namespaces.py | 412 | 'Tx.ptx.cp_async.bulk.tensor.g2c(2, tm, 0, tm, 0, 1, Tx.uint64(0), 0, 0, 1, "")', |
| CRITICAL | tests/python/tirx/test_printer_tir_namespaces.py | 425 | 'Tx.ptx.cp_async.bulk.tensor.g2c_prefetch(2, tm, Tx.uint64(0), 0, 0, 0, "")', |
| CRITICAL | tests/python/tirx/test_printer_tir_namespaces.py | 429 | 'tm = Tx.handle()\nTx.ptx.cp_async.bulk.tensor.s2g(2, 0, tm, Tx.uint64(0), 0, 0, 0, "")', |
| CRITICAL | tests/python/tirx/test_parser_printer.py | 1897 | Tx.ptx.cp_async.bulk.tensor.g2c( |
| CRITICAL | tests/python/tirx/test_parser_printer.py | 1918 | Tx.ptx.cp_async.bulk.tensor.s2g( |
| CRITICAL | tests/python/tirx/test_parser_printer.py | 1938 | Tx.ptx.cp_async.bulk.tensor.g2c_prefetch( |
| CRITICAL | tests/python/tirx/test_parser_printer.py | 1959 | Tx.ptx.cp_async.bulk.tensor.s2g_reduce( |
| CRITICAL | tests/python/tirx/test_verifier.py | 263 | Tx.ptx.cp_async.bulk.tensor.g2c(2, A_smem.data, bar.data, Tx.address_of(A_map), 0, 1, "", 0, |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 679 | Tx.ptx.cp_async.bulk.tensor.g2c(len(shape), A_smem.access_ptr(Buffer.WRITE, offs |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 682 | Tx.ptx.cp_async.bulk.tensor.g2c(len(shape), A_smem.access_ptr(Buffer.WRITE, offs |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 685 | Tx.ptx.cp_async.bulk.tensor.g2c(len(shape), A_smem.access_ptr(Buffer.WRITE, offs |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 688 | Tx.ptx.cp_async.bulk.tensor.g2c(len(shape), A_smem.access_ptr(Buffer.WRITE, offs |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 697 | Tx.ptx.cp_async.bulk.tensor.s2g(len(shape), A_smem.access_ptr("r", offset=0), Tx |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 347 | Tx.ptx.cp_async.bulk.tensor.g2c(len(shape), A_smem.data, Tx.address_of(bar), Tx.address_ |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 356 | Tx.ptx.cp_async.bulk.tensor.s2g(len(shape), A_smem.access_ptr("r", offset=0), Tx.address |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 509 | Tx.ptx.cp_async.bulk.tensor.g2c(len(shape), A_smem.data, Tx.address_of(bar), Tx.address_ |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 518 | Tx.ptx.cp_async.bulk.tensor.s2g(len(shape), A_smem.access_ptr("r", offset=0), Tx.address |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 598 | Tx.ptx.cp_async.bulk.tensor.g2c(len(shape), A_smem.data, Tx.address_of(bar), Tx. |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 607 | Tx.ptx.cp_async.bulk.tensor.s2g(len(shape), A_smem.access_ptr("r", offset=0), Tx |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 758 | Tx.ptx.cp_async.bulk.tensor.s2g(len(shape), A_smem.access_ptr("r", offset=0), Tx.address_of(A_ma |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 847 | Tx.ptx.cp_async.bulk.tensor.g2c(len(shapeA), A_smem.data, Tx.address_of(bar), Tx.address_of(A_ma |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 848 | Tx.ptx.cp_async.bulk.tensor.g2c(len(shapeB), B_smem.data, Tx.address_of(bar), Tx.address_of(B_ma |
| CRITICAL | tests/python/tirx/codegen/test_codegen_hopper.py | 1005 | Tx.ptx.cp_async.bulk.tensor.g2c(len(shapeB), B_smem.data, Tx.address_of(bar), Tx.address_of(B_ma |
| CRITICAL | tests/python/tirx/codegen/test_codegen_cuda.py | 303 | Tx.ptx.cp_async.bulk.tensor.g2c( |
| CRITICAL | tests/python/tirx/codegen/test_codegen_cuda.py | 315 | Tx.ptx.cp_async.bulk.tensor.g2c( |
| CRITICAL | tests/python/tirx/codegen/test_codegen_cuda.py | 327 | Tx.ptx.cp_async.bulk.tensor.s2g( |
| CRITICAL | tests/python/tirx/codegen/test_codegen_cuda.py | 331 | Tx.ptx.cp_async.bulk.tensor.g2c_bar_addr( |
| CRITICAL | tests/python/tirx/codegen/test_codegen_cuda.py | 344 | Tx.ptx.cp_async.bulk.tensor.g2c_bar_addr( |
| CRITICAL | tests/python/tirx/codegen/test_codegen_cuda.py | 357 | Tx.ptx.cp_async.bulk.tensor.g2c_bar_addr( |
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | python/tvm/topi/testing/poolnd_python.py | 163 | # Create a padded array, and a boolean mask showing which values are padded values |
| MEDIUM | python/tvm/te/operation.py | 178 | # The following code is equivalent to numpy.cumsum |
| MEDIUM | python/tvm/relax/base_py_module.py | 481 | # Create a wrapper that handles both instance methods and static functions |
| MEDIUM | python/tvm/relax/relax_to_pyfunc_converter.py | 87 | # Create a copy of the current IRModule |
| MEDIUM | python/tvm/relax/relax_to_pyfunc_converter.py | 128 | # Create the Python function |
| MEDIUM | python/tvm/relax/frontend/tflite/tflite_frontend.py | 7593 | # Define a tf.Module containing a tf.function with network definition |
| MEDIUM | python/tvm/relax/frontend/nn/core.py | 359 | """This method provides an iterator over module parameters, |
| MEDIUM | python/tvm/relax/frontend/nn/core.py | 376 | """This method provides an iterator over module parameters, |
| MEDIUM | python/tvm/relax/frontend/onnx/onnx_frontend.py | 3716 | # Create a tensor of zeros then scatter our data through it. |
| MEDIUM | …n/tvm/relax/frontend/torch/base_fx_graph_translator.py | 1413 | # Create a placeholder for indices (empty tensor with same shape as output) |
| MEDIUM | …n/tvm/relax/frontend/torch/base_fx_graph_translator.py | 1429 | # Create a placeholder for indices (empty tensor with same shape as output) |
| MEDIUM | …n/tvm/relax/frontend/torch/base_fx_graph_translator.py | 1445 | # Create a placeholder for indices (empty tensor with same shape as output) |
| MEDIUM | python/tvm/relax/frontend/torch/fx_translator.py | 1230 | # Define the module |
| MEDIUM | …vm/relax/frontend/torch/exported_program_translator.py | 185 | # This method is called for batch_norm in training mode |
| MEDIUM | …vm/relax/frontend/torch/exported_program_translator.py | 2130 | # Define the module |
| MEDIUM | python/tvm/relax/testing/nn.py | 228 | # Define a linear layer |
| MEDIUM | python/tvm/relax/transform/transform.py | 1116 | # Define the pass input IRModule |
| MEDIUM | python/tvm/relax/transform/transform.py | 1127 | # Define the customized legalization function for "relax.add" |
| MEDIUM | python/tvm/tirx/stmt_functor.py | 493 | # Create the data_or_idx parameter based on what's available |
| MEDIUM | python/tvm/tirx/expr.py | 244 | # This class is not manipulated by C++. So use python's identity check function is sufficient |
| MEDIUM | python/tvm/tirx/expr.py | 284 | # This class is not manipulated by C++. So use python's identity check function is sufficient |
| MEDIUM | …tvm/tirx/operator/tile_primitive/trn/binary/default.py | 74 | # Define the implementation function |
| MEDIUM | ci/scripts/github/github_tag_teams.py | 272 | # Create the new PR/issue body |
| MEDIUM | tests/python/driver/test_compile.py | 63 | # Define a simple Relax program |
| MEDIUM | tests/python/tirx-base/test_tir_expr_functor.py | 830 | # Create a simple expression |
| MEDIUM | tests/python/tirx-base/test_tir_expr_functor.py | 833 | # Create a mutator that replaces 'n' with a constant |
| MEDIUM | tests/python/tirx-base/test_tir_stmt_functor.py | 994 | # Create a new IntImm with negated value |
| MEDIUM | tests/python/runtime/test_executable.py | 107 | # Create a temporary directory for the library |
| MEDIUM | tests/python/runtime/test_executable.py | 181 | # Create an executable |
| MEDIUM | tests/python/runtime/test_executable.py | 237 | # Create an executable |
| MEDIUM | tests/python/s_tir/dlight/test_gpu_fallback.py | 203 | # This function is identical, except that it is explicitly |
| MEDIUM | tests/python/contrib/test_util.py | 77 | # Creating a new TempDirectory should fail now |
| MEDIUM | …ython/contrib/test_hexagon/test_benchmark_maxpool2d.py | 227 | # Create the host-side directory for this benchmark run's files / logs... |
| MEDIUM | …on/contrib/test_hexagon/test_benchmark_elemwise_add.py | 121 | # The below code is commented is commented to avoid unreachable error |
| MEDIUM | …on/contrib/test_hexagon/test_benchmark_elemwise_add.py | 187 | # Create the host-side directory for this benchmark run's files / logs... |
| MEDIUM | …on/contrib/test_hexagon/test_benchmark_elemwise_add.py | 218 | # Create an actual Hexagon-native shared object file, initially stored on the |
| MEDIUM | …on/contrib/test_hexagon/test_benchmark_elemwise_add.py | 244 | # Create the target-side tensors to hold the primfunc's inputs and outputs... |
| MEDIUM | tests/python/codegen/test_target_codegen_llvm.py | 50 | # Create an intrinsic that returns void. |
| MEDIUM | tests/python/tirx-transform/test_tir_functor.py | 266 | # Create a simple for loop |
| MEDIUM | tests/python/tirx-transform/test_tir_functor.py | 298 | # Create an evaluate statement with an expression |
| MEDIUM | …/tirx-analysis/test_tir_analysis_verify_well_formed.py | 446 | # Create an undeclared buffer using an explicit data pointer that is NOT |
| MEDIUM | …ime_builtin_paged_attention_kv_cache_mla_flashinfer.py | 407 | # Create a mask similar to np.tril and np.triu. |
| MEDIUM | tests/python/relax/test_pytorch_integration.py | 160 | # Define a custom function |
| MEDIUM | tests/python/relax/test_base_py_module.py | 115 | # Create a simple IRModule without TIR functions for GPU testing |
| MEDIUM | tests/python/relax/test_dlpack_integration.py | 213 | # Create a simple IRModule |
| MEDIUM | …st_runtime_builtin_paged_attention_kv_cache_mla_tir.py | 389 | # Create a mask similar to np.tril and np.triu. |
| MEDIUM | tests/python/tirx/transform/test_stmt_functor.py | 1006 | # Create a new IntImm with negated value |
| MEDIUM | tests/python/tirx/transform/test_expr_functor.py | 830 | # Create a simple expression |
| MEDIUM | tests/python/tirx/transform/test_expr_functor.py | 833 | # Create a mutator that replaces 'n' with a constant |
| MEDIUM | tests/scripts/task_config_build_gpu_other.sh | 19 | # This file is a compiler test to ensure that runtimes can compile |
| MEDIUM | docs/conf.py | 23 | # This file is execfile()d with the current directory set to its |
| MEDIUM | docs/how_to/tutorials/cross_compilation_and_rpc.py | 413 | # The following function shows how we run an OpenCL kernel remotely |
| MEDIUM | docs/how_to/tutorials/customize_opt.py | 121 | # Define a new pass for CUBLAS dispatch |
| MEDIUM | docs/how_to/tutorials/import_model.py | 307 | # Define a simple TF module and convert to TFLite. |
| MEDIUM | docs/how_to/tutorials/optimize_llm.py | 530 | # Create the KVCache |
| MEDIUM | docs/get_started/tutorials/ir_module.py | 57 | # Create a dummy model |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | python/tvm/ir/instrument.py | 338 | except Exception: # pylint: disable=broad-exception-caught |
| LOW | python/tvm/ir/base.py | 35 | except Exception: |
| LOW | python/tvm/target/tag_registry/metal.py | 30 | except Exception: # pylint: disable=broad-except |
| MEDIUM | python/tvm/target/tag_registry/metal.py | 24 | def _register_metal_tag(name, max_threads, shared_mem, warp_size, mcpu): |
| LOW | python/tvm/runtime/_tensor.py | 114 | except Exception: |
| LOW | python/tvm/s_tir/schedule/_type_checker.py | 350 | except Exception: |
| LOW | …hon/tvm/s_tir/meta_schedule/post_optimization/space.py | 237 | except Exception: # pylint: disable=broad-except, invalid-name |
| LOW | …hon/tvm/s_tir/meta_schedule/post_optimization/space.py | 251 | except Exception: # pylint: disable=broad-except, invalid-name |
| LOW | python/tvm/s_tir/meta_schedule/runner/local_runner.py | 310 | except Exception as exception: # pylint: disable=broad-except |
| LOW | python/tvm/s_tir/meta_schedule/runner/rpc_runner.py | 127 | except Exception as exception: # pylint: disable=broad-except |
| MEDIUM | python/tvm/s_tir/meta_schedule/runner/rpc_runner.py | 119 | def result(self) -> RunnerResult: |
| LOW | …n/tvm/s_tir/meta_schedule/testing/validate_database.py | 768 | except Exception as e: # pylint: disable=broad-except, invalid-name |
| LOW | python/tvm/s_tir/meta_schedule/testing/tune_utils.py | 107 | except Exception as exc: # pylint: disable=broad-except |
| LOW | python/tvm/s_tir/dlight/adreno/pool.py | 91 | except Exception: # pylint: disable=broad-except |
| LOW | python/tvm/s_tir/dlight/adreno/fallback.py | 82 | except Exception: # pylint: disable=broad-exception-caught |
| LOW | python/tvm/s_tir/dlight/adreno/fallback.py | 94 | except Exception: # pylint: disable=broad-exception-caught |
| LOW | python/tvm/s_tir/dlight/analysis/common_analysis.py | 296 | except Exception: # pylint: disable=broad-except |
| LOW | python/tvm/s_tir/dlight/analysis/common_analysis.py | 399 | except Exception: |
| LOW | python/tvm/contrib/download.py | 123 | except Exception as err: |
| LOW | python/tvm/script/parser/core/parser.py | 338 | except Exception as err: # pylint: disable=broad-except |
| LOW | python/tvm/script/parser/core/parser.py | 629 | except Exception as diag_err: |
| LOW | python/tvm/script/parser/core/parser.py | 667 | except Exception as err: # pylint: disable=broad-except |
| MEDIUM | python/tvm/script/parser/core/parser.py | 59 | def context(): |
| LOW | python/tvm/script/parser/core/entry.py | 142 | except Exception as err: # pylint: disable=broad-exception-caught |
| LOW | python/tvm/script/parser/core/entry.py | 172 | except Exception as error: |
| MEDIUM | python/tvm/script/parser/core/entry.py | 173 | print(f"Error calling Python function {pyfunc.__name__}: {error}") |
| LOW | python/tvm/script/parser/core/evaluator.py | 268 | except Exception as err: # pylint: disable=broad-except |
| LOW | python/tvm/script/parser/core/evaluator.py | 287 | except Exception as err: # pylint: disable=broad-except |
| LOW | python/tvm/script/parser/core/evaluator.py | 496 | except Exception as err: # pylint: disable=broad-except |
| LOW | python/tvm/script/parser/ir/entry.py | 105 | except Exception: # pylint: disable=broad-exception-caught |
| LOW | python/tvm/support/popen_pool.py | 440 | except Exception as exception: |
| MEDIUM | python/tvm/support/popen_pool.py | 122 | def __del__(self): |
| LOW | python/tvm/exec/popen_worker.py | 108 | except Exception as exception: |
| LOW | python/tvm/relax/base_py_module.py | 147 | except Exception as error: |
| LOW | python/tvm/relax/base_py_module.py | 155 | except Exception as error: |
| LOW | python/tvm/relax/base_py_module.py | 449 | except Exception as error: |
| LOW | python/tvm/relax/frontend/nn/exporter.py | 39 | except Exception as exception: |
| MEDIUM | python/tvm/relax/frontend/onnx/onnx_frontend.py | 5510 | print(f"Error converting operator {op_name}, with inputs: {inputs}") |
| LOW | python/tvm/relax/frontend/onnx/onnx_frontend.py | 5735 | except Exception as exception: # pylint: disable=c-extension-no-member, broad-except |
| LOW | python/tvm/relax/frontend/torch/fx_translator.py | 1253 | except Exception: |
| LOW | python/tvm/relax/op/_op_gradient.py | 70 | except Exception as error: |
| LOW | python/tvm/relax/op/_op_gradient.py | 81 | except Exception as error: |
| LOW | python/tvm/relax/script/parser/parser.py | 107 | except Exception as err: # pylint: disable=broad-except |
| LOW | python/tvm/relax/script/parser/parser.py | 117 | except Exception as err: # pylint: disable=broad-except |
| LOW | python/tvm/tirx/layout.py | 145 | except Exception: |
| LOW | python/tvm/tirx/bench.py | 192 | except Exception: |
| LOW | python/tvm/tirx/bench.py | 216 | except Exception: |
| LOW | python/tvm/tirx/bench.py | 222 | except Exception: |
| LOW | python/tvm/tirx/script/__init__.py | 57 | except Exception: |
| MEDIUM | python/tvm/tirx/script/__init__.py | 54 | def _fn(*args, workspace=None, config=None, dispatch=None, **kwargs): |
| LOW | python/tvm/tirx/script/parser/parser.py | 407 | except Exception: # pylint: disable=broad-except |
| LOW | python/tvm/tirx/script/parser/parser.py | 491 | except Exception: # pylint: disable=broad-except |
| LOW | python/tvm/tirx/script/parser/parser.py | 638 | except Exception: # pylint: disable=broad-except |
| LOW | python/tvm/tirx/script/parser/parser.py | 895 | except Exception: # pylint: disable=broad-except |
| LOW | python/tvm/tirx/script/builder/ir.py | 2866 | except Exception: # pragma: no cover |
| LOW | python/tvm/tirx/script/builder/ir.py | 2881 | except Exception: # pragma: no cover |
| LOW | python/tvm/tirx/script/builder/ir.py | 3414 | except Exception: |
| LOW | python/tvm/tirx/script/builder/ir.py | 3432 | except Exception: |
| LOW | python/tvm/tirx/operator/intrinsics/_schema.py | 172 | except Exception: |
| LOW | python/tvm/tirx/operator/intrinsics/_schema.py | 179 | except Exception: |
| 44 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | python/tvm/topi/vision/nms.py | 220 | # Step 1: Reorder data by sorted score |
| LOW | python/tvm/topi/vision/nms.py | 241 | # Step 2: Apply NMS - greedy suppression |
| LOW | python/tvm/ir/transform.py | 295 | The following code block decorates a module pass class. |
| LOW | python/tvm/s_tir/meta_schedule/runner/local_runner.py | 146 | # Step 1: create the local runtime module |
| LOW | python/tvm/s_tir/meta_schedule/runner/local_runner.py | 149 | # Step 2: Allocate input arguments |
| LOW | python/tvm/s_tir/meta_schedule/runner/local_runner.py | 157 | # Step 3: Run time_evaluator |
| LOW | python/tvm/s_tir/meta_schedule/runner/rpc_runner.py | 394 | # Step 3: Allocate input arguments |
| LOW | python/tvm/s_tir/meta_schedule/runner/rpc_runner.py | 402 | # Step 4: Run time_evaluator |
| LOW | python/tvm/script/parser/ir/parser.py | 54 | # Step 1: Check if this class inherits from BasePyModule |
| LOW | python/tvm/script/parser/ir/parser.py | 147 | # We need to check if we're in a class that inherits from BasePyModule |
| LOW | python/tvm/relax/transform/transform.py | 1688 | The following code block decorates a function pass class. |
| LOW | python/tvm/relax/transform/transform.py | 1832 | The following code block decorates a dataflowblock pass class. |
| LOW | python/tvm/tirx/build.py | 183 | # Step 0: Determine the target in environment |
| LOW | python/tvm/tirx/build.py | 191 | # Step 1: Determine the target to search for tirx pipeline |
| LOW | python/tvm/tirx/build.py | 202 | # Step 2: Determine the host target |
| LOW | python/tvm/tirx/build.py | 214 | # Step 3: Bind the target to the input module |
| LOW | python/tvm/tirx/build.py | 217 | # Step 4: Apply the tirx pipeline |
| LOW | python/tvm/tirx/build.py | 229 | # Step 5: Get host and device modules |
| LOW | python/tvm/tirx/build.py | 232 | # Step 6: Apply finalization passes |
| LOW | python/tvm/tirx/transform/function_pass.py | 110 | The following code block decorates a function pass class. |
| LOW | include/tvm/topi/nn/pooling.h | 572 | // In the case of ceil_mode=True, we need to check if the last pooling window is valid. |
| LOW | include/tvm/ir/config_schema.h | 125 | // Step 1: validate/coerce and materialize options in declaration order |
| LOW | include/tvm/ir/config_schema.h | 140 | // Step 2: unknown-key policy |
| LOW | include/tvm/ir/config_schema.h | 157 | // Step 3: whole-object canonicalization (last) |
| LOW | tests/python/relax/test_pytorch_integration.py | 279 | # Step 1: TIR matmul |
| LOW | tests/python/relax/test_pytorch_integration.py | 284 | # Step 2: ReLU |
| LOW | tests/python/relax/test_pytorch_integration.py | 289 | # Step 3: Softmax via call_dps_packed |
| LOW | tests/python/relax/test_pytorch_integration.py | 294 | # Step 4: Identity function |
| LOW | tests/python/relax/test_base_py_module_printer.py | 437 | # We need to check if they exist in the IRModule's pyfuncs |
| LOW | docs/how_to/tutorials/cross_compilation_and_rpc.py | 498 | # Step 1: Define and Export PyTorch Model |
| LOW | docs/how_to/tutorials/cross_compilation_and_rpc.py | 524 | # Step 2: Convert to Relax and Prepare for Compilation |
| LOW | docs/how_to/tutorials/cross_compilation_and_rpc.py | 536 | # Step 3: Cross-Compile for Target Device |
| LOW | docs/how_to/tutorials/cross_compilation_and_rpc.py | 590 | # Step 4: Deploy to Remote Device via RPC |
| LOW | docs/how_to/tutorials/cross_compilation_and_rpc.py | 653 | # Step 5: Run Inference on Remote Device |
| LOW | docs/how_to/tutorials/cross_compilation_and_rpc.py | 692 | # Step 6: Performance Evaluation (Optional) |
| LOW | …s/how_to/tutorials/mix_python_and_tvm_with_pymodule.py | 72 | # Step 1: Your First Hybrid Module |
| LOW | …s/how_to/tutorials/mix_python_and_tvm_with_pymodule.py | 125 | # Step 2: Debugging — The Main Selling Point |
| LOW | …s/how_to/tutorials/mix_python_and_tvm_with_pymodule.py | 189 | # Step 3: A Realistic Pipeline — Python, TIR, and Packed Functions |
| LOW | …s/how_to/tutorials/mix_python_and_tvm_with_pymodule.py | 262 | # Step 4: Relax-to-Python Converter — Verify at Any Compilation Stage |
| LOW | …s/how_to/tutorials/mix_python_and_tvm_with_pymodule.py | 334 | # Step 5: R.call_py_func — Python Callbacks in Compiled IR |
| LOW | …s/how_to/tutorials/mix_python_and_tvm_with_pymodule.py | 385 | # Step 6: Cross-Level Calls and Symbolic Shapes |
| LOW | docs/how_to/tutorials/bring_your_own_codegen.py | 59 | # Step 1: Import the backend to register its patterns |
| LOW | docs/how_to/tutorials/bring_your_own_codegen.py | 84 | # Step 2: Define a model |
| LOW | docs/how_to/tutorials/bring_your_own_codegen.py | 105 | # Step 3: Partition the graph |
| LOW | docs/how_to/tutorials/bring_your_own_codegen.py | 133 | # Step 4: Run codegen |
| LOW | docs/how_to/tutorials/bring_your_own_codegen.py | 149 | # Step 5: Build and run |
| LOW | docs/how_to/tutorials/bring_your_own_codegen.py | 169 | # Step 6: Conv2D + ReLU |
| LOW | docs/how_to/tutorials/export_and_load_executable.py | 247 | # # Step 1: Load the compiled library |
| LOW | docs/how_to/tutorials/export_and_load_executable.py | 250 | # # Step 2: Create Virtual Machine |
| LOW | docs/how_to/tutorials/export_and_load_executable.py | 254 | # # Step 3: Load parameters from the .npz file |
| LOW | docs/how_to/tutorials/export_and_load_executable.py | 259 | # # Step 4: Prepare input data |
| LOW | docs/how_to/tutorials/export_and_load_executable.py | 263 | # # Step 5: Run inference (pass input followed by all parameters) |
| LOW | docs/how_to/tutorials/export_and_load_executable.py | 266 | # # Step 6: Extract result (output may be tuple or single Tensor) |
| LOW | docs/how_to/tutorials/export_and_load_executable.py | 342 | # # Step 1: Cross-compile for ARM target (on local machine) |
| LOW | docs/how_to/tutorials/export_and_load_executable.py | 347 | # # Step 2: Connect to remote device RPC server |
| LOW | docs/how_to/tutorials/export_and_load_executable.py | 350 | # # Step 3: Upload the compiled library and parameters |
| LOW | docs/how_to/tutorials/export_and_load_executable.py | 354 | # # Step 4: Load and run on remote device |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | python/tvm/topi/sparse_reshape.py | 134 | # Check if prev_shape and new_shape are equal |
| LOW | python/tvm/contrib/hexagon/build.py | 115 | # Check if adb available via PATH |
| LOW | python/tvm/contrib/hexagon/build.py | 119 | # Check if adb available via env vars or default directories |
| LOW | python/tvm/script/parser/ir/parser.py | 146 | # Check if Python functions are allowed in this context |
| LOW | python/tvm/testing/plugin.py | 129 | # Check if any explicit parametrizations exist, and apply one |
| LOW | python/tvm/relax/relax_to_pyfunc_converter.py | 419 | # Check if it's a function parameter |
| LOW | python/tvm/relax/relax_to_pyfunc_converter.py | 424 | # Check if it's a bound variable |
| LOW | python/tvm/relax/frontend/tflite/tflite_frontend.py | 847 | # Check if the tensors are quantized. Parse if yes. |
| LOW | python/tvm/relax/frontend/tflite/tflite_frontend.py | 3250 | # Check if the input tensor is quantized, call QNN op |
| LOW | …n/tvm/relax/frontend/torch/base_fx_graph_translator.py | 163 | # Check if this is a null_value expression |
| LOW | …n/tvm/relax/frontend/torch/base_fx_graph_translator.py | 1957 | # Check if all indices can be squeezed to 1D for sequential take |
| LOW | python/tvm/relax/frontend/torch/fx_translator.py | 105 | # Check if input is integer type and convert to float32 if needed |
| LOW | python/tvm/relax/frontend/torch/fx_translator.py | 115 | # Check if input is integer type and convert to float32 if needed |
| LOW | …vm/relax/frontend/torch/exported_program_translator.py | 104 | # Check if input is integer type and convert to float32 if needed |
| LOW | …vm/relax/frontend/torch/exported_program_translator.py | 114 | # Check if input is integer type and convert to float32 if needed |
| LOW | python/tvm/relax/backend/cuda/cutlass.py | 137 | # Check if any dimensions are symbolic. |
| LOW | python/tvm/relax/backend/cuda/cudnn.py | 55 | # Check if the data types of input and weights are supported by cuDNN BYOC |
| LOW | python/tvm/relax/script/parser/parser.py | 123 | # Check if it calls into a func |
| LOW | python/tvm/relax/script/parser/parser.py | 147 | # Check if it is a recursive function |
| LOW | python/tvm/tirx/stmt_functor.py | 737 | # Check if anything changed |
| LOW | python/tvm/tirx/buffer.py | 351 | # Check if all positional args are integers/PrimExprs with dtype int32 or int64 (the shape) # noqa: E501 |
| LOW | python/tvm/tirx/script/builder/external_kernel.py | 155 | # Check if NVSHMEM is used - requires cubin output for device library linking |
| LOW | python/tvm/tirx/lang/tile_scheduler.py | 378 | # Check if we're in the full block region |
| LOW | …thon/tvm/tirx/operator/tile_primitive/trn/dim_utils.py | 225 | # Check if there is a direct mapping |
| LOW | tests/lint/check_asf_header.py | 308 | # Check if file should be skipped |
| LOW | tests/lint/check_asf_header.py | 312 | # Check if this file type is supported |
| LOW | tests/python/contrib/test_example_npu.py | 98 | # Check if the example NPU runtime is available |
| LOW | tests/python/codegen/test_target_codegen_aarch64.py | 615 | # Check if the vscale_range() attribute exists |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 471 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 50 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 97 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 126 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 176 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 220 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 276 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 296 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 319 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 333 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 354 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 388 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 410 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 426 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 444 | # Check if the generated ir is expected |
| LOW | tests/python/tvmscript/test_tvmscript_ir_builder_tir.py | 505 | # Check if the generated ir is expected |
| LOW | tests/python/relax/test_tvmscript_pyfunc.py | 145 | # Check if functions attribute exists |
| LOW | tests/python/relax/test_tvmscript_pyfunc.py | 157 | # Check if script method exists |
| LOW | tests/python/relax/test_tvmscript_pyfunc.py | 169 | # Check if inheritance flag exists (this might not be set in all implementations) |
| LOW | tests/python/relax/test_tvmscript_pyfunc.py | 176 | # Check if original class is preserved (this might not be set in all implementations) |
| LOW | tests/python/relax/test_base_py_module.py | 73 | # Check if target contains "cuda" instead of exact match |
| LOW | tests/scripts/ci.py | 552 | # Check if it should be a flag |
| LOW | docs/how_to/tutorials/customize_opt.py | 125 | # Check if CUBLAS is enabled |
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | python/tvm/contrib/cutlass/gemm_profiler.py | 170 | for (int i = 1; i < argc && i < 4; ++i) { |
| HIGH | python/tvm/relax/frontend/tflite/tflite_flexbuffer.py | 87 | # Keys are strings here. The format is all strings separated by null, followed by back |
| HIGH | python/tvm/tirx/operator/intrinsics/cuda/header.py | 643 | GmmaDescriptor(GmmaDescriptor && t) noexcept : desc_(t.desc_) {} |
| HIGH | python/tvm/tirx/operator/intrinsics/cuda/header.py | 652 | GmmaDescriptor& operator=(GmmaDescriptor && t) noexcept { |
| HIGH | …ython/s_tir/meta_schedule/test_meta_schedule_runner.py | 624 | global repeated_args_before # pylint: disable=global-variable-undefined, invalid-name |
| HIGH | …ython/s_tir/meta_schedule/test_meta_schedule_runner.py | 638 | global repeated_args_before # pylint: disable=global-variable-undefined, invalid-name |
| HIGH | …ython/s_tir/meta_schedule/test_meta_schedule_runner.py | 732 | global repeated_args_before # pylint: disable=global-variable-undefined, invalid-name |
| HIGH | …ython/s_tir/meta_schedule/test_meta_schedule_runner.py | 751 | global repeated_args_before # pylint: disable=global-variable-undefined, invalid-name |
| HIGH | …ython/s_tir/meta_schedule/test_meta_schedule_runner.py | 844 | global repeated_args_before # pylint: disable=global-variable-undefined, invalid-name |
| HIGH | …ython/s_tir/meta_schedule/test_meta_schedule_runner.py | 857 | global repeated_args_before # pylint: disable=global-variable-undefined, invalid-name |
| HIGH | …s/python/tirx-transform/test_tir_transform_simplify.py | 361 | simplification that ``!(x || y) == !x && !y``. |
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | version.py | 110 | # Remove the v prefix, mainly to be robust |
| LOW | python/tvm/topi/nn/qnn.py | 67 | # When disabled, just pass through the input values. |
| LOW | python/tvm/topi/nn/qnn.py | 157 | # When disabled simply return the input tensor. |
| LOW | python/tvm/script/parser/core/diagnostics.py | 87 | # Since `findsource` is an internal API of inspect, we just use it |
| LOW | python/tvm/testing/utils.py | 1369 | # this use case can just return the decorated function. |
| LOW | python/tvm/relax/relax_to_pyfunc_converter.py | 642 | # Simple fallback: just add the tensors |
| LOW | python/tvm/relax/frontend/onnx/onnx_frontend.py | 1942 | # from other frameworks and not actually used in TVM. We'll just use |
| LOW | …n/tvm/relax/frontend/torch/base_fx_graph_translator.py | 2291 | # by the translator, and therefore we just return a copy of the input. |
| LOW | …vm/relax/frontend/torch/exported_program_translator.py | 1785 | "bernoulli.p": lambda node: self.env[node.args[0]], # Dropout: just return input |
| LOW | python/tvm/relax/training/loss.py | 364 | # In other cases where ignore_index is not needed, just use the simpe product. |
| LOW | python/tvm/tirx/op.py | 2895 | # mode 2, simply use it with multiple Exprs: |
| LOW | tests/python/tirx-transform/test_tir_functor.py | 164 | # For demonstration, just return the original for now |
| MEDIUM | tests/python/relax/test_relax_to_pyfunc_converter.py | 354 | """Test that operator mapping is comprehensive.""" |
| MEDIUM | tests/python/relax/test_base_py_module_printer.py | 359 | """Test comprehensive error handling and validation.""" |
| MEDIUM | tests/python/relax/test_base_py_module_printer.py | 363 | """Robust data processing with comprehensive error handling.""" |
| MEDIUM | docs/how_to/tutorials/e2e_opt_model.py | 97 | # leverage MetaSchedule to tune the model and store the tuning logs to the database. We also |
| MEDIUM | docs/how_to/tutorials/export_and_load_executable.py | 327 | # for a comprehensive guide on: |
| MEDIUM | docs/how_to/tutorials/optimize_llm.py | 169 | # - QKV projection: We leverage the horizontal fusion on QKV projection and fuse them into one |
| MEDIUM | docs/how_to/tutorials/optimize_llm.py | 171 | # - Attention: We leverage the horizontal fusion on attention and fuse the QKV projection and |
| MEDIUM | docs/deep_dive/tensor_ir/tutorials/tir_creation.py | 262 | # The aforementioned lambda expression encapsulates the computation: |
| MEDIUM | …cs/deep_dive/tensor_ir/tutorials/tir_transformation.py | 175 | # Alternatively, we can output the IRModule in conjunction with the historical trace. |
| MEDIUM | docs/get_started/tutorials/quick_start.py | 113 | # Apache TVM leverage ``pipeline`` to transform and optimize program. |
| MEDIUM | docs/get_started/tutorials/quick_start.py | 125 | # In this tutorial we only demonstrate the overall flow, by leverage ``zero`` optimization |
| LOW | apps/ios_rpc/tests/ios_rpc_test.py | 59 | # If we don't want to do metal and only use cpu, just set target to be target |
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | python/tvm/topi/nn/pad.py | 28 | Calculates the output shape of a tensor after applying padding. Args: data (tvm.te.Tensor): The input |
| HIGH | python/tvm/tirx/operator/intrinsics/cuda/utils.py | 40 | Validate that cta_group is 1 or 2 and return it as int. Args: cta_group: The cta_group value (can be int or |
| HIGH | python/tvm/tirx/operator/intrinsics/cuda/utils.py | 62 | Validate that value is within range and is a power of two. Args: value: The value to validate min_v |
| HIGH | …irx/operator/tile_primitive/cuda/gemm_async/tcgen05.py | 296 | Schedule an asynchronous GEMM operation using tcgen05.mma (Blackwell Tensor Core). Computes C = A @ B (with optiona |
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | python/tvm/relax/frontend/tflite/tflite_frontend.py | 7611 | # Alternatively you can use tensorflow.keras.application wrapped arounf a tf.function |
| MEDIUM | include/tvm/s_tir/transform.h | 201 | * which is the three blocks as discussed above. |
| MEDIUM | docs/how_to/tutorials/export_and_load_executable.py | 319 | # you can use TVM's RPC mechanism to cross-compile, upload, and run the model |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | …thon/tvm/contrib/hexagon/profiling/process_lwp_data.py | 178 | def process_data(data, func_info, so_ld_addr): |
| LOW | tests/python/relax/test_tvmscript_printer_relax.py | 38 | def test_function(): |
| LOW | tests/python/relax/test_expr_functor.py | 540 | def test_function(): |
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | tests/python/relax/test_tvmscript_parser.py | 748 |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | .claude/scripts/monitor_gpu.sh | 7 | # Usage: |