├── .asf.yaml ├── .clang-format ├── .gitattributes ├── .github ├── CODEOWNERSHIP ├── ISSUE_TEMPLATE │ ├── bug-report.md │ ├── ci-problem.md │ ├── config.yml │ ├── documentation.md │ ├── feature-tracking.md │ └── flaky-test.md ├── actions │ └── setup │ │ └── action.yml ├── dependabot.yml ├── disabled_workflows │ └── pr_comment_bot.yml └── workflows │ ├── cc_bot.yml │ ├── github-command-test.yml │ ├── main.yml │ ├── mlc.yml │ ├── nightly_docker_update.yml │ ├── ping_reviewers.yml │ ├── tag_teams.yml │ ├── tvmbot.yml │ ├── update_last_successful_branch.yml │ ├── update_nightly_branch.yml │ └── upload_ci_resource.yml ├── .gitignore ├── .gitmodules ├── .pre-commit-config.yaml ├── 3rdparty ├── compiler-rt │ └── builtin_fp16.h ├── libcrc │ ├── .gitignore │ ├── include │ │ └── checksum.h │ ├── src │ │ └── crcccitt.c │ └── tab │ │ └── gentab_ccitt.inc ├── mlperftiny │ ├── README.md │ └── api │ │ ├── internally_implemented.cpp │ │ ├── internally_implemented.h │ │ └── submitter_implemented.h ├── mscclpp │ └── include │ │ ├── common.h │ │ ├── msccl.cuh │ │ └── msccl.h ├── nvbench │ └── l2_cache_flush.h ├── picojson │ ├── README.md │ ├── picojson.h │ └── test_picojson.cpp └── tensorrt_llm │ ├── custom_allreduce_kernels.cu │ └── custom_allreduce_kernels.h ├── CMakeLists.txt ├── CONTRIBUTORS.md ├── KEYS ├── LICENSE ├── Makefile ├── NOTICE ├── README.md ├── apps ├── android_rpc │ ├── .gitignore │ ├── README.md │ ├── app │ │ ├── .gitignore │ │ ├── build.gradle │ │ └── src │ │ │ └── main │ │ │ ├── AndroidManifest.xml │ │ │ ├── java │ │ │ └── org │ │ │ │ └── apache │ │ │ │ └── tvm │ │ │ │ └── tvmrpc │ │ │ │ ├── MainActivity.java │ │ │ │ ├── RPCActivity.java │ │ │ │ ├── RPCAndroidWatchdog.java │ │ │ │ └── RPCProcessor.java │ │ │ ├── jni │ │ │ ├── Android.mk │ │ │ ├── Application.mk │ │ │ ├── make │ │ │ │ └── config.mk │ │ │ └── tvm_runtime.h │ │ │ └── res │ │ │ ├── layout │ │ │ ├── activity_main.xml │ │ │ ├── activity_rpc.xml │ │ │ ├── content_main.xml │ │ │ └── content_rpc.xml │ │ │ ├── mipmap-hdpi │ │ │ └── ic_launcher.png │ │ │ ├── mipmap-mdpi │ │ │ └── ic_launcher.png │ │ │ └── values │ │ │ ├── colors.xml │ │ │ ├── strings.xml │ │ │ └── styles.xml │ ├── build.gradle │ ├── dev_tools │ │ ├── gen_keystore.sh │ │ └── sign_apk.sh │ ├── gradle.properties │ ├── settings.gradle │ └── tests │ │ └── android_rpc_test.py ├── cpp_rpc │ ├── CMakeLists.txt │ ├── README.md │ ├── main.cc │ ├── rpc_env.cc │ ├── rpc_env.h │ ├── rpc_server.cc │ ├── rpc_server.h │ ├── rpc_tracker_client.h │ ├── win32_process.cc │ └── win32_process.h ├── hexagon_api │ ├── CMakeLists.txt │ └── README.md ├── hexagon_launcher │ ├── CMakeLists.txt │ ├── README.md │ ├── cmake │ │ ├── HexagonLauncher.cmake │ │ ├── android │ │ │ └── CMakeLists.txt │ │ └── hexagon │ │ │ └── CMakeLists.txt │ ├── launcher_android.cc │ ├── launcher_core.cc │ ├── launcher_core.h │ ├── launcher_hexagon.cc │ ├── launcher_main.cc │ ├── launcher_rpc.idl │ ├── launcher_util.cc │ └── launcher_util.h └── ios_rpc │ ├── .gitignore │ ├── CMakeLists.txt │ ├── README.md │ ├── init_proj.py │ ├── tests │ └── ios_rpc_test.py │ ├── tvmrpc.xcodeproj │ ├── project.pbxproj │ ├── project.xcworkspace │ │ └── contents.xcworkspacedata │ └── xcshareddata │ │ └── xcschemes │ │ └── tvmrpc.xcscheme │ └── tvmrpc │ ├── AppDelegate.h │ ├── AppDelegate.m │ ├── Assets.xcassets │ └── AppIcon.appiconset │ │ └── Contents.json │ ├── Base.lproj │ ├── LaunchScreen.storyboard │ └── Main.storyboard │ ├── Info.plist │ ├── RPCArgs.h │ ├── RPCArgs.mm │ ├── RPCServer.h │ ├── RPCServer.mm │ ├── TVMRuntime.mm │ ├── ViewController.h │ ├── ViewController.mm │ └── main.m ├── ci ├── README.md ├── jenkins │ ├── .gitignore │ ├── README.md │ ├── data.py │ ├── docker-images.ini │ ├── generate.py │ ├── generated │ │ ├── arm_jenkinsfile.groovy │ │ ├── cpu_jenkinsfile.groovy │ │ ├── docker_jenkinsfile.groovy │ │ ├── gpu_jenkinsfile.groovy │ │ ├── hexagon_jenkinsfile.groovy │ │ ├── i386_jenkinsfile.groovy │ │ ├── lint_jenkinsfile.groovy │ │ └── wasm_jenkinsfile.groovy │ ├── mlc_jenkinsfile.groovy │ ├── templates │ │ ├── arm_jenkinsfile.groovy.j2 │ │ ├── cpu_jenkinsfile.groovy.j2 │ │ ├── docker_jenkinsfile.groovy.j2 │ │ ├── gpu_jenkinsfile.groovy.j2 │ │ ├── hexagon_jenkinsfile.groovy.j2 │ │ ├── i386_jenkinsfile.groovy.j2 │ │ ├── lint_jenkinsfile.groovy.j2 │ │ ├── utils │ │ │ ├── Build.groovy.j2 │ │ │ ├── Prepare.groovy.j2 │ │ │ ├── Test.groovy.j2 │ │ │ ├── base.groovy.j2 │ │ │ └── macros.j2 │ │ └── wasm_jenkinsfile.groovy.j2 │ └── unity_jenkinsfile.groovy └── scripts │ ├── github │ ├── __init__.py │ ├── github_cc_reviewers.py │ ├── github_commenter.py │ ├── github_docs_comment.py │ ├── github_pr_comment.py │ ├── github_skipped_tests_comment.py │ ├── github_tag_teams.py │ ├── github_tvmbot.py │ ├── ping_reviewers.py │ └── update_branch.py │ └── jenkins │ ├── check_pr.py │ ├── cmd_utils.py │ ├── determine_docker_images.py │ ├── git_change_docker.sh │ ├── git_change_docs.sh │ ├── git_skip_ci.py │ ├── git_skip_ci_globs.py │ ├── git_utils.py │ ├── http_utils.py │ ├── open_docker_update_pr.py │ ├── pytest_ids.py │ ├── pytest_wrapper.py │ ├── retry.sh │ ├── s3.py │ ├── should_rebuild_docker.py │ └── should_run_slow_tests.py ├── cmake ├── config.cmake ├── libs │ └── Libbacktrace.cmake ├── modules │ ├── CUDA.cmake │ ├── ClangFlags.cmake │ ├── Git.cmake │ ├── Hexagon.cmake │ ├── HexagonSDK.cmake │ ├── LLVM.cmake │ ├── LibInfo.cmake │ ├── Logging.cmake │ ├── Metal.cmake │ ├── OpenCL.cmake │ ├── OpenMP.cmake │ ├── ROCM.cmake │ ├── Vulkan.cmake │ └── contrib │ │ ├── AMX.cmake │ │ ├── ArmComputeLib.cmake │ │ ├── BLAS.cmake │ │ ├── BNNS.cmake │ │ ├── CLML.cmake │ │ ├── CUTLASS.cmake │ │ ├── CoreML.cmake │ │ ├── DNNL.cmake │ │ ├── MSC.cmake │ │ ├── MSCCLPP.cmake │ │ ├── Mrvl.cmake │ │ ├── NNAPI.cmake │ │ ├── PAPI.cmake │ │ ├── Posit.cmake │ │ ├── Random.cmake │ │ ├── Sort.cmake │ │ ├── TFLite.cmake │ │ ├── TensorRT.cmake │ │ └── vllm.cmake └── utils │ ├── CCache.cmake │ ├── CRTConfig.cmake │ ├── FindCUDA.cmake │ ├── FindLLVM.cmake │ ├── FindNCCL.cmake │ ├── FindNVSHMEM.cmake │ ├── FindOpenCL.cmake │ ├── FindRCCL.cmake │ ├── FindROCM.cmake │ ├── FindVulkan.cmake │ ├── Linker.cmake │ ├── Summary.cmake │ └── Utils.cmake ├── conda ├── Dockerfile.template ├── build-environment.yaml ├── build_cpu.sh ├── build_cuda.sh ├── build_win.bat ├── condarc ├── recipe │ ├── bld.bat │ ├── build.sh │ ├── conda_build_config.yaml │ ├── cross-linux.cmake │ ├── install_libtvm.bat │ ├── install_libtvm.sh │ ├── install_tvm_python.bat │ ├── install_tvm_python.sh │ └── meta.yaml └── render_cuda_dockerfiles.py ├── configs ├── host │ └── default.json └── test │ ├── compile_config_test.json │ └── tune_config_test.json ├── conftest.py ├── docker ├── Dockerfile.ci_adreno ├── Dockerfile.ci_arm ├── Dockerfile.ci_cpu ├── Dockerfile.ci_gpu ├── Dockerfile.ci_hexagon ├── Dockerfile.ci_i386 ├── Dockerfile.ci_jekyll ├── Dockerfile.ci_lint ├── Dockerfile.ci_wasm ├── Dockerfile.conda_cpu ├── Dockerfile.conda_cuda100 ├── Dockerfile.conda_cuda90 ├── Dockerfile.docs ├── README.md ├── bash.sh ├── build.sh ├── clear-stale-images.sh ├── dev_common.sh ├── install │ ├── install_tvm_cpu.sh │ ├── install_tvm_gpu.sh │ ├── ubuntu2004_install_core.sh │ ├── ubuntu2004_install_python.sh │ ├── ubuntu2004_install_python_package.sh │ ├── ubuntu2004_install_redis.sh │ ├── ubuntu2204_install_aarch64_cross_compile.sh │ ├── ubuntu2204_install_llvm.sh │ ├── ubuntu2204_manual_install_llvm.sh │ ├── ubuntu2204_manual_install_llvm_cross_aarch64.sh │ ├── ubuntu_download_arm_compute_lib_binaries.sh │ ├── ubuntu_download_csinn2_compute_lib.sh │ ├── ubuntu_download_xuantie_gcc_linux.sh │ ├── ubuntu_download_xuantie_gcc_newlib.sh │ ├── ubuntu_download_xuantie_qemu.sh │ ├── ubuntu_init_zephyr_project.sh │ ├── ubuntu_install_androidsdk.sh │ ├── ubuntu_install_aprofile_aem.sh │ ├── ubuntu_install_arduino.sh │ ├── ubuntu_install_boost.sh │ ├── ubuntu_install_clang_format.sh │ ├── ubuntu_install_cmake_source.sh │ ├── ubuntu_install_conda.sh │ ├── ubuntu_install_core.sh │ ├── ubuntu_install_coreml.sh │ ├── ubuntu_install_dgl.sh │ ├── ubuntu_install_dnnl.sh │ ├── ubuntu_install_emscripten.sh │ ├── ubuntu_install_freertos.sh │ ├── ubuntu_install_gluoncv.sh │ ├── ubuntu_install_golang.sh │ ├── ubuntu_install_googletest.sh │ ├── ubuntu_install_gradle.sh │ ├── ubuntu_install_hexagon.sh │ ├── ubuntu_install_java.sh │ ├── ubuntu_install_jax.sh │ ├── ubuntu_install_libtorch.sh │ ├── ubuntu_install_libxsmm.sh │ ├── ubuntu_install_llvm.sh │ ├── ubuntu_install_llvm_from_source.sh │ ├── ubuntu_install_nnef.sh │ ├── ubuntu_install_nnpack.sh │ ├── ubuntu_install_nodejs.sh │ ├── ubuntu_install_nrfjprog.sh │ ├── ubuntu_install_onnx.sh │ ├── ubuntu_install_opencl.sh │ ├── ubuntu_install_papi.sh │ ├── ubuntu_install_python.sh │ ├── ubuntu_install_python_package.sh │ ├── ubuntu_install_rat.sh │ ├── ubuntu_install_redis.sh │ ├── ubuntu_install_rocm.sh │ ├── ubuntu_install_rust.sh │ ├── ubuntu_install_sbt.sh │ ├── ubuntu_install_sccache.sh │ ├── ubuntu_install_sphinx.sh │ ├── ubuntu_install_spike_sim.sh │ ├── ubuntu_install_tensorflow.sh │ ├── ubuntu_install_tensorflow_aarch64.sh │ ├── ubuntu_install_tflite.sh │ ├── ubuntu_install_universal.sh │ ├── ubuntu_install_vulkan.sh │ ├── ubuntu_install_wasmtime.sh │ ├── ubuntu_install_zephyr.sh │ ├── ubuntu_install_zephyr_sdk.sh │ └── ubuntu_setup_tz.sh ├── lint.sh ├── python │ ├── bootstrap-requirements.txt │ ├── bootstrap │ │ ├── .gitignore │ │ ├── generate.sh │ │ └── lockfiles │ │ │ ├── constraints-3.7.txt │ │ │ ├── constraints-3.8.txt │ │ │ ├── constraints-3.9.txt │ │ │ ├── requirements-3.7.txt │ │ │ ├── requirements-3.8.txt │ │ │ └── requirements-3.9.txt │ └── ci-constraints.txt ├── utils │ └── apt-install-and-clear.sh └── with_the_same_user ├── docs ├── .gitignore ├── Doxyfile ├── Makefile ├── README.md ├── _static │ ├── css │ │ └── tvm_theme.css │ └── img │ │ ├── README │ │ ├── tvm-logo-small.png │ │ └── tvm-logo-square.png ├── arch │ ├── device_target_interactions.rst │ ├── index.rst │ ├── introduction_to_module_serialization.rst │ ├── pass_infra.rst │ ├── runtime.rst │ └── runtimes │ │ └── vulkan.rst ├── conf.py ├── contribute │ ├── ci.rst │ ├── code_guide.rst │ ├── code_review.rst │ ├── committer_guide.rst │ ├── community.rst │ ├── document.rst │ ├── error_handling.rst │ ├── git_howto.rst │ ├── index.rst │ ├── pull_request.rst │ └── release_process.rst ├── deep_dive │ ├── relax │ │ ├── abstraction.rst │ │ ├── index.rst │ │ ├── learning.rst │ │ └── tutorials │ │ │ ├── README.txt │ │ │ ├── relax_creation.py │ │ │ └── relax_transformation.py │ └── tensor_ir │ │ ├── abstraction.rst │ │ ├── index.rst │ │ ├── learning.rst │ │ └── tutorials │ │ ├── README.txt │ │ ├── tir_creation.py │ │ └── tir_transformation.py ├── download_3rdparty_embeds.py ├── errors.rst ├── genindex.rst ├── get_started │ ├── overview.rst │ └── tutorials │ │ ├── README.txt │ │ ├── ir_module.py │ │ └── quick_start.py ├── how_to │ ├── dev │ │ ├── index.rst │ │ ├── pytest_target_parametrization.rst │ │ └── setup_rpc_system.rst │ └── tutorials │ │ ├── README.txt │ │ ├── cross_compilation_and_rpc.py │ │ ├── customize_opt.py │ │ ├── e2e_opt_model.py │ │ ├── export_and_load_executable.py │ │ └── optimize_llm.py ├── index.rst ├── install │ ├── docker.rst │ ├── from_source.rst │ └── index.rst └── reference │ ├── api │ ├── links.rst │ └── python │ │ ├── contrib.rst │ │ ├── dlight.rst │ │ ├── driver.rst │ │ ├── error.rst │ │ ├── index.rst │ │ ├── instrument.rst │ │ ├── ir.rst │ │ ├── meta_schedule.rst │ │ ├── relax │ │ ├── analysis.rst │ │ ├── block_builder.rst │ │ ├── frontend.rst │ │ ├── op.rst │ │ ├── relax.rst │ │ └── transform.rst │ │ ├── rpc.rst │ │ ├── runtime │ │ ├── disco.rst │ │ ├── profiling.rst │ │ ├── runtime.rst │ │ └── vm.rst │ │ ├── target.rst │ │ ├── te.rst │ │ ├── tir │ │ ├── analysis.rst │ │ ├── schedule.rst │ │ ├── stmt_functor.rst │ │ ├── tir.rst │ │ └── transform.rst │ │ ├── topi.rst │ │ └── transform.rst │ ├── publications.rst │ └── security.rst ├── include └── tvm │ ├── arith │ ├── analyzer.h │ ├── bound.h │ ├── int_set.h │ ├── int_solver.h │ ├── iter_affine_map.h │ └── pattern.h │ ├── ir │ ├── analysis.h │ ├── attrs.h │ ├── diagnostic.h │ ├── env_func.h │ ├── expr.h │ ├── function.h │ ├── global_info.h │ ├── global_var_supply.h │ ├── instrument.h │ ├── module.h │ ├── name_supply.h │ ├── op.h │ ├── replace_global_vars.h │ ├── source_map.h │ ├── transform.h │ ├── type.h │ └── type_functor.h │ ├── meta_schedule │ ├── arg_info.h │ ├── builder.h │ ├── cost_model.h │ ├── database.h │ ├── extracted_task.h │ ├── feature_extractor.h │ ├── measure_callback.h │ ├── measure_candidate.h │ ├── mutator.h │ ├── postproc.h │ ├── profiler.h │ ├── runner.h │ ├── schedule │ │ ├── cpu │ │ │ └── .gitignore │ │ ├── cuda │ │ │ └── thread_bind.h │ │ ├── generic │ │ │ └── winograd.h │ │ └── x86 │ │ │ └── .gitignore │ ├── schedule_rule.h │ ├── search_strategy.h │ ├── space_generator.h │ ├── task_scheduler.h │ └── tune_context.h │ ├── node │ ├── attr_registry_map.h │ ├── cast.h │ ├── functor.h │ ├── node.h │ ├── reflection.h │ ├── repr_printer.h │ ├── script_printer.h │ ├── serialization.h │ ├── structural_equal.h │ └── structural_hash.h │ ├── relax │ ├── analysis.h │ ├── attrs │ │ ├── ccl.h │ │ ├── create.h │ │ ├── datatype.h │ │ ├── distributed.h │ │ ├── image.h │ │ ├── index.h │ │ ├── linear_algebra.h │ │ ├── manipulate.h │ │ ├── nn.h │ │ ├── op.h │ │ ├── qdq.h │ │ ├── sampling.h │ │ ├── search.h │ │ ├── sorting.h │ │ ├── statistical.h │ │ └── vision.h │ ├── backend.h │ ├── binding_rewrite.h │ ├── block_builder.h │ ├── dataflow_matcher.h │ ├── dataflow_pattern.h │ ├── dataflow_pattern_functor.h │ ├── distributed │ │ ├── axis_group_graph.h │ │ ├── global_info.h │ │ ├── struct_info.h │ │ └── transform.h │ ├── exec_builder.h │ ├── expr.h │ ├── expr_functor.h │ ├── nested_msg.h │ ├── op_attr_types.h │ ├── struct_info.h │ ├── struct_info_functor.h │ ├── tir_pattern.h │ ├── transform.h │ ├── type.h │ └── utils.h │ ├── runtime │ ├── base.h │ ├── builtin_fp16.h │ ├── c_backend_api.h │ ├── contrib │ │ └── papi.h │ ├── data_type.h │ ├── device_api.h │ ├── disco │ │ ├── builtin.h │ │ ├── cuda_ipc_memory.h │ │ ├── disco_worker.h │ │ └── session.h │ ├── int_tuple.h │ ├── logging.h │ ├── memory │ │ └── memory_manager.h │ ├── module.h │ ├── nvtx.h │ ├── object.h │ ├── packed_func.h │ ├── profiling.h │ ├── serializer.h │ ├── tensor.h │ ├── threading_backend.h │ └── vm │ │ ├── builtin.h │ │ ├── bytecode.h │ │ ├── executable.h │ │ ├── tensor_cache_support.h │ │ └── vm.h │ ├── script │ ├── ir_builder │ │ ├── base.h │ │ ├── ir │ │ │ ├── frame.h │ │ │ └── ir.h │ │ ├── relax │ │ │ ├── frame.h │ │ │ └── ir.h │ │ └── tir │ │ │ ├── frame.h │ │ │ └── ir.h │ └── printer │ │ ├── doc.h │ │ ├── ir_docsifier.h │ │ └── ir_docsifier_functor.h │ ├── support │ ├── parallel_for.h │ ├── random_engine.h │ └── with.h │ ├── target │ ├── codegen.h │ ├── tag.h │ ├── target.h │ ├── target_info.h │ ├── target_kind.h │ └── virtual_device.h │ ├── te │ ├── operation.h │ └── tensor.h │ ├── tir │ ├── analysis.h │ ├── block_dependence_info.h │ ├── block_scope.h │ ├── buffer.h │ ├── builtin.h │ ├── data_layout.h │ ├── data_type_rewriter.h │ ├── expr.h │ ├── expr_functor.h │ ├── function.h │ ├── index_map.h │ ├── op.h │ ├── op_attr_types.h │ ├── schedule │ │ ├── instruction.h │ │ ├── schedule.h │ │ ├── state.h │ │ └── trace.h │ ├── stmt.h │ ├── stmt_functor.h │ ├── transform.h │ ├── utils.h │ └── var.h │ └── topi │ ├── broadcast.h │ ├── contrib │ ├── cublas.h │ └── rocblas.h │ ├── detail │ ├── array_utils.h │ ├── broadcast.h │ ├── constant_utils.h │ ├── extern.h │ ├── fuse.h │ ├── pad_utils.h │ ├── ravel_unravel.h │ ├── strided_slice.h │ └── tensor_utils.h │ ├── einsum.h │ ├── elemwise.h │ ├── nn.h │ ├── nn │ ├── bias_add.h │ ├── bnn.h │ ├── dense.h │ ├── dilate.h │ ├── flatten.h │ ├── group_norm.h │ ├── instance_norm.h │ ├── layer_norm.h │ ├── local_response_norm.h │ ├── mapping.h │ ├── pooling.h │ ├── rms_norm.h │ └── softmax.h │ ├── reduction.h │ ├── tags.h │ ├── transform.h │ ├── utils.h │ └── vision │ └── reorg.h ├── jvm ├── README.md ├── assembly │ ├── linux-x86_64 │ │ ├── pom.xml │ │ └── src │ │ │ └── main │ │ │ └── assembly │ │ │ └── assembly.xml │ ├── osx-x86_64 │ │ ├── pom.xml │ │ └── src │ │ │ └── main │ │ │ └── assembly │ │ │ └── assembly.xml │ └── pom.xml ├── conf │ ├── google_checks.xml │ └── log4j.properties ├── core │ ├── pom.xml │ └── src │ │ ├── main │ │ └── java │ │ │ └── org │ │ │ └── apache │ │ │ └── tvm │ │ │ ├── API.java │ │ │ ├── APIInternal.java │ │ │ ├── Base.java │ │ │ ├── Device.java │ │ │ ├── Function.java │ │ │ ├── LibInfo.java │ │ │ ├── Module.java │ │ │ ├── NativeLibraryLoader.java │ │ │ ├── TVMObject.java │ │ │ ├── TVMType.java │ │ │ ├── TVMValue.java │ │ │ ├── TVMValueBytes.java │ │ │ ├── TVMValueDouble.java │ │ │ ├── TVMValueHandle.java │ │ │ ├── TVMValueLong.java │ │ │ ├── TVMValueNull.java │ │ │ ├── TVMValueString.java │ │ │ ├── Tensor.java │ │ │ ├── TensorBase.java │ │ │ ├── TypeIndex.java │ │ │ └── rpc │ │ │ ├── Client.java │ │ │ ├── ConnectProxyServerProcessor.java │ │ │ ├── ConnectTrackerServerProcessor.java │ │ │ ├── NativeServerLoop.java │ │ │ ├── RPC.java │ │ │ ├── RPCSession.java │ │ │ ├── RPCWatchdog.java │ │ │ ├── Server.java │ │ │ ├── ServerProcessor.java │ │ │ ├── SocketChannel.java │ │ │ ├── StandaloneServerProcessor.java │ │ │ ├── TVMRemoteDevice.java │ │ │ └── Utils.java │ │ └── test │ │ ├── java │ │ └── org │ │ │ └── apache │ │ │ └── tvm │ │ │ ├── FunctionTest.java │ │ │ ├── ModuleTest.java │ │ │ ├── TensorTest.java │ │ │ ├── TestUtils.java │ │ │ └── rpc │ │ │ └── RPCTest.java │ │ └── scripts │ │ ├── prepare_test_libs.py │ │ └── test_rpc_proxy_server.py ├── native │ ├── linux-x86_64 │ │ └── pom.xml │ ├── osx-x86_64 │ │ └── pom.xml │ ├── pom.xml │ └── src │ │ └── main │ │ └── native │ │ ├── jni_helper_func.h │ │ └── org_apache_tvm_native_c_api.cc └── pom.xml ├── licenses ├── LICENSE.blockingconcurrentqueue.txt ├── LICENSE.builtin_fp16.txt ├── LICENSE.cma.txt ├── LICENSE.cnpy.txt ├── LICENSE.concurrentqueue.txt ├── LICENSE.cutlass.txt ├── LICENSE.cutlass_fpA_intB_gemm.txt ├── LICENSE.l2_cache_flush.txt ├── LICENSE.libcrc.txt ├── LICENSE.libflash_attn.txt ├── LICENSE.picojson.txt ├── LICENSE.rang.txt ├── LICENSE.tensorrt_llm.txt ├── LICENSE.vllm.txt └── LICENSE.zlib.txt ├── mypy.ini ├── pyproject.toml ├── python ├── .gitignore ├── gen_requirements.py ├── pyproject.toml └── tvm │ ├── __init__.py │ ├── arith │ ├── __init__.py │ ├── _ffi_api.py │ ├── analyzer.py │ ├── bound.py │ ├── int_set.py │ ├── int_solver.py │ ├── iter_affine_map.py │ └── pattern.py │ ├── base.py │ ├── contrib │ ├── __init__.py │ ├── cblas.py │ ├── cc.py │ ├── clang.py │ ├── coreml_runtime.py │ ├── cublas.py │ ├── cublaslt.py │ ├── cudnn.py │ ├── cutlass │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── attention_operation.py │ │ ├── build.py │ │ ├── conv2d_operation.py │ │ ├── conv2d_profiler.py │ │ ├── gemm_operation.py │ │ ├── gemm_profiler.py │ │ ├── gen_conv2d.py │ │ ├── gen_gemm.py │ │ ├── gen_tensor_op.py │ │ ├── layer_norm_operation.py │ │ ├── library.py │ │ └── rms_norm_operation.py │ ├── dlpack.py │ ├── dnnl.py │ ├── download.py │ ├── emcc.py │ ├── hexagon │ │ ├── __init__.py │ │ ├── _ci_env_check.py │ │ ├── build.py │ │ ├── generate_take_op.py │ │ ├── hexagon_profiler.py │ │ ├── hexagon_unary_ops.py │ │ ├── meta_schedule.py │ │ ├── profiling │ │ │ └── process_lwp_data.py │ │ ├── pytest_plugin.py │ │ ├── session.py │ │ └── tools.py │ ├── hipblas.py │ ├── miopen.py │ ├── mkl.py │ ├── mps.py │ ├── mrvl.py │ ├── msc │ │ ├── __init__.py │ │ ├── core │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ ├── codegen │ │ │ │ ├── __init__.py │ │ │ │ ├── codegen.py │ │ │ │ └── sources.py │ │ │ ├── frontend │ │ │ │ ├── __init__.py │ │ │ │ └── translate.py │ │ │ ├── gym │ │ │ │ ├── __init__.py │ │ │ │ ├── agent │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_agent.py │ │ │ │ │ ├── method.py │ │ │ │ │ └── search_agent.py │ │ │ │ ├── control │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── configer.py │ │ │ │ │ ├── controller.py │ │ │ │ │ ├── namespace.py │ │ │ │ │ ├── service.py │ │ │ │ │ └── worker.py │ │ │ │ ├── environment │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_env.py │ │ │ │ │ ├── method.py │ │ │ │ │ ├── prune_env.py │ │ │ │ │ └── quantize_env.py │ │ │ │ └── namespace.py │ │ │ ├── ir │ │ │ │ ├── __init__.py │ │ │ │ └── graph.py │ │ │ ├── runtime │ │ │ │ ├── __init__.py │ │ │ │ ├── hook.py │ │ │ │ ├── jit.py │ │ │ │ └── runner.py │ │ │ ├── tools │ │ │ │ ├── __init__.py │ │ │ │ ├── configer.py │ │ │ │ ├── distill │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── configer.py │ │ │ │ │ ├── distiller.py │ │ │ │ │ └── method.py │ │ │ │ ├── execute.py │ │ │ │ ├── prune │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── configer.py │ │ │ │ │ ├── method.py │ │ │ │ │ └── pruner.py │ │ │ │ ├── quantize │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── configer.py │ │ │ │ │ ├── method.py │ │ │ │ │ └── quantizer.py │ │ │ │ ├── tool.py │ │ │ │ └── track │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── configer.py │ │ │ │ │ ├── method.py │ │ │ │ │ └── tracker.py │ │ │ ├── transform │ │ │ │ ├── __init__.py │ │ │ │ ├── pattern.py │ │ │ │ └── transform.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── arguments.py │ │ │ │ ├── dataset.py │ │ │ │ ├── expr.py │ │ │ │ ├── file.py │ │ │ │ ├── info.py │ │ │ │ ├── log.py │ │ │ │ ├── message.py │ │ │ │ ├── namespace.py │ │ │ │ └── register.py │ │ ├── framework │ │ │ ├── __init__.py │ │ │ ├── tensorflow │ │ │ │ ├── __init__.py │ │ │ │ ├── _ffi_api.py │ │ │ │ ├── codegen │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── codegen.py │ │ │ │ ├── frontend │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── translate.py │ │ │ │ ├── runtime │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── runner.py │ │ │ │ └── tools │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── distill │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── distiller.py │ │ │ │ │ ├── prune │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── pruner.py │ │ │ │ │ ├── quantize │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── quantizer.py │ │ │ │ │ └── track │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── tracker.py │ │ │ ├── tensorrt │ │ │ │ ├── __init__.py │ │ │ │ ├── _ffi_api.py │ │ │ │ ├── codegen │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── codegen.py │ │ │ │ │ ├── sources.py │ │ │ │ │ └── utils.py │ │ │ │ ├── frontend │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── translate.py │ │ │ │ ├── runtime │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── runner.py │ │ │ │ ├── tools │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── distill │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── distiller.py │ │ │ │ │ ├── prune │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── pruner.py │ │ │ │ │ ├── quantize │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── method.py │ │ │ │ │ │ └── quantizer.py │ │ │ │ │ └── track │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── tracker.py │ │ │ │ └── transform │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── pattern.py │ │ │ │ │ └── transform.py │ │ │ ├── torch │ │ │ │ ├── __init__.py │ │ │ │ ├── _ffi_api.py │ │ │ │ ├── codegen │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── codegen.py │ │ │ │ ├── frontend │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── translate.py │ │ │ │ ├── runtime │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── jit.py │ │ │ │ │ └── runner.py │ │ │ │ └── tools │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── distill │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── distiller.py │ │ │ │ │ └── method.py │ │ │ │ │ ├── prune │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── pruner.py │ │ │ │ │ ├── quantize │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── method.py │ │ │ │ │ └── quantizer.py │ │ │ │ │ └── track │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── tracker.py │ │ │ └── tvm │ │ │ │ ├── __init__.py │ │ │ │ ├── _ffi_api.py │ │ │ │ ├── codegen │ │ │ │ ├── __init__.py │ │ │ │ └── codegen.py │ │ │ │ ├── runtime │ │ │ │ ├── __init__.py │ │ │ │ └── runner.py │ │ │ │ └── tools │ │ │ │ ├── __init__.py │ │ │ │ ├── distill │ │ │ │ ├── __init__.py │ │ │ │ └── distiller.py │ │ │ │ ├── prune │ │ │ │ ├── __init__.py │ │ │ │ └── pruner.py │ │ │ │ ├── quantize │ │ │ │ ├── __init__.py │ │ │ │ ├── method.py │ │ │ │ └── quantizer.py │ │ │ │ └── track │ │ │ │ ├── __init__.py │ │ │ │ └── tracker.py │ │ ├── pipeline │ │ │ ├── __init__.py │ │ │ ├── dynamic.py │ │ │ ├── manager.py │ │ │ ├── pipeline.py │ │ │ ├── utils.py │ │ │ ├── worker.py │ │ │ └── wrapper.py │ │ └── plugin │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ ├── build.py │ │ │ ├── codegen │ │ │ ├── __init__.py │ │ │ ├── codegen.py │ │ │ └── sources.py │ │ │ ├── op │ │ │ ├── __init__.py │ │ │ └── _ffi_api.py │ │ │ ├── register.py │ │ │ └── utils.py │ ├── ndk.py │ ├── nnpack.py │ ├── nvcc.py │ ├── pickle_memoize.py │ ├── popen_pool.py │ ├── random.py │ ├── rocblas.py │ ├── rocm.py │ ├── rpc.py │ ├── spirv.py │ ├── tar.py │ ├── tflite_runtime.py │ ├── thrust.py │ ├── tvmjs.py │ ├── utils.py │ └── xcode.py │ ├── dlight │ ├── __init__.py │ ├── analysis │ │ ├── __init__.py │ │ ├── common_analysis.py │ │ └── gemv.py │ ├── base │ │ ├── __init__.py │ │ ├── common_schedules.py │ │ ├── schedule_rule.py │ │ ├── transform.py │ │ └── utils.py │ ├── benchmark │ │ ├── __init__.py │ │ ├── bench.py │ │ ├── extract.py │ │ └── utils.py │ ├── cpu │ │ ├── __init__.py │ │ ├── base.py │ │ └── gemv.py │ └── gpu │ │ ├── __init__.py │ │ ├── base.py │ │ ├── fallback.py │ │ ├── gemv.py │ │ ├── general_reduction.py │ │ ├── low_batch_gemv.py │ │ ├── matmul.py │ │ ├── reduction.py │ │ ├── rmsnorm.py │ │ └── transpose.py │ ├── driver │ ├── __init__.py │ ├── _ffi_api.py │ └── build_module.py │ ├── error.py │ ├── exec │ ├── __init__.py │ ├── disco_remote_socket_session.py │ ├── disco_worker.py │ ├── gpu_memory_bandwidth.py │ ├── popen_worker.py │ ├── query_rpc_tracker.py │ ├── rpc_proxy.py │ ├── rpc_server.py │ └── rpc_tracker.py │ ├── ffi.py │ ├── ir │ ├── __init__.py │ ├── _ffi_analysis_api.py │ ├── _ffi_api.py │ ├── _ffi_instrument_api.py │ ├── _ffi_transform_api.py │ ├── analysis.py │ ├── attrs.py │ ├── base.py │ ├── container.py │ ├── diagnostics │ │ ├── __init__.py │ │ └── _ffi_api.py │ ├── expr.py │ ├── function.py │ ├── global_info.py │ ├── instrument.py │ ├── json_compact.py │ ├── module.py │ ├── op.py │ ├── supply.py │ ├── transform.py │ ├── type.py │ └── type_relation.py │ ├── libinfo.py │ ├── meta_schedule │ ├── __init__.py │ ├── _ffi_api.py │ ├── arg_info.py │ ├── builder │ │ ├── __init__.py │ │ ├── builder.py │ │ └── local_builder.py │ ├── cost_model │ │ ├── __init__.py │ │ ├── cost_model.py │ │ ├── metric.py │ │ ├── mlp_model.py │ │ ├── random_model.py │ │ └── xgb_model.py │ ├── database │ │ ├── __init__.py │ │ ├── database.py │ │ ├── json_database.py │ │ ├── memory_database.py │ │ ├── ordered_union_database.py │ │ ├── schedule_fn_database.py │ │ └── union_database.py │ ├── extracted_task.py │ ├── feature_extractor │ │ ├── __init__.py │ │ ├── feature_extractor.py │ │ ├── per_store_feature.py │ │ └── random_feature_extractor.py │ ├── logging.py │ ├── measure_callback │ │ ├── __init__.py │ │ ├── add_to_database.py │ │ ├── measure_callback.py │ │ ├── remove_build_artifact.py │ │ └── update_cost_model.py │ ├── mutator │ │ ├── __init__.py │ │ ├── mutate_compute_location.py │ │ ├── mutate_parallel.py │ │ ├── mutate_thread_binding.py │ │ ├── mutate_tile_size.py │ │ ├── mutate_unroll.py │ │ └── mutator.py │ ├── post_optimization │ │ ├── __init__.py │ │ ├── droplet.py │ │ ├── post_opt.py │ │ ├── space.py │ │ └── utils.py │ ├── postproc │ │ ├── __init__.py │ │ ├── disallow_async_strided_mem_copy.py │ │ ├── disallow_dynamic_loop.py │ │ ├── postproc.py │ │ ├── rewrite_cooperative_fetch.py │ │ ├── rewrite_layout.py │ │ ├── rewrite_parallel_vectorize_unroll.py │ │ ├── rewrite_reduction_block.py │ │ ├── rewrite_tensorize.py │ │ ├── rewrite_unbound_block.py │ │ ├── verify_gpu_code.py │ │ └── verify_vtcm_limit.py │ ├── profiler.py │ ├── relax_integration.py │ ├── runner │ │ ├── __init__.py │ │ ├── config.py │ │ ├── local_runner.py │ │ ├── rpc_runner.py │ │ ├── runner.py │ │ └── utils.py │ ├── schedule │ │ ├── __init__.py │ │ ├── cpu │ │ │ └── __init__.py │ │ ├── cuda │ │ │ ├── __init__.py │ │ │ └── layout_transform.py │ │ ├── generic │ │ │ └── __init__.py │ │ └── x86 │ │ │ └── __init__.py │ ├── schedule_rule │ │ ├── __init__.py │ │ ├── add_rfactor.py │ │ ├── apply_custom_rule.py │ │ ├── auto_bind.py │ │ ├── auto_inline.py │ │ ├── cross_thread_reduction.py │ │ ├── multi_level_tiling.py │ │ ├── parallel_vectorize_unroll.py │ │ ├── random_compute_location.py │ │ └── schedule_rule.py │ ├── search_strategy │ │ ├── __init__.py │ │ ├── evolutionary_search.py │ │ ├── replay_func.py │ │ ├── replay_trace.py │ │ └── search_strategy.py │ ├── space_generator │ │ ├── __init__.py │ │ ├── post_order_apply.py │ │ ├── schedule_fn.py │ │ ├── space_generator.py │ │ └── space_generator_union.py │ ├── task_scheduler │ │ ├── __init__.py │ │ ├── gradient_based.py │ │ ├── round_robin.py │ │ └── task_scheduler.py │ ├── testing │ │ ├── __init__.py │ │ ├── custom_builder_runner.py │ │ ├── distributed_measure_candidates.py │ │ ├── dummy_object.py │ │ ├── local_rpc.py │ │ ├── space_generation.py │ │ ├── te_workload.py │ │ ├── tune_te.py │ │ ├── tune_utils.py │ │ └── validate_database.py │ ├── tir_integration.py │ ├── trace_apply.py │ ├── tune.py │ ├── tune_context.py │ └── utils.py │ ├── relax │ ├── __init__.py │ ├── _ffi_api.py │ ├── analysis │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── analysis.py │ │ └── estimate_memory_usage.py │ ├── backend │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── adreno │ │ │ ├── __init__.py │ │ │ ├── clml.py │ │ │ └── pipeline.py │ │ ├── contrib │ │ │ ├── __init__.py │ │ │ └── nnapi.py │ │ ├── cpu_generic │ │ │ ├── __init__.py │ │ │ └── pipeline.py │ │ ├── cuda │ │ │ ├── __init__.py │ │ │ ├── cublas.py │ │ │ ├── cudnn.py │ │ │ ├── cutlass.py │ │ │ ├── flashinfer.py │ │ │ └── pipeline.py │ │ ├── dispatch_sampling.py │ │ ├── dispatch_sort_scan.py │ │ ├── gpu_generic │ │ │ ├── __init__.py │ │ │ ├── cumsum.py │ │ │ ├── pipeline.py │ │ │ └── sampling.py │ │ ├── metal │ │ │ ├── __init__.py │ │ │ └── coreml.py │ │ ├── pattern_registry.py │ │ ├── patterns.py │ │ ├── rocm │ │ │ ├── __init__.py │ │ │ ├── hipblas.py │ │ │ └── pipeline.py │ │ └── utils.py │ ├── base_py_module.py │ ├── binding_rewrite.py │ ├── block_builder.py │ ├── distributed │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── global_info.py │ │ ├── struct_info.py │ │ └── transform │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ └── transform.py │ ├── dpl │ │ ├── __init__.py │ │ ├── _ffi.py │ │ ├── context.py │ │ ├── pattern.py │ │ └── rewrite.py │ ├── exec_builder.py │ ├── expr.py │ ├── expr_functor.py │ ├── frontend │ │ ├── __init__.py │ │ ├── common.py │ │ ├── nn │ │ │ ├── __init__.py │ │ │ ├── _tensor_op.py │ │ │ ├── core.py │ │ │ ├── exporter.py │ │ │ ├── extern.py │ │ │ ├── llm │ │ │ │ ├── __init__.py │ │ │ │ ├── kv_cache.py │ │ │ │ ├── position_embedding.py │ │ │ │ └── tree_attn.py │ │ │ ├── modules.py │ │ │ ├── op.py │ │ │ ├── spec.py │ │ │ ├── subroutine.py │ │ │ ├── torch.py │ │ │ └── visitor.py │ │ ├── onnx │ │ │ ├── __init__.py │ │ │ └── onnx_frontend.py │ │ ├── stablehlo │ │ │ ├── __init__.py │ │ │ └── stablehlo_translator.py │ │ └── torch │ │ │ ├── __init__.py │ │ │ ├── base_fx_graph_translator.py │ │ │ ├── dynamo.py │ │ │ ├── exported_program_translator.py │ │ │ └── fx_translator.py │ ├── ir │ │ └── instrument.py │ ├── op │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── _op_gradient.py │ │ ├── base.py │ │ ├── binary.py │ │ ├── builtin │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ └── builtin.py │ │ ├── ccl │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ └── ccl.py │ │ ├── create.py │ │ ├── datatype.py │ │ ├── distributed │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ └── distributed.py │ │ ├── grad │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ └── grad.py │ │ ├── image │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ └── image.py │ │ ├── index.py │ │ ├── linear_algebra.py │ │ ├── manipulate.py │ │ ├── mask.py │ │ ├── memory │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ ├── memory.py │ │ │ └── view.py │ │ ├── nn │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ └── nn.py │ │ ├── op_attrs.py │ │ ├── qdq.py │ │ ├── sampling.py │ │ ├── search.py │ │ ├── set.py │ │ ├── sorting.py │ │ ├── statistical.py │ │ ├── ternary.py │ │ ├── unary.py │ │ ├── vision │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ └── nms.py │ │ └── vm │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ └── vm.py │ ├── pipeline.py │ ├── relax_to_pyfunc_converter.py │ ├── struct_info.py │ ├── testing │ │ ├── __init__.py │ │ ├── ast_printer.py │ │ ├── attention.py │ │ ├── lib_comparator.py │ │ ├── matmul.py │ │ ├── nn.py │ │ ├── runtime_builtin.py │ │ ├── transform.py │ │ └── vm.py │ ├── training │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── loss.py │ │ ├── optimizer.py │ │ ├── setup_trainer.py │ │ ├── trainer.py │ │ └── utils.py │ ├── transform │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── attach_external_modules.py │ │ ├── fast_math.py │ │ ├── fold_batch_norm_to_conv2d_for_inference.py │ │ ├── fuse_transpose_matmul.py │ │ ├── ipc_allreduce_rewrite.py │ │ ├── lazy_transform_params.py │ │ ├── legalize_ops │ │ │ ├── __init__.py │ │ │ ├── binary.py │ │ │ ├── ccl.py │ │ │ ├── common.py │ │ │ ├── create.py │ │ │ ├── datatype.py │ │ │ ├── distributed.py │ │ │ ├── grad.py │ │ │ ├── image.py │ │ │ ├── index.py │ │ │ ├── inspect_op.py │ │ │ ├── linear_algebra.py │ │ │ ├── manipulate.py │ │ │ ├── nn.py │ │ │ ├── qdq.py │ │ │ ├── search.py │ │ │ ├── statistical.py │ │ │ ├── unary.py │ │ │ └── vision.py │ │ ├── lower_gpu_ipc_alloc_storage.py │ │ ├── optimize_layout_transform.py │ │ ├── remove_redundant_reshape.py │ │ └── transform.py │ ├── ty.py │ ├── type_converter.py │ ├── utils.py │ └── vm_build.py │ ├── rpc │ ├── __init__.py │ ├── _ffi_api.py │ ├── base.py │ ├── client.py │ ├── minrpc.py │ ├── proxy.py │ ├── server.py │ ├── server_ios_launcher.py │ ├── testing.py │ ├── tornado_util.py │ └── tracker.py │ ├── runtime │ ├── __init__.py │ ├── _ffi_api.py │ ├── _ffi_node_api.py │ ├── _tensor.py │ ├── container.py │ ├── device.py │ ├── disco │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── process_pool.py │ │ └── session.py │ ├── executable.py │ ├── module.py │ ├── object.py │ ├── object_generic.py │ ├── packed_func.py │ ├── params.py │ ├── profiling │ │ ├── __init__.py │ │ └── _ffi_api.py │ ├── script_printer.py │ ├── support.py │ └── vm.py │ ├── script │ ├── __init__.py │ ├── _ffi_api.py │ ├── highlight.py │ ├── ir_builder │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── base.py │ │ ├── ir │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ ├── frame.py │ │ │ └── ir.py │ │ ├── relax │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ ├── distributed │ │ │ │ ├── __init__.py │ │ │ │ ├── _ffi_api.py │ │ │ │ └── ir.py │ │ │ ├── frame.py │ │ │ └── ir.py │ │ └── tir │ │ │ ├── __init__.py │ │ │ ├── _ffi_api.py │ │ │ ├── external_kernel.py │ │ │ ├── frame.py │ │ │ ├── ir.py │ │ │ └── triton.py │ ├── parser │ │ ├── __init__.py │ │ ├── _core.py │ │ ├── core │ │ │ ├── __init__.py │ │ │ ├── diagnostics.py │ │ │ ├── dispatch.py │ │ │ ├── doc.py │ │ │ ├── doc_core.py │ │ │ ├── entry.py │ │ │ ├── error.py │ │ │ ├── evaluator.py │ │ │ ├── parser.py │ │ │ └── utils.py │ │ ├── ir │ │ │ ├── __init__.py │ │ │ ├── entry.py │ │ │ └── parser.py │ │ ├── relax │ │ │ ├── __init__.py │ │ │ ├── dist.py │ │ │ ├── entry.py │ │ │ └── parser.py │ │ └── tir │ │ │ ├── __init__.py │ │ │ ├── entry.py │ │ │ ├── operation.py │ │ │ └── parser.py │ ├── printer │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── doc.py │ │ └── doc_printer.py │ ├── relax.py │ └── tir.py │ ├── support.py │ ├── target │ ├── __init__.py │ ├── _ffi_api.py │ ├── codegen.py │ ├── compilation_config.py │ ├── datatype.py │ ├── detect_target.py │ ├── intrin.py │ ├── tag.py │ ├── target.py │ ├── virtual_device.py │ └── x86.py │ ├── te │ ├── __init__.py │ ├── _ffi_api.py │ ├── operation.py │ ├── tag.py │ └── tensor.py │ ├── testing │ ├── __init__.py │ ├── _ffi_api.py │ ├── attrs.py │ ├── plugin.py │ ├── popen_pool.py │ ├── runner.py │ ├── tir.py │ └── utils.py │ ├── tir │ ├── __init__.py │ ├── _ffi_api.py │ ├── analysis │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ └── analysis.py │ ├── block_dependence_info.py │ ├── block_scope.py │ ├── buffer.py │ ├── build.py │ ├── data_layout.py │ ├── expr.py │ ├── function.py │ ├── functor.py │ ├── generic.py │ ├── ir_builder.py │ ├── op.py │ ├── pipeline.py │ ├── schedule │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── _type_checker.py │ │ ├── analysis.py │ │ ├── instruction.py │ │ ├── schedule.py │ │ ├── state.py │ │ ├── testing.py │ │ ├── trace.py │ │ └── transform.py │ ├── stmt.py │ ├── stmt_functor.py │ ├── tensor_intrin │ │ ├── __init__.py │ │ ├── arm_cpu.py │ │ ├── cuda.py │ │ ├── dot_product_common.py │ │ ├── hexagon.py │ │ ├── metal.py │ │ ├── riscv_cpu.py │ │ ├── rocm.py │ │ └── x86.py │ └── transform │ │ ├── __init__.py │ │ ├── _ffi_api.py │ │ ├── function_pass.py │ │ └── transform.py │ └── topi │ ├── __init__.py │ ├── broadcast.py │ ├── cpp │ ├── __init__.py │ ├── cuda.py │ ├── generic.py │ ├── impl.py │ ├── nn.py │ ├── rocm.py │ ├── utils.py │ ├── vision │ │ ├── __init__.py │ │ └── yolo.py │ └── x86.py │ ├── einsum.py │ ├── generic_op_impl.py │ ├── gpu │ ├── __init__.py │ ├── scan.py │ └── sort.py │ ├── image │ ├── __init__.py │ ├── dilation2d.py │ ├── grid_sample.py │ └── resize.py │ ├── index_put.py │ ├── math.py │ ├── nn │ ├── __init__.py │ ├── batch_matmul.py │ ├── batch_norm.py │ ├── batch_to_space_nd.py │ ├── bitserial_conv2d.py │ ├── bitserial_dense.py │ ├── bitserial_util.py │ ├── bnn.py │ ├── conv1d.py │ ├── conv1d_transpose.py │ ├── conv2d.py │ ├── conv2d_transpose.py │ ├── conv3d.py │ ├── conv3d_transpose.py │ ├── correlation.py │ ├── deformable_conv2d.py │ ├── dense.py │ ├── depth_to_space.py │ ├── depthwise_conv2d.py │ ├── dilate.py │ ├── elemwise.py │ ├── fifo_buffer.py │ ├── flatten.py │ ├── group_norm.py │ ├── instance_norm.py │ ├── layer_norm.py │ ├── local_response_norm.py │ ├── loss.py │ ├── lstm.py │ ├── mapping.py │ ├── pad.py │ ├── pixel_shuffle.py │ ├── pooling.py │ ├── qnn.py │ ├── rms_norm.py │ ├── softmax.py │ ├── space_to_batch_nd.py │ ├── space_to_depth.py │ ├── upsampling.py │ ├── utils.py │ └── winograd_util.py │ ├── reduction.py │ ├── scan.py │ ├── scatter.py │ ├── scatter_elements.py │ ├── searchsorted.py │ ├── signal.py │ ├── slice_scatter.py │ ├── sort.py │ ├── sparse_reshape.py │ ├── tag.py │ ├── tensor.py │ ├── testing │ ├── __init__.py │ ├── adaptive_pool_python.py │ ├── attention_python.py │ ├── batch_matmul.py │ ├── batch_norm.py │ ├── batch_to_space_nd.py │ ├── common.py │ ├── conv1d_ncw_python.py │ ├── conv1d_transpose_ncw_python.py │ ├── conv2d_backcward_weight_python.py │ ├── conv2d_hwcn_python.py │ ├── conv2d_nchw_python.py │ ├── conv2d_nhwc_python.py │ ├── conv2d_transpose_python.py │ ├── conv3d_ncdhw_python.py │ ├── conv3d_ndhwc_python.py │ ├── conv3d_transpose_ncdhw_python.py │ ├── correlation_nchw_python.py │ ├── crop_and_resize_python.py │ ├── deformable_conv2d_python.py │ ├── dense.py │ ├── depth_to_space.py │ ├── depthwise_conv2d_python.py │ ├── dilate_python.py │ ├── gather_nd_python.py │ ├── gather_python.py │ ├── grid_sample_python.py │ ├── group_norm_python.py │ ├── instance_norm_python.py │ ├── l2_normalize_python.py │ ├── layer_norm_python.py │ ├── lrn_python.py │ ├── lstm_python.py │ ├── matrix_set_diag.py │ ├── nll_loss.py │ ├── one_hot.py │ ├── pool_grad_python.py │ ├── poolnd_python.py │ ├── reorg_python.py │ ├── resize_python.py │ ├── rms_norm_python.py │ ├── roi_align_python.py │ ├── roi_pool_python.py │ ├── searchsorted.py │ ├── sequence_mask_python.py │ ├── slice_axis_python.py │ ├── softmax_python.py │ ├── space_to_batch_nd.py │ ├── space_to_depth.py │ └── strided_slice_python.py │ ├── transform.py │ ├── unique.py │ ├── utils.py │ └── vision │ ├── __init__.py │ ├── nms.py │ └── nms_util.py ├── src ├── arith │ ├── analyzer.cc │ ├── bound_deducer.cc │ ├── canonical_simplify.cc │ ├── conjunctive_normal_form.cc │ ├── conjunctive_normal_form.h │ ├── const_fold.h │ ├── const_int_bound.cc │ ├── constraint_extract.cc │ ├── constraint_extract.h │ ├── detect_common_subexpr.cc │ ├── detect_linear_equation.cc │ ├── domain_touched.cc │ ├── int_constraints.cc │ ├── int_operator.h │ ├── int_set.cc │ ├── interval_set.h │ ├── ir_mutator_with_analyzer.cc │ ├── ir_mutator_with_analyzer.h │ ├── ir_visitor_with_analyzer.cc │ ├── ir_visitor_with_analyzer.h │ ├── iter_affine_map.cc │ ├── modular_set.cc │ ├── narrow_predicate_expression.cc │ ├── narrow_predicate_expression.h │ ├── pattern_match.h │ ├── presburger_set.cc │ ├── presburger_set.h │ ├── product_normal_form.h │ ├── rewrite_simplify.cc │ ├── rewrite_simplify.h │ ├── scalable_expression.cc │ ├── scalable_expression.h │ ├── solve_linear_equation.cc │ ├── solve_linear_inequality.cc │ ├── transitive_comparison_analyzer.cc │ ├── unwrap_vector_expr.cc │ └── unwrap_vector_expr.h ├── contrib │ ├── msc │ │ ├── core │ │ │ ├── codegen │ │ │ │ ├── base_codegen.h │ │ │ │ ├── code_stack.cc │ │ │ │ ├── code_stack.h │ │ │ │ ├── codegen_json.cc │ │ │ │ ├── codegen_json.h │ │ │ │ ├── codegen_utils.cc │ │ │ │ ├── codegen_utils.h │ │ │ │ ├── cpp_codegen.h │ │ │ │ └── py_codegen.h │ │ │ ├── ir │ │ │ │ ├── graph.cc │ │ │ │ ├── graph.h │ │ │ │ ├── graph_builder.cc │ │ │ │ ├── graph_builder.h │ │ │ │ ├── plugin.cc │ │ │ │ └── plugin.h │ │ │ ├── printer │ │ │ │ ├── cpp_printer.cc │ │ │ │ ├── cpp_printer.h │ │ │ │ ├── msc_base_printer.cc │ │ │ │ ├── msc_base_printer.h │ │ │ │ ├── msc_doc.cc │ │ │ │ ├── msc_doc.h │ │ │ │ ├── print_utils.cc │ │ │ │ ├── print_utils.h │ │ │ │ ├── prototxt_printer.cc │ │ │ │ ├── prototxt_printer.h │ │ │ │ ├── python_printer.cc │ │ │ │ └── python_printer.h │ │ │ ├── transform │ │ │ │ ├── bind_named_params.cc │ │ │ │ ├── bind_shape.cc │ │ │ │ ├── fuse_tuple.cc │ │ │ │ ├── inline_params.cc │ │ │ │ ├── layout_utils.cc │ │ │ │ ├── layout_utils.h │ │ │ │ ├── rewrite_utils.cc │ │ │ │ ├── rewrite_utils.h │ │ │ │ ├── set_byoc_attrs.cc │ │ │ │ ├── set_expr_layout.cc │ │ │ │ └── set_expr_name.cc │ │ │ ├── utils.cc │ │ │ └── utils.h │ │ ├── framework │ │ │ ├── tensorflow │ │ │ │ ├── codegen.cc │ │ │ │ ├── codegen.h │ │ │ │ ├── codegen_utils.h │ │ │ │ ├── config.h │ │ │ │ ├── tf_v1_opcode.cc │ │ │ │ └── tf_v1_opcode.h │ │ │ ├── tensorrt │ │ │ │ ├── codegen.cc │ │ │ │ ├── codegen.h │ │ │ │ ├── codegen_utils.h │ │ │ │ ├── tensorrt_opcode.cc │ │ │ │ ├── tensorrt_opcode.h │ │ │ │ └── transform_tensorrt.cc │ │ │ ├── torch │ │ │ │ ├── codegen.cc │ │ │ │ ├── codegen.h │ │ │ │ ├── codegen_utils.h │ │ │ │ ├── config.h │ │ │ │ ├── torch_opcode.cc │ │ │ │ └── torch_opcode.h │ │ │ └── tvm │ │ │ │ ├── codegen.cc │ │ │ │ ├── codegen.h │ │ │ │ ├── codegen_utils.h │ │ │ │ ├── config.h │ │ │ │ ├── relax_opcode.cc │ │ │ │ └── relax_opcode.h │ │ └── plugin │ │ │ ├── base_codegen.h │ │ │ ├── codegen_utils.h │ │ │ ├── tensorrt_codegen.cc │ │ │ ├── tensorrt_codegen.h │ │ │ ├── torch_codegen.cc │ │ │ ├── torch_codegen.h │ │ │ ├── tvm_codegen.cc │ │ │ └── tvm_codegen.h │ └── rust_extension.cc ├── ir │ ├── analysis.cc │ ├── apply_pass_to_function.cc │ ├── attr_functor.h │ ├── attrs.cc │ ├── diagnostic.cc │ ├── env_func.cc │ ├── expr.cc │ ├── function.cc │ ├── global_info.cc │ ├── global_var_supply.cc │ ├── instrument.cc │ ├── module.cc │ ├── name_supply.cc │ ├── op.cc │ ├── replace_global_vars.cc │ ├── source_map.cc │ ├── transform.cc │ ├── type.cc │ └── type_functor.cc ├── meta_schedule │ ├── arg_info.cc │ ├── builder │ │ └── builder.cc │ ├── cost_model │ │ └── cost_model.cc │ ├── database │ │ ├── database.cc │ │ ├── database_utils.cc │ │ ├── json_database.cc │ │ ├── memory_database.cc │ │ ├── ordered_union_database.cc │ │ ├── schedule_fn_database.cc │ │ └── union_database.cc │ ├── extracted_task.cc │ ├── feature_extractor │ │ ├── feature_extractor.cc │ │ └── per_store_feature.cc │ ├── measure_callback │ │ ├── add_to_database.cc │ │ ├── measure_callback.cc │ │ ├── remove_build_artifact.cc │ │ └── update_cost_model.cc │ ├── module_equality.cc │ ├── module_equality.h │ ├── mutator │ │ ├── mutate_compute_location.cc │ │ ├── mutate_parallel.cc │ │ ├── mutate_thread_binding.cc │ │ ├── mutate_tile_size.cc │ │ ├── mutate_unroll.cc │ │ └── mutator.cc │ ├── postproc │ │ ├── disallow_async_strided_mem_copy.cc │ │ ├── disallow_dynamic_loop.cc │ │ ├── postproc.cc │ │ ├── rewrite_cooperative_fetch.cc │ │ ├── rewrite_layout.cc │ │ ├── rewrite_parallel_vectorize_unroll.cc │ │ ├── rewrite_reduction_block.cc │ │ ├── rewrite_tensorize.cc │ │ ├── rewrite_unbound_block.cc │ │ ├── verify_gpu_code.cc │ │ └── verify_vtcm_limit.cc │ ├── profiler.cc │ ├── runner │ │ └── runner.cc │ ├── schedule │ │ ├── cpu │ │ │ └── winograd.cc │ │ ├── cuda │ │ │ ├── thread_bind.cc │ │ │ └── winograd.cc │ │ ├── generic │ │ │ └── winograd.cc │ │ └── x86 │ │ │ └── .gitignore │ ├── schedule_rule │ │ ├── add_rfactor.cc │ │ ├── apply_custom_rule.cc │ │ ├── auto_bind.cc │ │ ├── auto_inline.cc │ │ ├── cross_thread_reduction.cc │ │ ├── multi_level_tiling.cc │ │ ├── multi_level_tiling.h │ │ ├── multi_level_tiling_tensor_core.cc │ │ ├── multi_level_tiling_wide_vector.cc │ │ ├── multi_level_tiling_with_intrin.cc │ │ ├── parallel_vectorize_unroll.cc │ │ ├── random_compute_location.cc │ │ └── schedule_rule.cc │ ├── search_strategy │ │ ├── evolutionary_search.cc │ │ ├── replay_func.cc │ │ ├── replay_trace.cc │ │ └── search_strategy.cc │ ├── space_generator │ │ ├── post_order_apply.cc │ │ ├── schedule_fn.cc │ │ ├── space_generator.cc │ │ └── space_generator_union.cc │ ├── task_scheduler │ │ ├── gradient_based.cc │ │ ├── round_robin.cc │ │ └── task_scheduler.cc │ ├── trace_apply.cc │ ├── trace_apply.h │ ├── tune_context.cc │ └── utils.h ├── node │ ├── attr_registry.h │ ├── container_printing.cc │ ├── reflection.cc │ ├── repr_printer.cc │ ├── script_printer.cc │ ├── serialization.cc │ ├── structural_equal.cc │ └── structural_hash.cc ├── relax │ ├── analysis │ │ ├── analysis.cc │ │ ├── collect_call_map.cc │ │ ├── computable_at_compile_time.cc │ │ ├── detect_recursion.cc │ │ ├── graph_partitioner.cc │ │ ├── graph_partitioner.h │ │ ├── layout_transformation.cc │ │ ├── shape_analysis.cc │ │ ├── struct_info_analysis.cc │ │ ├── tir_op_pattern_kind.cc │ │ ├── udchain.cc │ │ ├── var2value.cc │ │ └── well_formed.cc │ ├── backend │ │ ├── contrib │ │ │ ├── clml │ │ │ │ └── codegen.cc │ │ │ ├── codegen_c │ │ │ │ └── codegen_c.h │ │ │ ├── codegen_json │ │ │ │ └── codegen_json.h │ │ │ ├── cublas │ │ │ │ └── codegen.cc │ │ │ ├── cudnn │ │ │ │ └── codegen.cc │ │ │ ├── cutlass │ │ │ │ └── codegen.cc │ │ │ ├── dnnl │ │ │ │ └── codegen.cc │ │ │ ├── hipblas │ │ │ │ └── codegen.cc │ │ │ ├── nnapi │ │ │ │ └── codegen.cc │ │ │ ├── tensorrt │ │ │ │ └── codegen.cc │ │ │ ├── utils.cc │ │ │ └── utils.h │ │ ├── pattern_registry.cc │ │ ├── pattern_registry.h │ │ ├── task_extraction.cc │ │ └── vm │ │ │ ├── codegen_vm.cc │ │ │ ├── codegen_vm_tir.cc │ │ │ ├── exec_builder.cc │ │ │ ├── lower_runtime_builtin.cc │ │ │ └── vm_shape_lower.cc │ ├── distributed │ │ ├── axis_group_graph.cc │ │ ├── global_info.cc │ │ ├── struct_info.cc │ │ └── transform │ │ │ ├── legalize_redistribute.cc │ │ │ ├── lower_distir.cc │ │ │ ├── lower_global_view_to_local_view.cc │ │ │ ├── propagate_sharding.cc │ │ │ ├── utils.cc │ │ │ └── utils.h │ ├── ir │ │ ├── binding_rewrite.cc │ │ ├── block_builder.cc │ │ ├── dataflow_block_rewriter.cc │ │ ├── dataflow_expr_rewriter.cc │ │ ├── dataflow_matcher.cc │ │ ├── dataflow_matcher.h │ │ ├── dataflow_pattern.cc │ │ ├── dataflow_pattern_functor.cc │ │ ├── dataflow_rewriter.h │ │ ├── emit_te.cc │ │ ├── emit_te.h │ │ ├── expr.cc │ │ ├── expr_functor.cc │ │ ├── py_expr_functor.cc │ │ ├── struct_info.cc │ │ ├── struct_info_functor.cc │ │ ├── tir_pattern.cc │ │ ├── transform.cc │ │ └── type.cc │ ├── op │ │ ├── ccl │ │ │ ├── ccl.cc │ │ │ └── ccl.h │ │ ├── distributed │ │ │ ├── binary.cc │ │ │ ├── binary.h │ │ │ ├── ccl.cc │ │ │ ├── distributed.cc │ │ │ ├── distributed.h │ │ │ ├── linear_algebra.cc │ │ │ ├── linear_algebra.h │ │ │ ├── manipulate.cc │ │ │ ├── manipulate.h │ │ │ ├── nn.cc │ │ │ ├── nn.h │ │ │ ├── op.cc │ │ │ ├── statistical.cc │ │ │ ├── statistical.h │ │ │ ├── unary.cc │ │ │ ├── unary.h │ │ │ ├── utils.cc │ │ │ └── utils.h │ │ ├── image │ │ │ ├── resize.cc │ │ │ └── resize.h │ │ ├── memory │ │ │ ├── view.cc │ │ │ └── view.h │ │ ├── nn │ │ │ ├── attention.cc │ │ │ ├── attention.h │ │ │ ├── convolution.cc │ │ │ ├── convolution.h │ │ │ ├── nn.cc │ │ │ ├── nn.h │ │ │ ├── pooling.cc │ │ │ └── pooling.h │ │ ├── op.cc │ │ ├── op_common.cc │ │ ├── op_common.h │ │ ├── tensor │ │ │ ├── binary.cc │ │ │ ├── binary.h │ │ │ ├── create.cc │ │ │ ├── create.h │ │ │ ├── datatype.cc │ │ │ ├── datatype.h │ │ │ ├── grad.cc │ │ │ ├── grad.h │ │ │ ├── index.cc │ │ │ ├── index.h │ │ │ ├── inspect.cc │ │ │ ├── inspect.h │ │ │ ├── linear_algebra.cc │ │ │ ├── linear_algebra.h │ │ │ ├── manipulate.cc │ │ │ ├── manipulate.h │ │ │ ├── qdq.cc │ │ │ ├── qdq.h │ │ │ ├── sampling.cc │ │ │ ├── sampling.h │ │ │ ├── search.cc │ │ │ ├── search.h │ │ │ ├── set.cc │ │ │ ├── set.h │ │ │ ├── sorting.cc │ │ │ ├── sorting.h │ │ │ ├── statistical.cc │ │ │ ├── statistical.h │ │ │ ├── ternary.cc │ │ │ ├── ternary.h │ │ │ ├── unary.cc │ │ │ └── unary.h │ │ └── vision │ │ │ ├── nms.cc │ │ │ └── nms.h │ ├── testing │ │ └── transform.cc │ ├── training │ │ ├── utils.cc │ │ └── utils.h │ ├── transform │ │ ├── adjust_matmul_order.cc │ │ ├── allocate_workspace.cc │ │ ├── alter_op_impl.cc │ │ ├── annotate_tir_op_pattern.cc │ │ ├── attach_attr_layout_free_buffers.cc │ │ ├── attach_global_symbol.cc │ │ ├── bind_params.cc │ │ ├── bind_symbolic_vars.cc │ │ ├── bundle_model_params.cc │ │ ├── call_tir_rewrite.cc │ │ ├── canonicalize_bindings.cc │ │ ├── combine_parallel_matmul.cc │ │ ├── compute_prim_value.cc │ │ ├── convert_dataflow.cc │ │ ├── convert_layout.cc │ │ ├── dataflow_inplace.cc │ │ ├── dead_code_elimination.cc │ │ ├── decompose_ops.cc │ │ ├── eliminate_common_subexpr.cc │ │ ├── expand_matmul_of_sum.cc │ │ ├── expand_tuple_arguments.cc │ │ ├── few_shot_tuning.cc │ │ ├── fold_constant.cc │ │ ├── fuse_ops.cc │ │ ├── fuse_tir.cc │ │ ├── gradient.cc │ │ ├── gradient_simplifier.cc │ │ ├── gradient_simplifier.h │ │ ├── infer_amp_utils.cc │ │ ├── infer_amp_utils.h │ │ ├── infer_layout_utils.cc │ │ ├── infer_layout_utils.h │ │ ├── inline_functions.cc │ │ ├── kill_after_last_use.cc │ │ ├── lambda_lift.cc │ │ ├── lazy_transform_params.cc │ │ ├── legalize_ops.cc │ │ ├── lift_transform_params.cc │ │ ├── lower_alloc_tensor.cc │ │ ├── merge_composite_functions.cc │ │ ├── meta_schedule.cc │ │ ├── normalize.cc │ │ ├── realize_vdevice.cc │ │ ├── remove_purity_checking.cc │ │ ├── remove_unused_outputs.cc │ │ ├── remove_unused_parameters.cc │ │ ├── reorder_permute_dims_after_concat.cc │ │ ├── reorder_take_after_matmul.cc │ │ ├── replace_global_vars.cc │ │ ├── rewrite_cuda_graph.cc │ │ ├── rewrite_dataflow_reshape.cc │ │ ├── run_codegen.cc │ │ ├── split_call_tir_by_pattern.cc │ │ ├── split_layout_rewrite_preproc.cc │ │ ├── static_plan_block_memory.cc │ │ ├── to_mixed_precision.cc │ │ ├── to_non_dataflow.cc │ │ ├── topological_sort.cc │ │ ├── update_param_struct_info.cc │ │ ├── update_vdevice.cc │ │ ├── utils.cc │ │ └── utils.h │ └── utils.cc ├── runtime │ ├── builtin_fp16.cc │ ├── const_loader_module.cc │ ├── const_loader_module.h │ ├── contrib │ │ ├── amx │ │ │ └── amx_config.cc │ │ ├── arm_compute_lib │ │ │ ├── acl_allocator.cc │ │ │ ├── acl_allocator.h │ │ │ ├── acl_runtime.cc │ │ │ ├── acl_utils.cc │ │ │ └── acl_utils.h │ │ ├── bnns │ │ │ ├── bnns_json_runtime.cc │ │ │ └── bnns_wrp.h │ │ ├── cblas │ │ │ ├── cblas.cc │ │ │ ├── dnnl_blas.cc │ │ │ ├── gemm_common.h │ │ │ └── mkl.cc │ │ ├── clml │ │ │ ├── clml_memory_planner.cc │ │ │ ├── clml_memory_planner.h │ │ │ ├── clml_runtime.cc │ │ │ ├── clml_runtime.h │ │ │ ├── clml_utils.cc │ │ │ └── clml_utils.h │ │ ├── coreml │ │ │ ├── coreml_runtime.h │ │ │ └── coreml_runtime.mm │ │ ├── cublas │ │ │ ├── cublas.cc │ │ │ ├── cublas_json_runtime.cc │ │ │ ├── cublas_utils.cc │ │ │ └── cublas_utils.h │ │ ├── cudnn │ │ │ ├── conv_backward.cc │ │ │ ├── conv_forward.cc │ │ │ ├── cudnn_frontend │ │ │ │ ├── attention.cc │ │ │ │ └── attention.h │ │ │ ├── cudnn_json_runtime.cc │ │ │ ├── cudnn_utils.cc │ │ │ ├── cudnn_utils.h │ │ │ └── softmax.cc │ │ ├── curand │ │ │ ├── curand.cc │ │ │ ├── helper_cuda_kernels.cu │ │ │ └── helper_cuda_kernels.h │ │ ├── cutlass │ │ │ ├── fp16_group_gemm.cuh │ │ │ ├── fp16_group_gemm_runner_sm100.cuh │ │ │ ├── fp16_group_gemm_runner_sm90.cuh │ │ │ ├── fp16_group_gemm_sm100.cu │ │ │ ├── fp16_group_gemm_sm90.cu │ │ │ ├── fp8_gemm.cu │ │ │ ├── fp8_group_gemm_sm90.cu │ │ │ ├── fp8_groupwise_scaled_gemm.cuh │ │ │ ├── fp8_groupwise_scaled_gemm_runner_sm100.cuh │ │ │ ├── fp8_groupwise_scaled_gemm_runner_sm90.cuh │ │ │ ├── fp8_groupwise_scaled_gemm_sm100.cu │ │ │ ├── fp8_groupwise_scaled_gemm_sm90.cu │ │ │ ├── fp8_groupwise_scaled_group_gemm_runner_sm100.cuh │ │ │ ├── fp8_groupwise_scaled_group_gemm_sm100.cu │ │ │ ├── gemm_runner.cuh │ │ │ └── weight_preprocess.cc │ │ ├── dnnl │ │ │ ├── dnnl.cc │ │ │ ├── dnnl_json_runtime.cc │ │ │ ├── dnnl_kernel.h │ │ │ ├── dnnl_tensor_requisite.h │ │ │ ├── dnnl_utils.cc │ │ │ └── dnnl_utils.h │ │ ├── edgetpu │ │ │ ├── edgetpu_runtime.cc │ │ │ └── edgetpu_runtime.h │ │ ├── hipblas │ │ │ ├── hipblas.cc │ │ │ ├── hipblas_json_runtime.cc │ │ │ ├── hipblas_utils.cc │ │ │ └── hipblas_utils.h │ │ ├── json │ │ │ ├── json_node.h │ │ │ └── json_runtime.h │ │ ├── miopen │ │ │ ├── conv_forward.cc │ │ │ ├── miopen_utils.cc │ │ │ ├── miopen_utils.h │ │ │ └── softmax.cc │ │ ├── mps │ │ │ ├── conv.mm │ │ │ ├── gemm.mm │ │ │ ├── mps_utils.h │ │ │ └── mps_utils.mm │ │ ├── mrvl │ │ │ ├── mrvl_base64.h │ │ │ ├── mrvl_hw_runtime.cc │ │ │ ├── mrvl_runtime.cc │ │ │ ├── mrvl_sw_runtime_lib.cc │ │ │ └── mrvl_sw_runtime_lib.h │ │ ├── msc │ │ │ └── tensorrt_runtime.cc │ │ ├── mscclpp │ │ │ └── allreduce.cu │ │ ├── nnapi │ │ │ ├── nnapi_builder.cc │ │ │ ├── nnapi_builder.h │ │ │ ├── nnapi_ops.cc │ │ │ ├── nnapi_ops.h │ │ │ └── nnapi_runtime.cc │ │ ├── nvshmem │ │ │ ├── init.cc │ │ │ ├── kv_transfer.cu │ │ │ └── memory_allocator.cc │ │ ├── papi │ │ │ └── papi.cc │ │ ├── random │ │ │ ├── mt_random_engine.cc │ │ │ └── random.cc │ │ ├── rocblas │ │ │ └── rocblas.cc │ │ ├── sort │ │ │ └── sort.cc │ │ ├── tensorrt │ │ │ ├── tensorrt_builder.cc │ │ │ ├── tensorrt_builder.h │ │ │ ├── tensorrt_calibrator.h │ │ │ ├── tensorrt_logger.h │ │ │ ├── tensorrt_ops.cc │ │ │ ├── tensorrt_ops.h │ │ │ ├── tensorrt_runtime.cc │ │ │ └── tensorrt_utils.h │ │ ├── tflite │ │ │ ├── tflite_runtime.cc │ │ │ └── tflite_runtime.h │ │ ├── thrust │ │ │ └── thrust.cu │ │ └── vllm │ │ │ ├── attention_kernels.cu │ │ │ ├── attention_utils.cuh │ │ │ ├── cache_alloc.cc │ │ │ ├── cache_kernels.cu │ │ │ └── dtype_float16.h │ ├── cpu_device_api.cc │ ├── cuda │ │ ├── cuda_common.h │ │ ├── cuda_device_api.cc │ │ ├── cuda_module.cc │ │ ├── cuda_module.h │ │ └── l2_cache_flush.cc │ ├── device_api.cc │ ├── disco │ │ ├── bcast_session.cc │ │ ├── bcast_session.h │ │ ├── builtin.cc │ │ ├── cuda_ipc │ │ │ ├── cuda_ipc_memory.cc │ │ │ └── custom_allreduce.cc │ │ ├── disco_worker.cc │ │ ├── disco_worker_thread.h │ │ ├── distributed │ │ │ └── socket_session.cc │ │ ├── loader.cc │ │ ├── message_queue.h │ │ ├── nccl │ │ │ ├── nccl.cc │ │ │ └── nccl_context.h │ │ ├── process_session.cc │ │ ├── protocol.h │ │ ├── session.cc │ │ ├── threaded_session.cc │ │ └── utils.h │ ├── file_utils.cc │ ├── file_utils.h │ ├── hexagon │ │ ├── README.md │ │ ├── hexagon_buffer.cc │ │ ├── hexagon_buffer.h │ │ ├── hexagon_buffer_manager.h │ │ ├── hexagon_common.cc │ │ ├── hexagon_common.h │ │ ├── hexagon_device_api.cc │ │ ├── hexagon_device_api.h │ │ ├── hexagon_htp.cc │ │ ├── hexagon_htp.h │ │ ├── hexagon_hvx.cc │ │ ├── hexagon_hvx.h │ │ ├── hexagon_module.cc │ │ ├── hexagon_module.h │ │ ├── hexagon_power_manager.cc │ │ ├── hexagon_power_manager.h │ │ ├── hexagon_thread_manager.cc │ │ ├── hexagon_thread_manager.h │ │ ├── hexagon_user_dma.cc │ │ ├── hexagon_user_dma.h │ │ ├── hexagon_user_dma_descriptors.h │ │ ├── hexagon_user_dma_instructions.h │ │ ├── hexagon_user_dma_registers.h │ │ ├── hexagon_vtcm_pool.cc │ │ ├── hexagon_vtcm_pool.h │ │ ├── ops │ │ │ ├── conv2d.h │ │ │ ├── conv2d_fp16_hvx.cc │ │ │ ├── conv2d_quant_hvx.cc │ │ │ └── conv_utils.cc │ │ ├── profiler │ │ │ ├── README.md │ │ │ ├── lwp_handler.S │ │ │ ├── prof_utils.cc │ │ │ └── prof_utils.h │ │ ├── qhl │ │ │ └── qhl_wrapper.cc │ │ ├── ring_buffer.h │ │ └── rpc │ │ │ ├── android │ │ │ └── session.cc │ │ │ ├── android_bash.sh.template │ │ │ ├── hexagon │ │ │ └── rpc_server.cc │ │ │ ├── hexagon_rpc.idl │ │ │ └── simulator │ │ │ ├── hexagon_sim_proto.h │ │ │ ├── rpc_server.cc │ │ │ └── session.cc │ ├── logging.cc │ ├── memory │ │ ├── memory_manager.cc │ │ ├── naive_allocator.h │ │ └── pooled_allocator.h │ ├── meta_data.h │ ├── metal │ │ ├── metal_common.h │ │ ├── metal_device_api.mm │ │ ├── metal_module.h │ │ └── metal_module.mm │ ├── minrpc │ │ ├── minrpc_server.h │ │ ├── posix_popen_server │ │ │ └── posix_popen_server.cc │ │ └── rpc_reference.h │ ├── module.cc │ ├── nvtx.cc │ ├── opencl │ │ ├── opencl_common.h │ │ ├── opencl_device_api.cc │ │ ├── opencl_module.cc │ │ ├── opencl_module.h │ │ ├── opencl_module_spirv.cc │ │ └── opencl_wrapper │ │ │ ├── README.md │ │ │ └── opencl_wrapper.cc │ ├── pack_args.h │ ├── profiling.cc │ ├── regex.cc │ ├── regex.h │ ├── rocm │ │ ├── rocm_common.h │ │ ├── rocm_device_api.cc │ │ ├── rocm_module.cc │ │ └── rocm_module.h │ ├── rpc │ │ ├── rpc_channel.cc │ │ ├── rpc_channel.h │ │ ├── rpc_device_api.cc │ │ ├── rpc_endpoint.cc │ │ ├── rpc_endpoint.h │ │ ├── rpc_event_impl.cc │ │ ├── rpc_local_session.cc │ │ ├── rpc_local_session.h │ │ ├── rpc_module.cc │ │ ├── rpc_pipe_impl.cc │ │ ├── rpc_server_env.cc │ │ ├── rpc_session.cc │ │ ├── rpc_session.h │ │ ├── rpc_socket_impl.cc │ │ └── rpc_socket_impl.h │ ├── source_utils.cc │ ├── source_utils.h │ ├── spirv │ │ └── spirv_shader.h │ ├── static_library.cc │ ├── static_library.h │ ├── tensor.cc │ ├── texture.h │ ├── thread_map.h │ ├── thread_pool.cc │ ├── thread_storage_scope.h │ ├── threading_backend.cc │ ├── vm │ │ ├── attn_backend.cc │ │ ├── attn_backend.h │ │ ├── attn_utils.h │ │ ├── builtin.cc │ │ ├── bytecode.cc │ │ ├── cuda │ │ │ └── cuda_graph_builtin.cc │ │ ├── executable.cc │ │ ├── hexagon │ │ │ └── builtin.cc │ │ ├── kv_state.cc │ │ ├── kv_state.h │ │ ├── lm_support.cc │ │ ├── paged_kv_cache.cc │ │ ├── rnn_state.cc │ │ ├── tensor_cache_support.cc │ │ └── vm.cc │ ├── vulkan │ │ ├── README.md │ │ ├── vulkan_amdrgp.cc │ │ ├── vulkan_amdrgp.h │ │ ├── vulkan_buffer.cc │ │ ├── vulkan_buffer.h │ │ ├── vulkan_common.cc │ │ ├── vulkan_common.h │ │ ├── vulkan_device.cc │ │ ├── vulkan_device.h │ │ ├── vulkan_device_api.cc │ │ ├── vulkan_device_api.h │ │ ├── vulkan_instance.cc │ │ ├── vulkan_instance.h │ │ ├── vulkan_module.cc │ │ ├── vulkan_module.h │ │ ├── vulkan_stream.cc │ │ ├── vulkan_stream.h │ │ ├── vulkan_wrapped_func.cc │ │ └── vulkan_wrapped_func.h │ ├── workspace_pool.cc │ └── workspace_pool.h ├── script │ ├── ir_builder │ │ ├── base.cc │ │ ├── ir │ │ │ ├── frame.cc │ │ │ ├── ir.cc │ │ │ └── utils.h │ │ ├── relax │ │ │ ├── distributed.cc │ │ │ ├── frame.cc │ │ │ ├── ir.cc │ │ │ └── utils.h │ │ └── tir │ │ │ ├── frame.cc │ │ │ ├── ir.cc │ │ │ └── utils.h │ └── printer │ │ ├── doc.cc │ │ ├── doc_printer │ │ ├── base_doc_printer.cc │ │ ├── base_doc_printer.h │ │ └── python_doc_printer.cc │ │ ├── ir │ │ ├── distributed.cc │ │ ├── ir.cc │ │ ├── misc.cc │ │ └── utils.h │ │ ├── ir_docsifier.cc │ │ ├── relax │ │ ├── binding.cc │ │ ├── call.cc │ │ ├── distributed.cc │ │ ├── expr.cc │ │ ├── function.cc │ │ ├── region.cc │ │ ├── struct_info.cc │ │ ├── tir.cc │ │ ├── type.cc │ │ └── utils.h │ │ ├── tir │ │ ├── block.cc │ │ ├── buffer.cc │ │ ├── expr.cc │ │ ├── for_loop.cc │ │ ├── function.cc │ │ ├── ir.cc │ │ ├── stmt.cc │ │ └── utils.h │ │ └── utils.h ├── support │ ├── arena.h │ ├── array.h │ ├── base64.h │ ├── errno_handling.h │ ├── ffi_testing.cc │ ├── generic_arena.h │ ├── hexdump.cc │ ├── hexdump.h │ ├── libinfo.cc │ ├── nd_int_set.h │ ├── ordered_map.h │ ├── ordered_set.h │ ├── parallel_for.cc │ ├── pipe.h │ ├── process_id.h │ ├── ring_buffer.h │ ├── scalars.cc │ ├── scalars.h │ ├── socket.h │ ├── ssize.h │ ├── str_escape.h │ ├── table_printer.h │ └── utils.h ├── target │ ├── build_common.h │ ├── codegen.cc │ ├── datatype │ │ ├── myfloat │ │ │ └── myfloat.cc │ │ ├── posit │ │ │ └── posit-wrapper.cc │ │ ├── registry.cc │ │ └── registry.h │ ├── intrin_rule.cc │ ├── intrin_rule.h │ ├── llvm │ │ ├── codegen_aarch64.cc │ │ ├── codegen_amdgpu.cc │ │ ├── codegen_arm.cc │ │ ├── codegen_blob.cc │ │ ├── codegen_blob.h │ │ ├── codegen_cpu.cc │ │ ├── codegen_cpu.h │ │ ├── codegen_hexagon.cc │ │ ├── codegen_llvm.cc │ │ ├── codegen_llvm.h │ │ ├── codegen_nvptx.cc │ │ ├── codegen_params.cc │ │ ├── codegen_params.h │ │ ├── codegen_x86_64.cc │ │ ├── intrin_rule_hexagon.cc │ │ ├── intrin_rule_llvm.cc │ │ ├── intrin_rule_llvm.h │ │ ├── intrin_rule_nvptx.cc │ │ ├── intrin_rule_rocm.cc │ │ ├── llvm_instance.cc │ │ ├── llvm_instance.h │ │ ├── llvm_module.cc │ │ └── llvm_module.h │ ├── opt │ │ ├── README │ │ ├── build_cuda_off.cc │ │ ├── build_cuda_on.cc │ │ ├── build_hexagon_off.cc │ │ ├── build_metal_off.cc │ │ ├── build_opencl_off.cc │ │ └── build_rocm_off.cc │ ├── parsers │ │ ├── aprofile.cc │ │ ├── aprofile.h │ │ ├── cpu.cc │ │ ├── cpu.h │ │ ├── mprofile.cc │ │ └── mprofile.h │ ├── source │ │ ├── codegen_c.cc │ │ ├── codegen_c.h │ │ ├── codegen_c_host.cc │ │ ├── codegen_c_host.h │ │ ├── codegen_cuda.cc │ │ ├── codegen_cuda.h │ │ ├── codegen_metal.cc │ │ ├── codegen_metal.h │ │ ├── codegen_opencl.cc │ │ ├── codegen_opencl.h │ │ ├── codegen_params.cc │ │ ├── codegen_params.h │ │ ├── codegen_source_base.cc │ │ ├── codegen_source_base.h │ │ ├── codegen_webgpu.cc │ │ ├── codegen_webgpu.h │ │ ├── intrin_rule_cuda.cc │ │ ├── intrin_rule_metal.cc │ │ ├── intrin_rule_opencl.cc │ │ ├── intrin_rule_webgpu.cc │ │ ├── literal │ │ │ ├── cuda_half_t.h │ │ │ └── cuda_int8_t.h │ │ ├── ptx.cc │ │ ├── ptx.h │ │ └── source_module.cc │ ├── spirv │ │ ├── build_vulkan.cc │ │ ├── codegen_spirv.cc │ │ ├── codegen_spirv.h │ │ ├── intrin_rule_spirv.cc │ │ ├── ir_builder.cc │ │ ├── ir_builder.h │ │ ├── spirv_support.cc │ │ ├── spirv_support.h │ │ ├── spirv_utils.cc │ │ └── spirv_utils.h │ ├── tag.cc │ ├── target.cc │ ├── target_info.cc │ ├── target_kind.cc │ └── virtual_device.cc ├── te │ ├── operation │ │ ├── compute_op.cc │ │ ├── create_primfunc.cc │ │ ├── create_primfunc.h │ │ ├── extern_op.cc │ │ ├── graph.cc │ │ ├── graph.h │ │ ├── placeholder_op.cc │ │ └── scan_op.cc │ └── tensor.cc ├── tir │ ├── analysis │ │ ├── block_access_region_detector.cc │ │ ├── buffer_access_lca_detector.cc │ │ ├── calculate_allocated_memory.cc │ │ ├── check_contains.cc │ │ ├── check_contains.h │ │ ├── collect_call_map.cc │ │ ├── control_flow_graph.cc │ │ ├── control_flow_graph.h │ │ ├── deep_equal.cc │ │ ├── estimate_flops.cc │ │ ├── expr_complexity.cc │ │ ├── identify_memcpy.cc │ │ ├── is_pure_function.cc │ │ ├── oob_checker.cc │ │ ├── side_effect.cc │ │ ├── stmt_finding.cc │ │ ├── var_touch.cc │ │ ├── var_use_def_analysis.cc │ │ ├── var_use_def_analysis.h │ │ ├── verify_gpu_code.cc │ │ ├── verify_memory.cc │ │ ├── verify_ssa.cc │ │ └── verify_well_formed.cc │ ├── ir │ │ ├── block_dependence_info.cc │ │ ├── block_scope.cc │ │ ├── buffer.cc │ │ ├── buffer_common.h │ │ ├── data_layout.cc │ │ ├── data_type_rewriter.cc │ │ ├── expr.cc │ │ ├── expr_functor.cc │ │ ├── function.cc │ │ ├── functor_common.h │ │ ├── index_map.cc │ │ ├── py_functor.cc │ │ ├── script │ │ │ ├── script_complete.cc │ │ │ └── script_complete.h │ │ ├── specialize.cc │ │ ├── stmt.cc │ │ ├── stmt_functor.cc │ │ ├── tir_visitor_with_path.cc │ │ ├── tir_visitor_with_path.h │ │ └── transform.cc │ ├── op │ │ ├── builtin.cc │ │ ├── op.cc │ │ └── runtime.cc │ ├── schedule │ │ ├── analysis.h │ │ ├── analysis │ │ │ ├── analysis.cc │ │ │ ├── layout.cc │ │ │ ├── reducer.cc │ │ │ └── verify.cc │ │ ├── concrete_schedule.cc │ │ ├── concrete_schedule.h │ │ ├── error.cc │ │ ├── error.h │ │ ├── instruction.cc │ │ ├── instruction_traits.h │ │ ├── ir_comparator.cc │ │ ├── ir_comparator.h │ │ ├── primitive.h │ │ ├── primitive │ │ │ ├── annotate.cc │ │ │ ├── annotate_buffer_access.cc │ │ │ ├── block_annotate.cc │ │ │ ├── blockize_tensorize.cc │ │ │ ├── cache_index.cc │ │ │ ├── cache_read_write.cc │ │ │ ├── compute_at.cc │ │ │ ├── compute_inline.cc │ │ │ ├── decompose_padding.cc │ │ │ ├── for_kind.cc │ │ │ ├── get_block_loop.cc │ │ │ ├── hide_buffer_access.cc │ │ │ ├── layout_transformation.cc │ │ │ ├── loop_transformation.cc │ │ │ ├── pad_einsum.cc │ │ │ ├── read_write_at.cc │ │ │ ├── reduction.cc │ │ │ ├── reorder_block_iter_var.cc │ │ │ ├── rolling_buffer.cc │ │ │ └── sampling.cc │ │ ├── schedule.cc │ │ ├── state.cc │ │ ├── trace.cc │ │ ├── traced_schedule.cc │ │ ├── traced_schedule.h │ │ ├── transform.cc │ │ ├── transform.h │ │ └── utils.h │ └── transforms │ │ ├── annotate_device_regions.cc │ │ ├── annotate_irregular_loop.cc │ │ ├── arg_binder.cc │ │ ├── arg_binder.h │ │ ├── bind_params.cc │ │ ├── bind_target.cc │ │ ├── bound_checker.cc │ │ ├── combine_context_call.cc │ │ ├── common_subexpr_elim.cc │ │ ├── common_subexpr_elim.h │ │ ├── common_subexpr_elim_tools.cc │ │ ├── common_subexpr_elim_tools.h │ │ ├── compact_buffer_region.cc │ │ ├── convert_blocks_to_opaque.cc │ │ ├── convert_for_loops_serial.cc │ │ ├── decorate_device_scope.cc │ │ ├── default_gpu_schedule.cc │ │ ├── dtype_conversion.cc │ │ ├── dtype_conversion.h │ │ ├── extract_constants.cc │ │ ├── flatten_buffer.cc │ │ ├── force_narrow_index_to_i32.cc │ │ ├── hoist_expression.cc │ │ ├── inject_double_buffer.cc │ │ ├── inject_permuted_layout.cc │ │ ├── inject_ptx_async_copy.cc │ │ ├── inject_ptx_ldg32.cc │ │ ├── inject_rolling_buffer.cc │ │ ├── inject_software_pipeline.cc │ │ ├── inject_virtual_thread.cc │ │ ├── inline_private_functions.cc │ │ ├── ir_utils.cc │ │ ├── ir_utils.h │ │ ├── lift_thread_binding.cc │ │ ├── loop_partition.cc │ │ ├── lower_async_dma.cc │ │ ├── lower_cross_thread_reduction.cc │ │ ├── lower_custom_datatypes.cc │ │ ├── lower_device_kernel_launch.cc │ │ ├── lower_device_storage_access_info.cc │ │ ├── lower_init_block.cc │ │ ├── lower_intrin.cc │ │ ├── lower_match_buffer.cc │ │ ├── lower_opaque_block.cc │ │ ├── lower_thread_allreduce.cc │ │ ├── lower_tvm_builtin.cc │ │ ├── lower_vtcm_alloc.cc │ │ ├── lower_warp_memory.cc │ │ ├── make_packed_api.cc │ │ ├── make_unpacked_api.cc │ │ ├── manifest_shared_memory_local_stage.cc │ │ ├── memhammer_coalesce.cc │ │ ├── memhammer_intermediate_stage.cc │ │ ├── memhammer_lower_auto_copy.cc │ │ ├── memhammer_rewrite_rule.h │ │ ├── memhammer_tensorcore_rewrite.cc │ │ ├── merge_shared_memory_allocations.cc │ │ ├── narrow_datatype.cc │ │ ├── plan_update_buffer_allocation_location.cc │ │ ├── primfunc_utils.cc │ │ ├── profile_instrumentation.cc │ │ ├── reduce_branching_through_overcompute.cc │ │ ├── remap_thread_axis.cc │ │ ├── remove_assume.cc │ │ ├── remove_no_op.cc │ │ ├── remove_no_op.h │ │ ├── remove_store_undef.cc │ │ ├── remove_weight_layout_rewrite_block.cc │ │ ├── renew_defs.cc │ │ ├── renormalize_split_pattern.cc │ │ ├── replace_global_vars.cc │ │ ├── replace_selected_expr.cc │ │ ├── replace_selected_expr.h │ │ ├── rewrite_unsafe_select.cc │ │ ├── simplify.cc │ │ ├── simplify.h │ │ ├── skip_assert.cc │ │ ├── split_host_device.cc │ │ ├── storage_access.cc │ │ ├── storage_access.h │ │ ├── storage_rewrite.cc │ │ ├── tensorcore_infer_fragment.cc │ │ ├── thread_storage_sync.cc │ │ ├── transform_mma_buffer_layout.cc │ │ ├── unify_thread_binding.cc │ │ ├── unroll_loop.cc │ │ ├── unsupported_dtype_legalize.cc │ │ ├── update_pointer_storage_scope.cc │ │ ├── update_pointer_storage_scope.h │ │ ├── using_assume_to_reduce_branches.cc │ │ └── vectorize_loop.cc └── topi │ ├── broadcast.cc │ ├── einsum.cc │ ├── elemwise.cc │ ├── nn.cc │ ├── reduction.cc │ ├── transform.cc │ ├── utils.cc │ └── vision.cc ├── tests ├── cpp-runtime │ ├── hexagon │ │ ├── hexagon_buffer_tests.cc │ │ ├── hexagon_conv_utils_test.h │ │ ├── hexagon_device_api_tests.cc │ │ ├── hexagon_fp16_utils_tests.cc │ │ ├── hexagon_quant_utils_tests.cc │ │ ├── hexagon_thread_manager_tests.cc │ │ ├── hexagon_user_dma_tests.cc │ │ ├── hexagon_vtcm_pool_tests.cc │ │ ├── ring_buffer_tests.cc │ │ ├── run_all_tests.cc │ │ └── run_unit_tests.cc │ └── opencl │ │ ├── aa_opencl_qcom_extn.cc │ │ ├── clml_memory_planner.cc │ │ ├── opencl_compile_to_bin.cc │ │ ├── opencl_nativeptr.cc │ │ ├── opencl_timer_test.cc │ │ └── texture_copy_test.cc ├── cpp │ ├── .gitignore │ ├── README.md │ ├── arith_integer_set_test.cc │ ├── arith_simplify_test.cc │ ├── data_type_rewriter_test.cc │ ├── expr_test.cc │ ├── ir_functor_test.cc │ ├── llvm_codegen_registry_test.cc │ ├── ndarray_test.cc │ ├── nested_msg_test.cc │ ├── object_protocol_test.cc │ ├── parallel_for_test.cc │ ├── pattern_match_test.cc │ ├── random_engine_test.cc │ ├── runtime │ │ ├── logging_test.cc │ │ └── memory │ │ │ └── memory_manager_tests.cc │ ├── support │ │ ├── ring_buffer_test.cc │ │ └── scalars_test.cc │ ├── support_test.cc │ ├── target │ │ ├── parsers │ │ │ ├── aprofile_test.cc │ │ │ └── mprofile_test.cc │ │ └── virtual_device_test.cc │ ├── target_test.cc │ ├── te_compute_test.cc │ ├── threading_backend_test.cc │ ├── tir_analysis_side_effect.cc │ ├── tir_scalable_datatype.cc │ └── topi_ewise_test.cc ├── lint │ ├── add_asf_header.py │ ├── blocklint.sh │ ├── check_asf_header.sh │ ├── check_cmake_options.py │ ├── check_file_type.py │ ├── cppdocs.sh │ ├── cpplint.sh │ ├── docker-format.sh │ ├── filter_untracked.py │ ├── flake8.sh │ ├── git-black.sh │ ├── git-clang-format.sh │ ├── jnilint.sh │ ├── pylint.sh │ ├── pylintrc │ ├── rat-excludes │ ├── trailing_newlines.py │ └── whitespace.sh ├── python │ ├── all-platform-minimal-test │ │ ├── README.md │ │ ├── test_minimal_target_codegen_llvm.py │ │ ├── test_runtime_ndarray.py │ │ └── test_runtime_packed_func.py │ ├── arith │ │ ├── test_arith_canonical_simplify.py │ │ ├── test_arith_const_int_bound.py │ │ ├── test_arith_deduce_bound.py │ │ ├── test_arith_detect_clip_bound.py │ │ ├── test_arith_detect_cse.py │ │ ├── test_arith_detect_linear_equation.py │ │ ├── test_arith_domain_touched.py │ │ ├── test_arith_intset.py │ │ ├── test_arith_iter_affine_map.py │ │ ├── test_arith_modular_set.py │ │ ├── test_arith_narrow_predicate_expression.py │ │ ├── test_arith_rewrite_simplify.py │ │ ├── test_arith_simplify.py │ │ ├── test_arith_solve_linear_equations.py │ │ └── test_arith_solve_linear_inequality.py │ ├── ci │ │ ├── __init__.py │ │ ├── sample_prs │ │ │ └── pr10786.json │ │ ├── test_ci.py │ │ ├── test_tvmbot.py │ │ └── test_utils.py │ ├── codegen │ │ ├── test_gpu_codegen_allreduce.py │ │ ├── test_inject_ptx_ldg32.py │ │ ├── test_target_codegen.py │ │ ├── test_target_codegen_aarch64.py │ │ ├── test_target_codegen_arm.py │ │ ├── test_target_codegen_blob.py │ │ ├── test_target_codegen_bool.py │ │ ├── test_target_codegen_c_host.py │ │ ├── test_target_codegen_cross_llvm.py │ │ ├── test_target_codegen_cuda.py │ │ ├── test_target_codegen_cuda_fp4.py │ │ ├── test_target_codegen_cuda_fp8.py │ │ ├── test_target_codegen_device.py │ │ ├── test_target_codegen_extern.py │ │ ├── test_target_codegen_gpu_common.py │ │ ├── test_target_codegen_hexagon.py │ │ ├── test_target_codegen_llvm.py │ │ ├── test_target_codegen_llvm_vla.py │ │ ├── test_target_codegen_metal.py │ │ ├── test_target_codegen_opencl.py │ │ ├── test_target_codegen_riscv.py │ │ ├── test_target_codegen_rocm.py │ │ ├── test_target_codegen_static_init.py │ │ ├── test_target_codegen_vulkan.py │ │ └── test_target_codegen_x86.py │ ├── conftest.py │ ├── contrib │ │ ├── pickle_memoize_script.py │ │ ├── test_android │ │ │ ├── __init__.py │ │ │ ├── infrastructure.py │ │ │ └── test_meta_schedule.py │ │ ├── test_cblas.py │ │ ├── test_ccache.py │ │ ├── test_coreml_runtime.py │ │ ├── test_cutlass_gemm.py │ │ ├── test_dlpack.py │ │ ├── test_edgetpu_runtime.py │ │ ├── test_hexagon │ │ │ ├── README.md │ │ │ ├── README_RPC.md │ │ │ ├── __init__.py │ │ │ ├── benchmark_util.py │ │ │ ├── conftest.py │ │ │ ├── conv2d │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── test_conv2d_blocked.md │ │ │ │ └── test_conv2d_conv2d.md │ │ │ ├── infrastructure.py │ │ │ ├── pytest_util.py │ │ │ ├── test_2d_physical_buffers.py │ │ │ ├── test_async_dma_pipeline.py │ │ │ ├── test_benchmark_elemwise_add.py │ │ │ ├── test_benchmark_maxpool2d.py │ │ │ ├── test_dma_builtin.py │ │ │ ├── test_fixed_point_conversion.py │ │ │ ├── test_memory_alloc.py │ │ │ ├── test_meta_schedule.py │ │ │ ├── test_parallel_hvx.py │ │ │ ├── test_parallel_hvx_load_vtcm.py │ │ │ ├── test_parallel_scalar.py │ │ │ ├── test_relax_2d_buffer_allocation.py │ │ │ ├── test_relax_integration.py │ │ │ ├── test_run_unit_tests.py │ │ │ ├── test_sigmoid.py │ │ │ ├── test_software_pipeline_async.py │ │ │ ├── test_take.py │ │ │ ├── test_thread_pool.py │ │ │ ├── test_vtcm.py │ │ │ └── test_vtcm_bandwidth.py │ │ ├── test_hipblas.py │ │ ├── test_memoize.py │ │ ├── test_mps.py │ │ ├── test_msc │ │ │ ├── test_graph_build.py │ │ │ ├── test_pipeline.py │ │ │ ├── test_plugin.py │ │ │ ├── test_runner.py │ │ │ ├── test_tools.py │ │ │ ├── test_transform.py │ │ │ ├── test_translate_relax.py │ │ │ ├── test_translate_tensorrt.py │ │ │ └── test_translate_torch.py │ │ ├── test_popen_pool.py │ │ ├── test_random.py │ │ ├── test_rocblas.py │ │ ├── test_rpc_proxy.py │ │ ├── test_rpc_tracker.py │ │ ├── test_sort.py │ │ ├── test_tflite_runtime.py │ │ ├── test_tir_triton_integration.py │ │ ├── test_tvmjs.py │ │ └── test_util.py │ ├── disco │ │ ├── test_callback.py │ │ ├── test_ccl.py │ │ ├── test_custom_allreduce.py │ │ ├── test_loader.py │ │ ├── test_nvshmem.py │ │ └── test_session.py │ ├── dlight │ │ ├── test_benchmark.py │ │ ├── test_cpu_gemv.py │ │ ├── test_gpu_conv.py │ │ ├── test_gpu_fallback.py │ │ ├── test_gpu_gemv.py │ │ ├── test_gpu_general_reduction.py │ │ ├── test_gpu_low_batch_gemv.py │ │ ├── test_gpu_matmul.py │ │ ├── test_gpu_matmul_tensorize.py │ │ ├── test_gpu_reduction.py │ │ ├── test_gpu_rmsnorm.py │ │ ├── test_gpu_transpose.py │ │ └── test_primitives.py │ ├── driver │ │ └── test_compile.py │ ├── ir │ │ ├── analysis │ │ │ └── test_collect_call_map.py │ │ ├── test_container_structural_equal.py │ │ ├── test_datatype_nv_fp4.py │ │ ├── test_datatype_nv_fp8.py │ │ ├── test_dtype.py │ │ ├── test_ir_attrs.py │ │ ├── test_ir_container.py │ │ ├── test_ir_type.py │ │ ├── test_node_reflection.py │ │ ├── test_pass_instrument.py │ │ ├── test_roundtrip_runtime_module.py │ │ └── test_transform_replace_global_var.py │ ├── meta_schedule │ │ ├── test_meta_schedule_arg_info.py │ │ ├── test_meta_schedule_builder.py │ │ ├── test_meta_schedule_cost_model.py │ │ ├── test_meta_schedule_database.py │ │ ├── test_meta_schedule_feature_extractor.py │ │ ├── test_meta_schedule_feature_extractor_per_store_feature.py │ │ ├── test_meta_schedule_measure_callback.py │ │ ├── test_meta_schedule_mutator_mutate_compute_location.py │ │ ├── test_meta_schedule_mutator_mutate_parallel.py │ │ ├── test_meta_schedule_mutator_mutate_thread_binding.py │ │ ├── test_meta_schedule_mutator_mutate_tile_size.py │ │ ├── test_meta_schedule_mutator_mutate_unroll.py │ │ ├── test_meta_schedule_post_order_apply.py │ │ ├── test_meta_schedule_postproc_disallow_async_strided_mem_copy.py │ │ ├── test_meta_schedule_postproc_disallow_dynamic_loop.py │ │ ├── test_meta_schedule_postproc_rewrite_cooperative_fetch.py │ │ ├── test_meta_schedule_postproc_rewrite_layout.py │ │ ├── test_meta_schedule_postproc_rewrite_parallel_vectorize_unroll.py │ │ ├── test_meta_schedule_postproc_rewrite_reduction_block.py │ │ ├── test_meta_schedule_postproc_rewrite_tensorize.py │ │ ├── test_meta_schedule_postproc_rewrite_unbound_block.py │ │ ├── test_meta_schedule_postproc_verify_gpu_code.py │ │ ├── test_meta_schedule_postproc_verify_vtcm_limit.py │ │ ├── test_meta_schedule_profiler.py │ │ ├── test_meta_schedule_runner.py │ │ ├── test_meta_schedule_schedule_rule_add_rfactor.py │ │ ├── test_meta_schedule_schedule_rule_apply_custom_rule.py │ │ ├── test_meta_schedule_schedule_rule_auto_bind.py │ │ ├── test_meta_schedule_schedule_rule_auto_inline.py │ │ ├── test_meta_schedule_schedule_rule_cross_thread_reduction.py │ │ ├── test_meta_schedule_schedule_rule_mlt.py │ │ ├── test_meta_schedule_schedule_rule_mlt_intrin.py │ │ ├── test_meta_schedule_schedule_rule_mlt_tc.py │ │ ├── test_meta_schedule_schedule_rule_parallel_vectorize_unroll.py │ │ ├── test_meta_schedule_schedule_rule_random_compute_location.py │ │ ├── test_meta_schedule_search_strategy.py │ │ ├── test_meta_schedule_space_cpu.py │ │ ├── test_meta_schedule_space_cuda.py │ │ ├── test_meta_schedule_space_cuda_async.py │ │ ├── test_meta_schedule_space_cuda_async_multiple_initialization.py │ │ ├── test_meta_schedule_space_generator.py │ │ ├── test_meta_schedule_space_post_opt.py │ │ ├── test_meta_schedule_task_scheduler.py │ │ ├── test_meta_schedule_trace_apply.py │ │ ├── test_meta_schedule_tune_context.py │ │ └── test_meta_schedule_tune_tir.py │ ├── nightly │ │ └── test_nnapi │ │ │ ├── __init__.py │ │ │ ├── conftest.py │ │ │ ├── infrastructure.py │ │ │ ├── test_from_exported_to_cuda.py │ │ │ ├── test_network.py │ │ │ └── test_ops.py │ ├── relax │ │ ├── backend │ │ │ └── clml │ │ │ │ ├── conftest.py │ │ │ │ ├── mod_utils.py │ │ │ │ ├── test_clml_codegen.py │ │ │ │ ├── test_op_exec_clml_codegen.py │ │ │ │ └── utils.py │ │ ├── conftest.py │ │ ├── distributed │ │ │ ├── test_distributed_dtensor_sinfo.py │ │ │ ├── test_distributed_transform_legalize_redistribute.py │ │ │ ├── test_distributed_transform_lower_distir.py │ │ │ ├── test_distributed_transform_lower_global_to_local_view.py │ │ │ ├── test_distributed_transform_propagate_sharding.py │ │ │ ├── test_distributed_tvmscript_parser.py │ │ │ └── test_distributed_tvmscript_printer.py │ │ ├── frontend_nn_extern_module.cc │ │ ├── nvshmem │ │ │ ├── test_runtime_builtin_kv_cache_transfer.py │ │ │ └── test_runtime_builtin_kv_cache_transfer_kernel.py │ │ ├── test_analysis.py │ │ ├── test_analysis_computable_at_compile_time.py │ │ ├── test_analysis_contains_impure_call.py │ │ ├── test_analysis_detect_recursion.py │ │ ├── test_analysis_estimate_memory_usage.py │ │ ├── test_analysis_struct_info_analysis.py │ │ ├── test_analysis_suggest_layout_transforms.py │ │ ├── test_analysis_well_formed.py │ │ ├── test_ast_printer.py │ │ ├── test_backend_dispatch_sampling.py │ │ ├── test_backend_dispatch_sort_scan.py │ │ ├── test_backend_transform_shape_lower.py │ │ ├── test_base_py_module.py │ │ ├── test_base_py_module_printer.py │ │ ├── test_base_py_module_symbolic_shape.py │ │ ├── test_bind_params.py │ │ ├── test_bind_symbolic_vars.py │ │ ├── test_binding_rewrite.py │ │ ├── test_blockbuilder_core.py │ │ ├── test_blockbuilder_emit_te.py │ │ ├── test_codegen_coreml.py │ │ ├── test_codegen_cublas.py │ │ ├── test_codegen_cudnn.py │ │ ├── test_codegen_cutlass.py │ │ ├── test_codegen_dnnl.py │ │ ├── test_codegen_hipblas.py │ │ ├── test_codegen_tensorrt.py │ │ ├── test_contrib_vllm.py │ │ ├── test_dataflow_inplace.py │ │ ├── test_dataflow_pattern.py │ │ ├── test_dataflow_rewriter.py │ │ ├── test_dlpack_integration.py │ │ ├── test_e2e_op_dynamic.py │ │ ├── test_eliminate_pad_branch_using_buffer_assumption.py │ │ ├── test_expr.py │ │ ├── test_expr_args_converter.py │ │ ├── test_expr_functor.py │ │ ├── test_fast_math_transform.py │ │ ├── test_frontend_common.py │ │ ├── test_frontend_dynamo.py │ │ ├── test_frontend_from_exported_program.py │ │ ├── test_frontend_from_fx.py │ │ ├── test_frontend_nn_debug.py │ │ ├── test_frontend_nn_exporter.py │ │ ├── test_frontend_nn_extern_module.py │ │ ├── test_frontend_nn_jit.py │ │ ├── test_frontend_nn_modules.py │ │ ├── test_frontend_nn_mutator.py │ │ ├── test_frontend_nn_op.py │ │ ├── test_frontend_nn_packing.py │ │ ├── test_frontend_nn_subroutines.py │ │ ├── test_frontend_nn_tensor.py │ │ ├── test_frontend_onnx.py │ │ ├── test_frontend_stablehlo.py │ │ ├── test_group_gemm_flashinfer.py │ │ ├── test_inline_functions.py │ │ ├── test_kill_after_last_use.py │ │ ├── test_lower_alloc_tensor.py │ │ ├── test_meta_schedule_relax_integration.py │ │ ├── test_op_binary.py │ │ ├── test_op_ccl.py │ │ ├── test_op_create.py │ │ ├── test_op_datatype.py │ │ ├── test_op_distributed.py │ │ ├── test_op_grad.py │ │ ├── test_op_gradient_numeric.py │ │ ├── test_op_image.py │ │ ├── test_op_index.py │ │ ├── test_op_inspect.py │ │ ├── test_op_linear_algebra.py │ │ ├── test_op_manipulate.py │ │ ├── test_op_misc.py │ │ ├── test_op_nn.py │ │ ├── test_op_nn_convolution.py │ │ ├── test_op_nn_pooling.py │ │ ├── test_op_qdq.py │ │ ├── test_op_sampling.py │ │ ├── test_op_search.py │ │ ├── test_op_set.py │ │ ├── test_op_sort.py │ │ ├── test_op_statistical.py │ │ ├── test_op_take.py │ │ ├── test_op_ternary.py │ │ ├── test_op_unary.py │ │ ├── test_op_view.py │ │ ├── test_op_vision.py │ │ ├── test_optimize_layout_transform.py │ │ ├── test_pipeline.py │ │ ├── test_pytorch_integration.py │ │ ├── test_relax_operators.py │ │ ├── test_relax_to_pyfunc_converter.py │ │ ├── test_remove_redundant_reshape.py │ │ ├── test_runtime_builtin.py │ │ ├── test_runtime_builtin_paged_attention_kv_cache_cpu.py │ │ ├── test_runtime_builtin_paged_attention_kv_cache_flashinfer.py │ │ ├── test_runtime_builtin_paged_attention_kv_cache_mla_flashinfer.py │ │ ├── test_runtime_builtin_paged_attention_kv_cache_mla_tir.py │ │ ├── test_runtime_builtin_paged_attention_kv_cache_tir.py │ │ ├── test_runtime_builtin_rnn_state.py │ │ ├── test_runtime_sampling_flashinfer.py │ │ ├── test_struct_info.py │ │ ├── test_testing_nn.py │ │ ├── test_tir_call_source_kernel.py │ │ ├── test_training_append_loss.py │ │ ├── test_training_loss.py │ │ ├── test_training_optimizer.py │ │ ├── test_training_optimizer_numeric.py │ │ ├── test_training_setup_trainer.py │ │ ├── test_training_trainer_numeric.py │ │ ├── test_transform.py │ │ ├── test_transform_adjust_matmul_order.py │ │ ├── test_transform_allocate_workspace.py │ │ ├── test_transform_alter_op_impl.py │ │ ├── test_transform_annotate_tir_op_pattern.py │ │ ├── test_transform_attach_attr_layout_free_buffers.py │ │ ├── test_transform_attach_global_symbol.py │ │ ├── test_transform_bind_params.py │ │ ├── test_transform_bind_symbolic_vars.py │ │ ├── test_transform_bundle_model_params.py │ │ ├── test_transform_canonicalize_bindings.py │ │ ├── test_transform_codegen_pass.py │ │ ├── test_transform_combine_parallel_matmul.py │ │ ├── test_transform_compute_prim_value.py │ │ ├── test_transform_convert_dataflow.py │ │ ├── test_transform_convert_layout.py │ │ ├── test_transform_cse.py │ │ ├── test_transform_dead_code_elimination.py │ │ ├── test_transform_decompose_ops.py │ │ ├── test_transform_expand_matmul_of_sum.py │ │ ├── test_transform_expand_tuple_args.py │ │ ├── test_transform_few_shot_tuning.py │ │ ├── test_transform_fold_batch_norm_to_conv2d.py │ │ ├── test_transform_fold_constant.py │ │ ├── test_transform_fuse_ops.py │ │ ├── test_transform_fuse_ops_by_pattern.py │ │ ├── test_transform_fuse_tir.py │ │ ├── test_transform_fuse_transpose_matmul.py │ │ ├── test_transform_gradient.py │ │ ├── test_transform_gradient_checkpoint.py │ │ ├── test_transform_gradient_numeric.py │ │ ├── test_transform_gradient_te_register.py │ │ ├── test_transform_inline_private_functions.py │ │ ├── test_transform_ipc_allreduce_rewrite.py │ │ ├── test_transform_lambda_lift.py │ │ ├── test_transform_lazy_transform_params.py │ │ ├── test_transform_legalize_ops.py │ │ ├── test_transform_legalize_ops_binary.py │ │ ├── test_transform_legalize_ops_ccl.py │ │ ├── test_transform_legalize_ops_create_datatype.py │ │ ├── test_transform_legalize_ops_distributed.py │ │ ├── test_transform_legalize_ops_grad.py │ │ ├── test_transform_legalize_ops_image.py │ │ ├── test_transform_legalize_ops_index_linear_algebra.py │ │ ├── test_transform_legalize_ops_manipulate.py │ │ ├── test_transform_legalize_ops_nn.py │ │ ├── test_transform_legalize_ops_qdq.py │ │ ├── test_transform_legalize_ops_search_statistical.py │ │ ├── test_transform_legalize_ops_unary.py │ │ ├── test_transform_lift_transform_params.py │ │ ├── test_transform_lower_gpu_ipc_alloc_storage.py │ │ ├── test_transform_merge_composite_functions.py │ │ ├── test_transform_meta_schedule_apply_database.py │ │ ├── test_transform_meta_schedule_tuning.py │ │ ├── test_transform_normalize.py │ │ ├── test_transform_normalize_global_var.py │ │ ├── test_transform_operator_specific_normalization.py │ │ ├── test_transform_realize_vdevice.py │ │ ├── test_transform_remove_unused_outputs.py │ │ ├── test_transform_remove_unused_parameters.py │ │ ├── test_transform_reorder_permute_dims_after_concat.py │ │ ├── test_transform_reorder_take_after_matmul.py │ │ ├── test_transform_rewrite_cuda_graph.py │ │ ├── test_transform_rewrite_dataflow_reshape.py │ │ ├── test_transform_split_layout_rewrite_preproc.py │ │ ├── test_transform_static_plan_block_memory.py │ │ ├── test_transform_to_mixed_precision.py │ │ ├── test_transform_topological_sort.py │ │ ├── test_transform_update_param_struct_info.py │ │ ├── test_transform_update_vdevice.py │ │ ├── test_tvmscript_ir_builder.py │ │ ├── test_tvmscript_parser.py │ │ ├── test_tvmscript_parser_op_arith_cmp.py │ │ ├── test_tvmscript_parser_op_create.py │ │ ├── test_tvmscript_parser_op_datatype.py │ │ ├── test_tvmscript_parser_op_grad.py │ │ ├── test_tvmscript_parser_op_image.py │ │ ├── test_tvmscript_parser_op_index.py │ │ ├── test_tvmscript_parser_op_linear_algebra.py │ │ ├── test_tvmscript_parser_op_manipulate.py │ │ ├── test_tvmscript_parser_op_nn.py │ │ ├── test_tvmscript_parser_op_search.py │ │ ├── test_tvmscript_parser_op_set.py │ │ ├── test_tvmscript_parser_op_sort.py │ │ ├── test_tvmscript_parser_op_statistical.py │ │ ├── test_tvmscript_parser_op_vision.py │ │ ├── test_tvmscript_printer_relax.py │ │ ├── test_tvmscript_pyfunc.py │ │ ├── test_utils.py │ │ ├── test_vm_alloc_storage_with_scope.py │ │ ├── test_vm_build.py │ │ ├── test_vm_builtin.py │ │ ├── test_vm_builtin_lower.py │ │ ├── test_vm_callback_function.py │ │ ├── test_vm_codegen_only.py │ │ ├── test_vm_codegen_tir.py │ │ ├── test_vm_cuda_graph.py │ │ ├── test_vm_execbuilder.py │ │ ├── test_vm_instrument.py │ │ ├── test_vm_multi_device.py │ │ └── test_vm_profiler.py │ ├── runtime │ │ ├── test_evaluator_with_preproc.py │ │ ├── test_executable.py │ │ ├── test_rpc_base.py │ │ ├── test_runtime_container.py │ │ ├── test_runtime_device_api.py │ │ ├── test_runtime_dlpack.py │ │ ├── test_runtime_error.py │ │ ├── test_runtime_extension.py │ │ ├── test_runtime_measure.py │ │ ├── test_runtime_module_export.py │ │ ├── test_runtime_module_load.py │ │ ├── test_runtime_module_property.py │ │ ├── test_runtime_nd_array.py │ │ ├── test_runtime_rpc.py │ │ └── test_runtime_trace.py │ ├── target │ │ ├── test_arm_target.py │ │ ├── test_llvm_features_info.py │ │ ├── test_riscv_features.py │ │ ├── test_target_parser_mprofile.py │ │ ├── test_target_target.py │ │ ├── test_virtual_device.py │ │ └── test_x86_features.py │ ├── te │ │ ├── test_te_create_primfunc.py │ │ ├── test_te_tag.py │ │ ├── test_te_tensor.py │ │ └── test_te_verify_compute.py │ ├── testing │ │ ├── test_filter_untracked.py │ │ ├── test_gen_requirements.py │ │ ├── test_testing.py │ │ ├── test_tvm_testing_before_after.py │ │ ├── test_tvm_testing_features.py │ │ └── test_type_annotation_checker.py │ ├── tir-analysis │ │ ├── test_tir_analysis_calculate_allocated_memory.py │ │ ├── test_tir_analysis_detect_buffer_access_lca.py │ │ ├── test_tir_analysis_estimate_tir_flops.py │ │ ├── test_tir_analysis_expr_deep_equal.py │ │ ├── test_tir_analysis_get_block_access_region.py │ │ ├── test_tir_analysis_identify_memcpy.py │ │ ├── test_tir_analysis_is_pure_function.py │ │ ├── test_tir_analysis_oob.py │ │ ├── test_tir_analysis_verify_ssa.py │ │ └── test_tir_analysis_verify_well_formed.py │ ├── tir-base │ │ ├── test_slice_tir.py │ │ ├── test_tir_base.py │ │ ├── test_tir_block_dependence_info.py │ │ ├── test_tir_buffer.py │ │ ├── test_tir_constructor.py │ │ ├── test_tir_data_layout.py │ │ ├── test_tir_host_func.py │ │ ├── test_tir_imm_values.py │ │ ├── test_tir_index_map.py │ │ ├── test_tir_intrin.py │ │ ├── test_tir_nodes.py │ │ ├── test_tir_op_types.py │ │ ├── test_tir_ops.py │ │ ├── test_tir_ptx_cp_async.py │ │ ├── test_tir_ptx_ldmatrix.py │ │ ├── test_tir_ptx_mma.py │ │ ├── test_tir_ptx_mma_sp.py │ │ ├── test_tir_renew_defs.py │ │ ├── test_tir_scalable_datatype.py │ │ ├── test_tir_specialize.py │ │ ├── test_tir_stmt_functor_ir_transform.py │ │ ├── test_tir_stmt_functor_substitute.py │ │ ├── test_tir_structural_equal_hash.py │ │ ├── test_tir_te_extern_primfunc.py │ │ ├── test_tir_texture_scope.py │ │ └── test_tir_unsafe_hide_buffer_access.py │ ├── tir-schedule │ │ ├── test_tir_schedule_analysis.py │ │ ├── test_tir_schedule_annotate_buffer_access.py │ │ ├── test_tir_schedule_block_scope.py │ │ ├── test_tir_schedule_blockize.py │ │ ├── test_tir_schedule_cache_index.py │ │ ├── test_tir_schedule_cache_read_write.py │ │ ├── test_tir_schedule_compute_at.py │ │ ├── test_tir_schedule_compute_inline.py │ │ ├── test_tir_schedule_decompose_padding.py │ │ ├── test_tir_schedule_error.py │ │ ├── test_tir_schedule_for_kind.py │ │ ├── test_tir_schedule_instruction.py │ │ ├── test_tir_schedule_merge.py │ │ ├── test_tir_schedule_pad_einsum.py │ │ ├── test_tir_schedule_partition.py │ │ ├── test_tir_schedule_read_write_at.py │ │ ├── test_tir_schedule_reduction.py │ │ ├── test_tir_schedule_reindex.py │ │ ├── test_tir_schedule_reorder.py │ │ ├── test_tir_schedule_reorder_block_iter_var.py │ │ ├── test_tir_schedule_rfactor.py │ │ ├── test_tir_schedule_rolling_buffer.py │ │ ├── test_tir_schedule_sampling.py │ │ ├── test_tir_schedule_set_axis_separator.py │ │ ├── test_tir_schedule_set_dtype.py │ │ ├── test_tir_schedule_set_scope.py │ │ ├── test_tir_schedule_split_fuse.py │ │ ├── test_tir_schedule_state.py │ │ ├── test_tir_schedule_state_cached_flags.py │ │ ├── test_tir_schedule_storage_align.py │ │ ├── test_tir_schedule_tensorize.py │ │ ├── test_tir_schedule_tensorize_ldmatrix_mma_numeric.py │ │ ├── test_tir_schedule_tensorize_mfma_numeric.py │ │ ├── test_tir_schedule_trace.py │ │ ├── test_tir_schedule_transform.py │ │ ├── test_tir_schedule_transform_layout.py │ │ └── test_tir_schedule_utilities.py │ ├── tir-transform │ │ ├── test_tir_functor.py │ │ ├── test_tir_inline_private_functions.py │ │ ├── test_tir_transform_annotate_device_regions.py │ │ ├── test_tir_transform_annotate_irregular_loop.py │ │ ├── test_tir_transform_bf16_legalize.py │ │ ├── test_tir_transform_combine_context_call.py │ │ ├── test_tir_transform_common_subexpr_elim.py │ │ ├── test_tir_transform_compact_buffer_region.py │ │ ├── test_tir_transform_convert_blocks_to_opaque.py │ │ ├── test_tir_transform_convert_for_loops_serial.py │ │ ├── test_tir_transform_convert_ssa.py │ │ ├── test_tir_transform_decorate_device_scope.py │ │ ├── test_tir_transform_device_kernel_launch.py │ │ ├── test_tir_transform_extract_constants.py │ │ ├── test_tir_transform_flatten_buffer.py │ │ ├── test_tir_transform_force_narrow_index_to_i32.py │ │ ├── test_tir_transform_fp8_legalize.py │ │ ├── test_tir_transform_helpers.py │ │ ├── test_tir_transform_hoist_expression.py │ │ ├── test_tir_transform_hoist_if.py │ │ ├── test_tir_transform_inject_double_buffer.py │ │ ├── test_tir_transform_inject_permuted_layout.py │ │ ├── test_tir_transform_inject_ptx_async_copy.py │ │ ├── test_tir_transform_inject_rolling_buffer.py │ │ ├── test_tir_transform_inject_software_pipeline.py │ │ ├── test_tir_transform_inject_virtual_thread.py │ │ ├── test_tir_transform_lift_thread_binding.py │ │ ├── test_tir_transform_loop_partition.py │ │ ├── test_tir_transform_lower_cross_thread_reduction.py │ │ ├── test_tir_transform_lower_device_storage_access_info.py │ │ ├── test_tir_transform_lower_init_block.py │ │ ├── test_tir_transform_lower_intrin.py │ │ ├── test_tir_transform_lower_match_buffer.py │ │ ├── test_tir_transform_lower_opaque_block.py │ │ ├── test_tir_transform_lower_thread_all_reduce.py │ │ ├── test_tir_transform_lower_tvm_builtin.py │ │ ├── test_tir_transform_make_packed_api.py │ │ ├── test_tir_transform_make_unpacked_api.py │ │ ├── test_tir_transform_manifest_shared_memory_local_stage.py │ │ ├── test_tir_transform_memhammer_lower_auto_copy.py │ │ ├── test_tir_transform_merge_dynamic_shared_memory_allocations.py │ │ ├── test_tir_transform_narrow_datatype.py │ │ ├── test_tir_transform_plan_update_buffer_allocation_location.py │ │ ├── test_tir_transform_pointer_value_type_rewrite.py │ │ ├── test_tir_transform_prim_func_pass.py │ │ ├── test_tir_transform_profiling_instr.py │ │ ├── test_tir_transform_reduce_branching_through_overcompute.py │ │ ├── test_tir_transform_remove_assume.py │ │ ├── test_tir_transform_remove_no_op.py │ │ ├── test_tir_transform_remove_undef.py │ │ ├── test_tir_transform_remove_weight_layout_rewrite_block.py │ │ ├── test_tir_transform_renormalize_split_pattern.py │ │ ├── test_tir_transform_rewrite_unsafe_select.py │ │ ├── test_tir_transform_simplify.py │ │ ├── test_tir_transform_split_host_device.py │ │ ├── test_tir_transform_storage_rewrite.py │ │ ├── test_tir_transform_thread_sync.py │ │ ├── test_tir_transform_unify_thread_binding.py │ │ ├── test_tir_transform_unroll_loop.py │ │ ├── test_tir_transform_vectorize.py │ │ └── test_transform_default_gpu_schedule.py │ └── tvmscript │ │ ├── test_tvmscript_complete.py │ │ ├── test_tvmscript_error_report.py │ │ ├── test_tvmscript_ir_builder_base.py │ │ ├── test_tvmscript_ir_builder_irmodule.py │ │ ├── test_tvmscript_ir_builder_tir.py │ │ ├── test_tvmscript_meta_programming.py │ │ ├── test_tvmscript_ops.py │ │ ├── test_tvmscript_parser_evaluator.py │ │ ├── test_tvmscript_parser_ir.py │ │ ├── test_tvmscript_parser_source.py │ │ ├── test_tvmscript_parser_tir.py │ │ ├── test_tvmscript_printer_annotation.py │ │ ├── test_tvmscript_printer_doc.py │ │ ├── test_tvmscript_printer_highlight.py │ │ ├── test_tvmscript_printer_ir.py │ │ ├── test_tvmscript_printer_metadata.py │ │ ├── test_tvmscript_printer_python_doc_printer.py │ │ ├── test_tvmscript_printer_structural_equal.py │ │ ├── test_tvmscript_printer_tir.py │ │ ├── test_tvmscript_printer_underlining.py │ │ ├── test_tvmscript_regression.py │ │ ├── test_tvmscript_roundtrip.py │ │ ├── test_tvmscript_syntax_sugar.py │ │ └── test_tvmscript_type.py └── scripts │ ├── ci.py │ ├── mlc │ ├── task_mlc_build.sh │ └── task_mlc_lint_cleanup.sh │ ├── release │ ├── .gitignore │ ├── README.md │ ├── gather_prs.py │ ├── list_rfcs.py │ ├── make_notes.py │ └── test_release_package.sh │ ├── request_hook │ └── request_hook.py │ ├── setup-adreno-env.sh │ ├── setup-pytest-env.sh │ ├── task_build.py │ ├── task_build_adreno_bins.sh │ ├── task_build_hexagon_api.sh │ ├── task_clean.sh │ ├── task_clear_pytest.sh │ ├── task_config_build_adreno.sh │ ├── task_config_build_arm.sh │ ├── task_config_build_cpu.sh │ ├── task_config_build_gpu.sh │ ├── task_config_build_gpu_other.sh │ ├── task_config_build_gpu_vulkan.sh │ ├── task_config_build_hexagon.sh │ ├── task_config_build_i386.sh │ ├── task_config_build_jvm.sh │ ├── task_config_build_mrvl.sh │ ├── task_config_build_static.sh │ ├── task_config_build_wasm.sh │ ├── task_cpp_unittest.sh │ ├── task_golang.sh │ ├── task_java_unittest.sh │ ├── task_lint.sh │ ├── task_mypy.sh │ ├── task_opencl_cpp_unittest.sh │ ├── task_python_adreno.sh │ ├── task_python_arm_compute_library.sh │ ├── task_python_docs.sh │ ├── task_python_hexagon.sh │ ├── task_python_integration.sh │ ├── task_python_integration_gpuonly.sh │ ├── task_python_integration_i386only.sh │ ├── task_python_nightly.sh │ ├── task_python_topi.sh │ ├── task_python_unittest.sh │ ├── task_python_unittest_gpuonly.sh │ ├── task_rust.sh │ ├── task_show_node_info.sh │ ├── task_web_wasm.sh │ └── unity │ ├── README │ ├── task_extra_lint.sh │ ├── task_python_relax.sh │ └── task_python_relax_gpuonly.sh ├── version.py └── web ├── .eslintignore ├── .eslintrc.json ├── .gitignore ├── Makefile ├── README.md ├── apps ├── browser │ ├── rpc_plugin.html │ └── rpc_server.html └── node │ ├── example.js │ ├── wasi_example.js │ └── wasi_rpc_server.js ├── emcc ├── decorate_as_wasi.py ├── preload.js ├── tvmjs_support.cc ├── wasm_runtime.cc └── webgpu_runtime.cc ├── jest.config.js ├── package-lock.json ├── package.json ├── rollup.config.mjs ├── run_jest.py ├── src ├── artifact_cache.ts ├── asyncify.ts ├── compact.ts ├── ctypes.ts ├── environment.ts ├── index.ts ├── memory.ts ├── rpc_server.ts ├── runtime.ts ├── support.ts ├── tvmjs_runtime_wasi.d.ts ├── types.ts └── webgpu.ts ├── tests ├── node │ ├── test_module_load.js │ ├── test_object.js │ ├── test_packed_func.js │ ├── test_random_generator.js │ ├── test_tensor.js │ └── test_vm.js └── python │ ├── prepare_test_libs.py │ ├── relax_rpc_test.py │ └── webgpu_rpc_test.py ├── tsconfig.json └── typedoc.json /.asf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.asf.yaml -------------------------------------------------------------------------------- /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.clang-format -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/CODEOWNERSHIP: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.github/CODEOWNERSHIP -------------------------------------------------------------------------------- /.github/actions/setup/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.github/actions/setup/action.yml -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/cc_bot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.github/workflows/cc_bot.yml -------------------------------------------------------------------------------- /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.github/workflows/mlc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.github/workflows/mlc.yml -------------------------------------------------------------------------------- /.github/workflows/tag_teams.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.github/workflows/tag_teams.yml -------------------------------------------------------------------------------- /.github/workflows/tvmbot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.github/workflows/tvmbot.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.gitmodules -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /3rdparty/libcrc/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/3rdparty/libcrc/.gitignore -------------------------------------------------------------------------------- /3rdparty/libcrc/src/crcccitt.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/3rdparty/libcrc/src/crcccitt.c -------------------------------------------------------------------------------- /3rdparty/mlperftiny/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/3rdparty/mlperftiny/README.md -------------------------------------------------------------------------------- /3rdparty/mscclpp/include/msccl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/3rdparty/mscclpp/include/msccl.h -------------------------------------------------------------------------------- /3rdparty/picojson/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/3rdparty/picojson/README.md -------------------------------------------------------------------------------- /3rdparty/picojson/picojson.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/3rdparty/picojson/picojson.h -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /CONTRIBUTORS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/CONTRIBUTORS.md -------------------------------------------------------------------------------- /KEYS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/KEYS -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/Makefile -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/README.md -------------------------------------------------------------------------------- /apps/android_rpc/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/android_rpc/.gitignore -------------------------------------------------------------------------------- /apps/android_rpc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/android_rpc/README.md -------------------------------------------------------------------------------- /apps/android_rpc/app/.gitignore: -------------------------------------------------------------------------------- 1 | /build 2 | -------------------------------------------------------------------------------- /apps/android_rpc/build.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/android_rpc/build.gradle -------------------------------------------------------------------------------- /apps/android_rpc/settings.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/android_rpc/settings.gradle -------------------------------------------------------------------------------- /apps/cpp_rpc/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/cpp_rpc/CMakeLists.txt -------------------------------------------------------------------------------- /apps/cpp_rpc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/cpp_rpc/README.md -------------------------------------------------------------------------------- /apps/cpp_rpc/main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/cpp_rpc/main.cc -------------------------------------------------------------------------------- /apps/cpp_rpc/rpc_env.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/cpp_rpc/rpc_env.cc -------------------------------------------------------------------------------- /apps/cpp_rpc/rpc_env.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/cpp_rpc/rpc_env.h -------------------------------------------------------------------------------- /apps/cpp_rpc/rpc_server.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/cpp_rpc/rpc_server.cc -------------------------------------------------------------------------------- /apps/cpp_rpc/rpc_server.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/cpp_rpc/rpc_server.h -------------------------------------------------------------------------------- /apps/cpp_rpc/win32_process.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/cpp_rpc/win32_process.cc -------------------------------------------------------------------------------- /apps/cpp_rpc/win32_process.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/cpp_rpc/win32_process.h -------------------------------------------------------------------------------- /apps/hexagon_api/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/hexagon_api/CMakeLists.txt -------------------------------------------------------------------------------- /apps/hexagon_api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/hexagon_api/README.md -------------------------------------------------------------------------------- /apps/hexagon_launcher/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/hexagon_launcher/README.md -------------------------------------------------------------------------------- /apps/ios_rpc/.gitignore: -------------------------------------------------------------------------------- 1 | rpc_config.txt 2 | -------------------------------------------------------------------------------- /apps/ios_rpc/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/ios_rpc/CMakeLists.txt -------------------------------------------------------------------------------- /apps/ios_rpc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/ios_rpc/README.md -------------------------------------------------------------------------------- /apps/ios_rpc/init_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/ios_rpc/init_proj.py -------------------------------------------------------------------------------- /apps/ios_rpc/tvmrpc/Info.plist: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/ios_rpc/tvmrpc/Info.plist -------------------------------------------------------------------------------- /apps/ios_rpc/tvmrpc/RPCArgs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/ios_rpc/tvmrpc/RPCArgs.h -------------------------------------------------------------------------------- /apps/ios_rpc/tvmrpc/RPCArgs.mm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/ios_rpc/tvmrpc/RPCArgs.mm -------------------------------------------------------------------------------- /apps/ios_rpc/tvmrpc/RPCServer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/ios_rpc/tvmrpc/RPCServer.h -------------------------------------------------------------------------------- /apps/ios_rpc/tvmrpc/RPCServer.mm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/ios_rpc/tvmrpc/RPCServer.mm -------------------------------------------------------------------------------- /apps/ios_rpc/tvmrpc/main.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/apps/ios_rpc/tvmrpc/main.m -------------------------------------------------------------------------------- /ci/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/README.md -------------------------------------------------------------------------------- /ci/jenkins/.gitignore: -------------------------------------------------------------------------------- 1 | /_venv 2 | -------------------------------------------------------------------------------- /ci/jenkins/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/jenkins/README.md -------------------------------------------------------------------------------- /ci/jenkins/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/jenkins/data.py -------------------------------------------------------------------------------- /ci/jenkins/docker-images.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/jenkins/docker-images.ini -------------------------------------------------------------------------------- /ci/jenkins/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/jenkins/generate.py -------------------------------------------------------------------------------- /ci/scripts/github/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/scripts/github/__init__.py -------------------------------------------------------------------------------- /ci/scripts/jenkins/check_pr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/scripts/jenkins/check_pr.py -------------------------------------------------------------------------------- /ci/scripts/jenkins/cmd_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/scripts/jenkins/cmd_utils.py -------------------------------------------------------------------------------- /ci/scripts/jenkins/git_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/scripts/jenkins/git_utils.py -------------------------------------------------------------------------------- /ci/scripts/jenkins/http_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/scripts/jenkins/http_utils.py -------------------------------------------------------------------------------- /ci/scripts/jenkins/pytest_ids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/scripts/jenkins/pytest_ids.py -------------------------------------------------------------------------------- /ci/scripts/jenkins/retry.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/scripts/jenkins/retry.sh -------------------------------------------------------------------------------- /ci/scripts/jenkins/s3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/ci/scripts/jenkins/s3.py -------------------------------------------------------------------------------- /cmake/config.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/config.cmake -------------------------------------------------------------------------------- /cmake/libs/Libbacktrace.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/libs/Libbacktrace.cmake -------------------------------------------------------------------------------- /cmake/modules/CUDA.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/CUDA.cmake -------------------------------------------------------------------------------- /cmake/modules/ClangFlags.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/ClangFlags.cmake -------------------------------------------------------------------------------- /cmake/modules/Git.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/Git.cmake -------------------------------------------------------------------------------- /cmake/modules/Hexagon.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/Hexagon.cmake -------------------------------------------------------------------------------- /cmake/modules/HexagonSDK.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/HexagonSDK.cmake -------------------------------------------------------------------------------- /cmake/modules/LLVM.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/LLVM.cmake -------------------------------------------------------------------------------- /cmake/modules/LibInfo.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/LibInfo.cmake -------------------------------------------------------------------------------- /cmake/modules/Logging.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/Logging.cmake -------------------------------------------------------------------------------- /cmake/modules/Metal.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/Metal.cmake -------------------------------------------------------------------------------- /cmake/modules/OpenCL.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/OpenCL.cmake -------------------------------------------------------------------------------- /cmake/modules/OpenMP.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/OpenMP.cmake -------------------------------------------------------------------------------- /cmake/modules/ROCM.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/ROCM.cmake -------------------------------------------------------------------------------- /cmake/modules/Vulkan.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/Vulkan.cmake -------------------------------------------------------------------------------- /cmake/modules/contrib/AMX.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/contrib/AMX.cmake -------------------------------------------------------------------------------- /cmake/modules/contrib/BLAS.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/contrib/BLAS.cmake -------------------------------------------------------------------------------- /cmake/modules/contrib/BNNS.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/contrib/BNNS.cmake -------------------------------------------------------------------------------- /cmake/modules/contrib/CLML.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/contrib/CLML.cmake -------------------------------------------------------------------------------- /cmake/modules/contrib/DNNL.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/contrib/DNNL.cmake -------------------------------------------------------------------------------- /cmake/modules/contrib/MSC.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/contrib/MSC.cmake -------------------------------------------------------------------------------- /cmake/modules/contrib/Mrvl.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/contrib/Mrvl.cmake -------------------------------------------------------------------------------- /cmake/modules/contrib/PAPI.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/contrib/PAPI.cmake -------------------------------------------------------------------------------- /cmake/modules/contrib/Sort.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/contrib/Sort.cmake -------------------------------------------------------------------------------- /cmake/modules/contrib/vllm.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/modules/contrib/vllm.cmake -------------------------------------------------------------------------------- /cmake/utils/CCache.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/CCache.cmake -------------------------------------------------------------------------------- /cmake/utils/CRTConfig.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/CRTConfig.cmake -------------------------------------------------------------------------------- /cmake/utils/FindCUDA.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/FindCUDA.cmake -------------------------------------------------------------------------------- /cmake/utils/FindLLVM.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/FindLLVM.cmake -------------------------------------------------------------------------------- /cmake/utils/FindNCCL.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/FindNCCL.cmake -------------------------------------------------------------------------------- /cmake/utils/FindNVSHMEM.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/FindNVSHMEM.cmake -------------------------------------------------------------------------------- /cmake/utils/FindOpenCL.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/FindOpenCL.cmake -------------------------------------------------------------------------------- /cmake/utils/FindRCCL.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/FindRCCL.cmake -------------------------------------------------------------------------------- /cmake/utils/FindROCM.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/FindROCM.cmake -------------------------------------------------------------------------------- /cmake/utils/FindVulkan.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/FindVulkan.cmake -------------------------------------------------------------------------------- /cmake/utils/Linker.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/Linker.cmake -------------------------------------------------------------------------------- /cmake/utils/Summary.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/Summary.cmake -------------------------------------------------------------------------------- /cmake/utils/Utils.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/cmake/utils/Utils.cmake -------------------------------------------------------------------------------- /conda/Dockerfile.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/Dockerfile.template -------------------------------------------------------------------------------- /conda/build-environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/build-environment.yaml -------------------------------------------------------------------------------- /conda/build_cpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/build_cpu.sh -------------------------------------------------------------------------------- /conda/build_cuda.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/build_cuda.sh -------------------------------------------------------------------------------- /conda/build_win.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/build_win.bat -------------------------------------------------------------------------------- /conda/condarc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/condarc -------------------------------------------------------------------------------- /conda/recipe/bld.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/recipe/bld.bat -------------------------------------------------------------------------------- /conda/recipe/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/recipe/build.sh -------------------------------------------------------------------------------- /conda/recipe/cross-linux.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/recipe/cross-linux.cmake -------------------------------------------------------------------------------- /conda/recipe/install_libtvm.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/recipe/install_libtvm.bat -------------------------------------------------------------------------------- /conda/recipe/install_libtvm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/recipe/install_libtvm.sh -------------------------------------------------------------------------------- /conda/recipe/meta.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/recipe/meta.yaml -------------------------------------------------------------------------------- /conda/render_cuda_dockerfiles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conda/render_cuda_dockerfiles.py -------------------------------------------------------------------------------- /configs/host/default.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/configs/host/default.json -------------------------------------------------------------------------------- /conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/conftest.py -------------------------------------------------------------------------------- /docker/Dockerfile.ci_adreno: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.ci_adreno -------------------------------------------------------------------------------- /docker/Dockerfile.ci_arm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.ci_arm -------------------------------------------------------------------------------- /docker/Dockerfile.ci_cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.ci_cpu -------------------------------------------------------------------------------- /docker/Dockerfile.ci_gpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.ci_gpu -------------------------------------------------------------------------------- /docker/Dockerfile.ci_hexagon: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.ci_hexagon -------------------------------------------------------------------------------- /docker/Dockerfile.ci_i386: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.ci_i386 -------------------------------------------------------------------------------- /docker/Dockerfile.ci_jekyll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.ci_jekyll -------------------------------------------------------------------------------- /docker/Dockerfile.ci_lint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.ci_lint -------------------------------------------------------------------------------- /docker/Dockerfile.ci_wasm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.ci_wasm -------------------------------------------------------------------------------- /docker/Dockerfile.conda_cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.conda_cpu -------------------------------------------------------------------------------- /docker/Dockerfile.conda_cuda100: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.conda_cuda100 -------------------------------------------------------------------------------- /docker/Dockerfile.conda_cuda90: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.conda_cuda90 -------------------------------------------------------------------------------- /docker/Dockerfile.docs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/Dockerfile.docs -------------------------------------------------------------------------------- /docker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/README.md -------------------------------------------------------------------------------- /docker/bash.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/bash.sh -------------------------------------------------------------------------------- /docker/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/build.sh -------------------------------------------------------------------------------- /docker/clear-stale-images.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/clear-stale-images.sh -------------------------------------------------------------------------------- /docker/dev_common.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/dev_common.sh -------------------------------------------------------------------------------- /docker/lint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/lint.sh -------------------------------------------------------------------------------- /docker/python/bootstrap/.gitignore: -------------------------------------------------------------------------------- 1 | /_venv 2 | -------------------------------------------------------------------------------- /docker/python/bootstrap/lockfiles/requirements-3.7.txt: -------------------------------------------------------------------------------- 1 | pip 2 | poetry 3 | setuptools 4 | -------------------------------------------------------------------------------- /docker/python/bootstrap/lockfiles/requirements-3.8.txt: -------------------------------------------------------------------------------- 1 | pip 2 | poetry 3 | setuptools 4 | -------------------------------------------------------------------------------- /docker/python/bootstrap/lockfiles/requirements-3.9.txt: -------------------------------------------------------------------------------- 1 | pip 2 | poetry 3 | setuptools 4 | -------------------------------------------------------------------------------- /docker/python/ci-constraints.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/python/ci-constraints.txt -------------------------------------------------------------------------------- /docker/with_the_same_user: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docker/with_the_same_user -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- 1 | doxygen 2 | modules 3 | -------------------------------------------------------------------------------- /docs/Doxyfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/Doxyfile -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/_static/css/tvm_theme.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/_static/css/tvm_theme.css -------------------------------------------------------------------------------- /docs/_static/img/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/_static/img/README -------------------------------------------------------------------------------- /docs/arch/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/arch/index.rst -------------------------------------------------------------------------------- /docs/arch/pass_infra.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/arch/pass_infra.rst -------------------------------------------------------------------------------- /docs/arch/runtime.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/arch/runtime.rst -------------------------------------------------------------------------------- /docs/arch/runtimes/vulkan.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/arch/runtimes/vulkan.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/contribute/ci.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/contribute/ci.rst -------------------------------------------------------------------------------- /docs/contribute/code_guide.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/contribute/code_guide.rst -------------------------------------------------------------------------------- /docs/contribute/code_review.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/contribute/code_review.rst -------------------------------------------------------------------------------- /docs/contribute/community.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/contribute/community.rst -------------------------------------------------------------------------------- /docs/contribute/document.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/contribute/document.rst -------------------------------------------------------------------------------- /docs/contribute/git_howto.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/contribute/git_howto.rst -------------------------------------------------------------------------------- /docs/contribute/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/contribute/index.rst -------------------------------------------------------------------------------- /docs/contribute/pull_request.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/contribute/pull_request.rst -------------------------------------------------------------------------------- /docs/deep_dive/relax/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/deep_dive/relax/index.rst -------------------------------------------------------------------------------- /docs/download_3rdparty_embeds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/download_3rdparty_embeds.py -------------------------------------------------------------------------------- /docs/errors.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/errors.rst -------------------------------------------------------------------------------- /docs/genindex.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/genindex.rst -------------------------------------------------------------------------------- /docs/get_started/overview.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/get_started/overview.rst -------------------------------------------------------------------------------- /docs/how_to/dev/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/how_to/dev/index.rst -------------------------------------------------------------------------------- /docs/how_to/tutorials/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/how_to/tutorials/README.txt -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/install/docker.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/install/docker.rst -------------------------------------------------------------------------------- /docs/install/from_source.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/install/from_source.rst -------------------------------------------------------------------------------- /docs/install/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/install/index.rst -------------------------------------------------------------------------------- /docs/reference/api/links.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/reference/api/links.rst -------------------------------------------------------------------------------- /docs/reference/api/python/ir.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/reference/api/python/ir.rst -------------------------------------------------------------------------------- /docs/reference/api/python/te.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/reference/api/python/te.rst -------------------------------------------------------------------------------- /docs/reference/publications.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/reference/publications.rst -------------------------------------------------------------------------------- /docs/reference/security.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/docs/reference/security.rst -------------------------------------------------------------------------------- /include/tvm/arith/analyzer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/arith/analyzer.h -------------------------------------------------------------------------------- /include/tvm/arith/bound.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/arith/bound.h -------------------------------------------------------------------------------- /include/tvm/arith/int_set.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/arith/int_set.h -------------------------------------------------------------------------------- /include/tvm/arith/int_solver.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/arith/int_solver.h -------------------------------------------------------------------------------- /include/tvm/arith/pattern.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/arith/pattern.h -------------------------------------------------------------------------------- /include/tvm/ir/analysis.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/analysis.h -------------------------------------------------------------------------------- /include/tvm/ir/attrs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/attrs.h -------------------------------------------------------------------------------- /include/tvm/ir/diagnostic.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/diagnostic.h -------------------------------------------------------------------------------- /include/tvm/ir/env_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/env_func.h -------------------------------------------------------------------------------- /include/tvm/ir/expr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/expr.h -------------------------------------------------------------------------------- /include/tvm/ir/function.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/function.h -------------------------------------------------------------------------------- /include/tvm/ir/global_info.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/global_info.h -------------------------------------------------------------------------------- /include/tvm/ir/instrument.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/instrument.h -------------------------------------------------------------------------------- /include/tvm/ir/module.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/module.h -------------------------------------------------------------------------------- /include/tvm/ir/name_supply.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/name_supply.h -------------------------------------------------------------------------------- /include/tvm/ir/op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/op.h -------------------------------------------------------------------------------- /include/tvm/ir/source_map.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/source_map.h -------------------------------------------------------------------------------- /include/tvm/ir/transform.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/transform.h -------------------------------------------------------------------------------- /include/tvm/ir/type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/type.h -------------------------------------------------------------------------------- /include/tvm/ir/type_functor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/ir/type_functor.h -------------------------------------------------------------------------------- /include/tvm/meta_schedule/schedule/cpu/.gitignore: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /include/tvm/meta_schedule/schedule/x86/.gitignore: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /include/tvm/node/cast.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/node/cast.h -------------------------------------------------------------------------------- /include/tvm/node/functor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/node/functor.h -------------------------------------------------------------------------------- /include/tvm/node/node.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/node/node.h -------------------------------------------------------------------------------- /include/tvm/node/reflection.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/node/reflection.h -------------------------------------------------------------------------------- /include/tvm/node/repr_printer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/node/repr_printer.h -------------------------------------------------------------------------------- /include/tvm/node/serialization.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/node/serialization.h -------------------------------------------------------------------------------- /include/tvm/relax/analysis.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/analysis.h -------------------------------------------------------------------------------- /include/tvm/relax/attrs/ccl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/attrs/ccl.h -------------------------------------------------------------------------------- /include/tvm/relax/attrs/create.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/attrs/create.h -------------------------------------------------------------------------------- /include/tvm/relax/attrs/image.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/attrs/image.h -------------------------------------------------------------------------------- /include/tvm/relax/attrs/index.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/attrs/index.h -------------------------------------------------------------------------------- /include/tvm/relax/attrs/nn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/attrs/nn.h -------------------------------------------------------------------------------- /include/tvm/relax/attrs/op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/attrs/op.h -------------------------------------------------------------------------------- /include/tvm/relax/attrs/qdq.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/attrs/qdq.h -------------------------------------------------------------------------------- /include/tvm/relax/attrs/search.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/attrs/search.h -------------------------------------------------------------------------------- /include/tvm/relax/attrs/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/attrs/vision.h -------------------------------------------------------------------------------- /include/tvm/relax/backend.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/backend.h -------------------------------------------------------------------------------- /include/tvm/relax/exec_builder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/exec_builder.h -------------------------------------------------------------------------------- /include/tvm/relax/expr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/expr.h -------------------------------------------------------------------------------- /include/tvm/relax/expr_functor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/expr_functor.h -------------------------------------------------------------------------------- /include/tvm/relax/nested_msg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/nested_msg.h -------------------------------------------------------------------------------- /include/tvm/relax/struct_info.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/struct_info.h -------------------------------------------------------------------------------- /include/tvm/relax/tir_pattern.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/tir_pattern.h -------------------------------------------------------------------------------- /include/tvm/relax/transform.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/transform.h -------------------------------------------------------------------------------- /include/tvm/relax/type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/type.h -------------------------------------------------------------------------------- /include/tvm/relax/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/relax/utils.h -------------------------------------------------------------------------------- /include/tvm/runtime/base.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/base.h -------------------------------------------------------------------------------- /include/tvm/runtime/data_type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/data_type.h -------------------------------------------------------------------------------- /include/tvm/runtime/device_api.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/device_api.h -------------------------------------------------------------------------------- /include/tvm/runtime/int_tuple.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/int_tuple.h -------------------------------------------------------------------------------- /include/tvm/runtime/logging.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/logging.h -------------------------------------------------------------------------------- /include/tvm/runtime/module.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/module.h -------------------------------------------------------------------------------- /include/tvm/runtime/nvtx.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/nvtx.h -------------------------------------------------------------------------------- /include/tvm/runtime/object.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/object.h -------------------------------------------------------------------------------- /include/tvm/runtime/profiling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/profiling.h -------------------------------------------------------------------------------- /include/tvm/runtime/serializer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/serializer.h -------------------------------------------------------------------------------- /include/tvm/runtime/tensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/tensor.h -------------------------------------------------------------------------------- /include/tvm/runtime/vm/builtin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/vm/builtin.h -------------------------------------------------------------------------------- /include/tvm/runtime/vm/vm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/runtime/vm/vm.h -------------------------------------------------------------------------------- /include/tvm/script/printer/doc.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/script/printer/doc.h -------------------------------------------------------------------------------- /include/tvm/support/with.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/support/with.h -------------------------------------------------------------------------------- /include/tvm/target/codegen.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/target/codegen.h -------------------------------------------------------------------------------- /include/tvm/target/tag.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/target/tag.h -------------------------------------------------------------------------------- /include/tvm/target/target.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/target/target.h -------------------------------------------------------------------------------- /include/tvm/target/target_info.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/target/target_info.h -------------------------------------------------------------------------------- /include/tvm/target/target_kind.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/target/target_kind.h -------------------------------------------------------------------------------- /include/tvm/te/operation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/te/operation.h -------------------------------------------------------------------------------- /include/tvm/te/tensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/te/tensor.h -------------------------------------------------------------------------------- /include/tvm/tir/analysis.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/analysis.h -------------------------------------------------------------------------------- /include/tvm/tir/block_scope.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/block_scope.h -------------------------------------------------------------------------------- /include/tvm/tir/buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/buffer.h -------------------------------------------------------------------------------- /include/tvm/tir/builtin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/builtin.h -------------------------------------------------------------------------------- /include/tvm/tir/data_layout.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/data_layout.h -------------------------------------------------------------------------------- /include/tvm/tir/expr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/expr.h -------------------------------------------------------------------------------- /include/tvm/tir/expr_functor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/expr_functor.h -------------------------------------------------------------------------------- /include/tvm/tir/function.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/function.h -------------------------------------------------------------------------------- /include/tvm/tir/index_map.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/index_map.h -------------------------------------------------------------------------------- /include/tvm/tir/op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/op.h -------------------------------------------------------------------------------- /include/tvm/tir/op_attr_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/op_attr_types.h -------------------------------------------------------------------------------- /include/tvm/tir/schedule/state.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/schedule/state.h -------------------------------------------------------------------------------- /include/tvm/tir/schedule/trace.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/schedule/trace.h -------------------------------------------------------------------------------- /include/tvm/tir/stmt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/stmt.h -------------------------------------------------------------------------------- /include/tvm/tir/stmt_functor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/stmt_functor.h -------------------------------------------------------------------------------- /include/tvm/tir/transform.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/transform.h -------------------------------------------------------------------------------- /include/tvm/tir/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/utils.h -------------------------------------------------------------------------------- /include/tvm/tir/var.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/tir/var.h -------------------------------------------------------------------------------- /include/tvm/topi/broadcast.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/broadcast.h -------------------------------------------------------------------------------- /include/tvm/topi/detail/extern.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/detail/extern.h -------------------------------------------------------------------------------- /include/tvm/topi/detail/fuse.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/detail/fuse.h -------------------------------------------------------------------------------- /include/tvm/topi/einsum.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/einsum.h -------------------------------------------------------------------------------- /include/tvm/topi/elemwise.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/elemwise.h -------------------------------------------------------------------------------- /include/tvm/topi/nn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/bias_add.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/bias_add.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/bnn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/bnn.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/dense.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/dense.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/dilate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/dilate.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/flatten.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/flatten.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/group_norm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/group_norm.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/layer_norm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/layer_norm.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/mapping.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/mapping.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/pooling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/pooling.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/rms_norm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/rms_norm.h -------------------------------------------------------------------------------- /include/tvm/topi/nn/softmax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/nn/softmax.h -------------------------------------------------------------------------------- /include/tvm/topi/reduction.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/reduction.h -------------------------------------------------------------------------------- /include/tvm/topi/tags.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/tags.h -------------------------------------------------------------------------------- /include/tvm/topi/transform.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/transform.h -------------------------------------------------------------------------------- /include/tvm/topi/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/utils.h -------------------------------------------------------------------------------- /include/tvm/topi/vision/reorg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/include/tvm/topi/vision/reorg.h -------------------------------------------------------------------------------- /jvm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/jvm/README.md -------------------------------------------------------------------------------- /jvm/assembly/osx-x86_64/pom.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/jvm/assembly/osx-x86_64/pom.xml -------------------------------------------------------------------------------- /jvm/assembly/pom.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/jvm/assembly/pom.xml -------------------------------------------------------------------------------- /jvm/conf/google_checks.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/jvm/conf/google_checks.xml -------------------------------------------------------------------------------- /jvm/conf/log4j.properties: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/jvm/conf/log4j.properties -------------------------------------------------------------------------------- /jvm/core/pom.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/jvm/core/pom.xml -------------------------------------------------------------------------------- /jvm/native/linux-x86_64/pom.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/jvm/native/linux-x86_64/pom.xml -------------------------------------------------------------------------------- /jvm/native/osx-x86_64/pom.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/jvm/native/osx-x86_64/pom.xml -------------------------------------------------------------------------------- /jvm/native/pom.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/jvm/native/pom.xml -------------------------------------------------------------------------------- /jvm/pom.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/jvm/pom.xml -------------------------------------------------------------------------------- /licenses/LICENSE.cma.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/licenses/LICENSE.cma.txt -------------------------------------------------------------------------------- /licenses/LICENSE.cnpy.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/licenses/LICENSE.cnpy.txt -------------------------------------------------------------------------------- /licenses/LICENSE.cutlass.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/licenses/LICENSE.cutlass.txt -------------------------------------------------------------------------------- /licenses/LICENSE.libcrc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/licenses/LICENSE.libcrc.txt -------------------------------------------------------------------------------- /licenses/LICENSE.picojson.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/licenses/LICENSE.picojson.txt -------------------------------------------------------------------------------- /licenses/LICENSE.rang.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/licenses/LICENSE.rang.txt -------------------------------------------------------------------------------- /licenses/LICENSE.vllm.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/licenses/LICENSE.vllm.txt -------------------------------------------------------------------------------- /licenses/LICENSE.zlib.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/licenses/LICENSE.zlib.txt -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/mypy.ini -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/pyproject.toml -------------------------------------------------------------------------------- /python/.gitignore: -------------------------------------------------------------------------------- 1 | build 2 | dist 3 | *.cpp 4 | requirements/*.txt 5 | -------------------------------------------------------------------------------- /python/gen_requirements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/gen_requirements.py -------------------------------------------------------------------------------- /python/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/pyproject.toml -------------------------------------------------------------------------------- /python/tvm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/__init__.py -------------------------------------------------------------------------------- /python/tvm/arith/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/arith/__init__.py -------------------------------------------------------------------------------- /python/tvm/arith/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/arith/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/arith/analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/arith/analyzer.py -------------------------------------------------------------------------------- /python/tvm/arith/bound.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/arith/bound.py -------------------------------------------------------------------------------- /python/tvm/arith/int_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/arith/int_set.py -------------------------------------------------------------------------------- /python/tvm/arith/int_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/arith/int_solver.py -------------------------------------------------------------------------------- /python/tvm/arith/pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/arith/pattern.py -------------------------------------------------------------------------------- /python/tvm/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/base.py -------------------------------------------------------------------------------- /python/tvm/contrib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/__init__.py -------------------------------------------------------------------------------- /python/tvm/contrib/cblas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/cblas.py -------------------------------------------------------------------------------- /python/tvm/contrib/cc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/cc.py -------------------------------------------------------------------------------- /python/tvm/contrib/clang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/clang.py -------------------------------------------------------------------------------- /python/tvm/contrib/cublas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/cublas.py -------------------------------------------------------------------------------- /python/tvm/contrib/cublaslt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/cublaslt.py -------------------------------------------------------------------------------- /python/tvm/contrib/cudnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/cudnn.py -------------------------------------------------------------------------------- /python/tvm/contrib/dlpack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/dlpack.py -------------------------------------------------------------------------------- /python/tvm/contrib/dnnl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/dnnl.py -------------------------------------------------------------------------------- /python/tvm/contrib/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/download.py -------------------------------------------------------------------------------- /python/tvm/contrib/emcc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/emcc.py -------------------------------------------------------------------------------- /python/tvm/contrib/hipblas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/hipblas.py -------------------------------------------------------------------------------- /python/tvm/contrib/miopen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/miopen.py -------------------------------------------------------------------------------- /python/tvm/contrib/mkl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/mkl.py -------------------------------------------------------------------------------- /python/tvm/contrib/mps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/mps.py -------------------------------------------------------------------------------- /python/tvm/contrib/mrvl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/mrvl.py -------------------------------------------------------------------------------- /python/tvm/contrib/ndk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/ndk.py -------------------------------------------------------------------------------- /python/tvm/contrib/nnpack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/nnpack.py -------------------------------------------------------------------------------- /python/tvm/contrib/nvcc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/nvcc.py -------------------------------------------------------------------------------- /python/tvm/contrib/popen_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/popen_pool.py -------------------------------------------------------------------------------- /python/tvm/contrib/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/random.py -------------------------------------------------------------------------------- /python/tvm/contrib/rocblas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/rocblas.py -------------------------------------------------------------------------------- /python/tvm/contrib/rocm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/rocm.py -------------------------------------------------------------------------------- /python/tvm/contrib/rpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/rpc.py -------------------------------------------------------------------------------- /python/tvm/contrib/spirv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/spirv.py -------------------------------------------------------------------------------- /python/tvm/contrib/tar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/tar.py -------------------------------------------------------------------------------- /python/tvm/contrib/thrust.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/thrust.py -------------------------------------------------------------------------------- /python/tvm/contrib/tvmjs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/tvmjs.py -------------------------------------------------------------------------------- /python/tvm/contrib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/utils.py -------------------------------------------------------------------------------- /python/tvm/contrib/xcode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/contrib/xcode.py -------------------------------------------------------------------------------- /python/tvm/dlight/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/dlight/__init__.py -------------------------------------------------------------------------------- /python/tvm/dlight/base/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/dlight/base/utils.py -------------------------------------------------------------------------------- /python/tvm/dlight/cpu/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/dlight/cpu/base.py -------------------------------------------------------------------------------- /python/tvm/dlight/cpu/gemv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/dlight/cpu/gemv.py -------------------------------------------------------------------------------- /python/tvm/dlight/gpu/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/dlight/gpu/base.py -------------------------------------------------------------------------------- /python/tvm/dlight/gpu/gemv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/dlight/gpu/gemv.py -------------------------------------------------------------------------------- /python/tvm/dlight/gpu/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/dlight/gpu/matmul.py -------------------------------------------------------------------------------- /python/tvm/dlight/gpu/rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/dlight/gpu/rmsnorm.py -------------------------------------------------------------------------------- /python/tvm/driver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/driver/__init__.py -------------------------------------------------------------------------------- /python/tvm/driver/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/driver/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/error.py -------------------------------------------------------------------------------- /python/tvm/exec/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/exec/__init__.py -------------------------------------------------------------------------------- /python/tvm/exec/disco_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/exec/disco_worker.py -------------------------------------------------------------------------------- /python/tvm/exec/popen_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/exec/popen_worker.py -------------------------------------------------------------------------------- /python/tvm/exec/rpc_proxy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/exec/rpc_proxy.py -------------------------------------------------------------------------------- /python/tvm/exec/rpc_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/exec/rpc_server.py -------------------------------------------------------------------------------- /python/tvm/exec/rpc_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/exec/rpc_tracker.py -------------------------------------------------------------------------------- /python/tvm/ffi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ffi.py -------------------------------------------------------------------------------- /python/tvm/ir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/__init__.py -------------------------------------------------------------------------------- /python/tvm/ir/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/ir/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/analysis.py -------------------------------------------------------------------------------- /python/tvm/ir/attrs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/attrs.py -------------------------------------------------------------------------------- /python/tvm/ir/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/base.py -------------------------------------------------------------------------------- /python/tvm/ir/container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/container.py -------------------------------------------------------------------------------- /python/tvm/ir/expr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/expr.py -------------------------------------------------------------------------------- /python/tvm/ir/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/function.py -------------------------------------------------------------------------------- /python/tvm/ir/global_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/global_info.py -------------------------------------------------------------------------------- /python/tvm/ir/instrument.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/instrument.py -------------------------------------------------------------------------------- /python/tvm/ir/json_compact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/json_compact.py -------------------------------------------------------------------------------- /python/tvm/ir/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/module.py -------------------------------------------------------------------------------- /python/tvm/ir/op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/op.py -------------------------------------------------------------------------------- /python/tvm/ir/supply.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/supply.py -------------------------------------------------------------------------------- /python/tvm/ir/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/transform.py -------------------------------------------------------------------------------- /python/tvm/ir/type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/type.py -------------------------------------------------------------------------------- /python/tvm/ir/type_relation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/ir/type_relation.py -------------------------------------------------------------------------------- /python/tvm/libinfo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/libinfo.py -------------------------------------------------------------------------------- /python/tvm/meta_schedule/tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/meta_schedule/tune.py -------------------------------------------------------------------------------- /python/tvm/relax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/__init__.py -------------------------------------------------------------------------------- /python/tvm/relax/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/relax/dpl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/dpl/__init__.py -------------------------------------------------------------------------------- /python/tvm/relax/dpl/_ffi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/dpl/_ffi.py -------------------------------------------------------------------------------- /python/tvm/relax/dpl/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/dpl/context.py -------------------------------------------------------------------------------- /python/tvm/relax/dpl/pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/dpl/pattern.py -------------------------------------------------------------------------------- /python/tvm/relax/dpl/rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/dpl/rewrite.py -------------------------------------------------------------------------------- /python/tvm/relax/exec_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/exec_builder.py -------------------------------------------------------------------------------- /python/tvm/relax/expr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/expr.py -------------------------------------------------------------------------------- /python/tvm/relax/expr_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/expr_functor.py -------------------------------------------------------------------------------- /python/tvm/relax/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/__init__.py -------------------------------------------------------------------------------- /python/tvm/relax/op/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/relax/op/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/base.py -------------------------------------------------------------------------------- /python/tvm/relax/op/binary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/binary.py -------------------------------------------------------------------------------- /python/tvm/relax/op/ccl/ccl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/ccl/ccl.py -------------------------------------------------------------------------------- /python/tvm/relax/op/create.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/create.py -------------------------------------------------------------------------------- /python/tvm/relax/op/datatype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/datatype.py -------------------------------------------------------------------------------- /python/tvm/relax/op/grad/grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/grad/grad.py -------------------------------------------------------------------------------- /python/tvm/relax/op/index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/index.py -------------------------------------------------------------------------------- /python/tvm/relax/op/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/mask.py -------------------------------------------------------------------------------- /python/tvm/relax/op/nn/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/nn/nn.py -------------------------------------------------------------------------------- /python/tvm/relax/op/op_attrs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/op_attrs.py -------------------------------------------------------------------------------- /python/tvm/relax/op/qdq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/qdq.py -------------------------------------------------------------------------------- /python/tvm/relax/op/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/sampling.py -------------------------------------------------------------------------------- /python/tvm/relax/op/search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/search.py -------------------------------------------------------------------------------- /python/tvm/relax/op/set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/set.py -------------------------------------------------------------------------------- /python/tvm/relax/op/sorting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/sorting.py -------------------------------------------------------------------------------- /python/tvm/relax/op/ternary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/ternary.py -------------------------------------------------------------------------------- /python/tvm/relax/op/unary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/unary.py -------------------------------------------------------------------------------- /python/tvm/relax/op/vm/vm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/op/vm/vm.py -------------------------------------------------------------------------------- /python/tvm/relax/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/pipeline.py -------------------------------------------------------------------------------- /python/tvm/relax/struct_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/struct_info.py -------------------------------------------------------------------------------- /python/tvm/relax/testing/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/testing/nn.py -------------------------------------------------------------------------------- /python/tvm/relax/testing/vm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/testing/vm.py -------------------------------------------------------------------------------- /python/tvm/relax/ty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/ty.py -------------------------------------------------------------------------------- /python/tvm/relax/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/utils.py -------------------------------------------------------------------------------- /python/tvm/relax/vm_build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/relax/vm_build.py -------------------------------------------------------------------------------- /python/tvm/rpc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/rpc/__init__.py -------------------------------------------------------------------------------- /python/tvm/rpc/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/rpc/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/rpc/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/rpc/base.py -------------------------------------------------------------------------------- /python/tvm/rpc/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/rpc/client.py -------------------------------------------------------------------------------- /python/tvm/rpc/minrpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/rpc/minrpc.py -------------------------------------------------------------------------------- /python/tvm/rpc/proxy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/rpc/proxy.py -------------------------------------------------------------------------------- /python/tvm/rpc/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/rpc/server.py -------------------------------------------------------------------------------- /python/tvm/rpc/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/rpc/testing.py -------------------------------------------------------------------------------- /python/tvm/rpc/tornado_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/rpc/tornado_util.py -------------------------------------------------------------------------------- /python/tvm/rpc/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/rpc/tracker.py -------------------------------------------------------------------------------- /python/tvm/runtime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/__init__.py -------------------------------------------------------------------------------- /python/tvm/runtime/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/runtime/_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/_tensor.py -------------------------------------------------------------------------------- /python/tvm/runtime/container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/container.py -------------------------------------------------------------------------------- /python/tvm/runtime/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/device.py -------------------------------------------------------------------------------- /python/tvm/runtime/executable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/executable.py -------------------------------------------------------------------------------- /python/tvm/runtime/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/module.py -------------------------------------------------------------------------------- /python/tvm/runtime/object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/object.py -------------------------------------------------------------------------------- /python/tvm/runtime/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/params.py -------------------------------------------------------------------------------- /python/tvm/runtime/support.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/support.py -------------------------------------------------------------------------------- /python/tvm/runtime/vm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/runtime/vm.py -------------------------------------------------------------------------------- /python/tvm/script/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/script/__init__.py -------------------------------------------------------------------------------- /python/tvm/script/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/script/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/script/highlight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/script/highlight.py -------------------------------------------------------------------------------- /python/tvm/script/printer/doc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/script/printer/doc.py -------------------------------------------------------------------------------- /python/tvm/script/relax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/script/relax.py -------------------------------------------------------------------------------- /python/tvm/script/tir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/script/tir.py -------------------------------------------------------------------------------- /python/tvm/support.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/support.py -------------------------------------------------------------------------------- /python/tvm/target/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/target/__init__.py -------------------------------------------------------------------------------- /python/tvm/target/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/target/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/target/codegen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/target/codegen.py -------------------------------------------------------------------------------- /python/tvm/target/datatype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/target/datatype.py -------------------------------------------------------------------------------- /python/tvm/target/intrin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/target/intrin.py -------------------------------------------------------------------------------- /python/tvm/target/tag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/target/tag.py -------------------------------------------------------------------------------- /python/tvm/target/target.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/target/target.py -------------------------------------------------------------------------------- /python/tvm/target/x86.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/target/x86.py -------------------------------------------------------------------------------- /python/tvm/te/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/te/__init__.py -------------------------------------------------------------------------------- /python/tvm/te/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/te/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/te/operation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/te/operation.py -------------------------------------------------------------------------------- /python/tvm/te/tag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/te/tag.py -------------------------------------------------------------------------------- /python/tvm/te/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/te/tensor.py -------------------------------------------------------------------------------- /python/tvm/testing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/testing/__init__.py -------------------------------------------------------------------------------- /python/tvm/testing/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/testing/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/testing/attrs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/testing/attrs.py -------------------------------------------------------------------------------- /python/tvm/testing/plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/testing/plugin.py -------------------------------------------------------------------------------- /python/tvm/testing/popen_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/testing/popen_pool.py -------------------------------------------------------------------------------- /python/tvm/testing/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/testing/runner.py -------------------------------------------------------------------------------- /python/tvm/testing/tir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/testing/tir.py -------------------------------------------------------------------------------- /python/tvm/testing/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/testing/utils.py -------------------------------------------------------------------------------- /python/tvm/tir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/__init__.py -------------------------------------------------------------------------------- /python/tvm/tir/_ffi_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/_ffi_api.py -------------------------------------------------------------------------------- /python/tvm/tir/block_scope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/block_scope.py -------------------------------------------------------------------------------- /python/tvm/tir/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/buffer.py -------------------------------------------------------------------------------- /python/tvm/tir/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/build.py -------------------------------------------------------------------------------- /python/tvm/tir/data_layout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/data_layout.py -------------------------------------------------------------------------------- /python/tvm/tir/expr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/expr.py -------------------------------------------------------------------------------- /python/tvm/tir/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/function.py -------------------------------------------------------------------------------- /python/tvm/tir/functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/functor.py -------------------------------------------------------------------------------- /python/tvm/tir/generic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/generic.py -------------------------------------------------------------------------------- /python/tvm/tir/ir_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/ir_builder.py -------------------------------------------------------------------------------- /python/tvm/tir/op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/op.py -------------------------------------------------------------------------------- /python/tvm/tir/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/pipeline.py -------------------------------------------------------------------------------- /python/tvm/tir/schedule/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/schedule/state.py -------------------------------------------------------------------------------- /python/tvm/tir/schedule/trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/schedule/trace.py -------------------------------------------------------------------------------- /python/tvm/tir/stmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/stmt.py -------------------------------------------------------------------------------- /python/tvm/tir/stmt_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/tir/stmt_functor.py -------------------------------------------------------------------------------- /python/tvm/topi/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/__init__.py -------------------------------------------------------------------------------- /python/tvm/topi/broadcast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/broadcast.py -------------------------------------------------------------------------------- /python/tvm/topi/cpp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/cpp/__init__.py -------------------------------------------------------------------------------- /python/tvm/topi/cpp/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/cpp/cuda.py -------------------------------------------------------------------------------- /python/tvm/topi/cpp/generic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/cpp/generic.py -------------------------------------------------------------------------------- /python/tvm/topi/cpp/impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/cpp/impl.py -------------------------------------------------------------------------------- /python/tvm/topi/cpp/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/cpp/nn.py -------------------------------------------------------------------------------- /python/tvm/topi/cpp/rocm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/cpp/rocm.py -------------------------------------------------------------------------------- /python/tvm/topi/cpp/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/cpp/utils.py -------------------------------------------------------------------------------- /python/tvm/topi/cpp/x86.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/cpp/x86.py -------------------------------------------------------------------------------- /python/tvm/topi/einsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/einsum.py -------------------------------------------------------------------------------- /python/tvm/topi/gpu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/gpu/__init__.py -------------------------------------------------------------------------------- /python/tvm/topi/gpu/scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/gpu/scan.py -------------------------------------------------------------------------------- /python/tvm/topi/gpu/sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/gpu/sort.py -------------------------------------------------------------------------------- /python/tvm/topi/image/resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/image/resize.py -------------------------------------------------------------------------------- /python/tvm/topi/index_put.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/index_put.py -------------------------------------------------------------------------------- /python/tvm/topi/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/math.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/__init__.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/batch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/batch_norm.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/bnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/bnn.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/conv1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/conv1d.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/conv2d.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/conv3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/conv3d.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/dense.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/dilate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/dilate.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/elemwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/elemwise.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/flatten.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/flatten.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/group_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/group_norm.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/layer_norm.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/loss.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/lstm.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/mapping.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/pad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/pad.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/pooling.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/qnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/qnn.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/rms_norm.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/softmax.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/upsampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/upsampling.py -------------------------------------------------------------------------------- /python/tvm/topi/nn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/nn/utils.py -------------------------------------------------------------------------------- /python/tvm/topi/reduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/reduction.py -------------------------------------------------------------------------------- /python/tvm/topi/scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/scan.py -------------------------------------------------------------------------------- /python/tvm/topi/scatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/scatter.py -------------------------------------------------------------------------------- /python/tvm/topi/searchsorted.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/searchsorted.py -------------------------------------------------------------------------------- /python/tvm/topi/signal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/signal.py -------------------------------------------------------------------------------- /python/tvm/topi/slice_scatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/slice_scatter.py -------------------------------------------------------------------------------- /python/tvm/topi/sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/sort.py -------------------------------------------------------------------------------- /python/tvm/topi/tag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/tag.py -------------------------------------------------------------------------------- /python/tvm/topi/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/tensor.py -------------------------------------------------------------------------------- /python/tvm/topi/testing/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/testing/dense.py -------------------------------------------------------------------------------- /python/tvm/topi/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/transform.py -------------------------------------------------------------------------------- /python/tvm/topi/unique.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/unique.py -------------------------------------------------------------------------------- /python/tvm/topi/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/utils.py -------------------------------------------------------------------------------- /python/tvm/topi/vision/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/python/tvm/topi/vision/nms.py -------------------------------------------------------------------------------- /src/arith/analyzer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/analyzer.cc -------------------------------------------------------------------------------- /src/arith/bound_deducer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/bound_deducer.cc -------------------------------------------------------------------------------- /src/arith/canonical_simplify.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/canonical_simplify.cc -------------------------------------------------------------------------------- /src/arith/const_fold.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/const_fold.h -------------------------------------------------------------------------------- /src/arith/const_int_bound.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/const_int_bound.cc -------------------------------------------------------------------------------- /src/arith/constraint_extract.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/constraint_extract.cc -------------------------------------------------------------------------------- /src/arith/constraint_extract.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/constraint_extract.h -------------------------------------------------------------------------------- /src/arith/domain_touched.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/domain_touched.cc -------------------------------------------------------------------------------- /src/arith/int_constraints.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/int_constraints.cc -------------------------------------------------------------------------------- /src/arith/int_operator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/int_operator.h -------------------------------------------------------------------------------- /src/arith/int_set.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/int_set.cc -------------------------------------------------------------------------------- /src/arith/interval_set.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/interval_set.h -------------------------------------------------------------------------------- /src/arith/iter_affine_map.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/iter_affine_map.cc -------------------------------------------------------------------------------- /src/arith/modular_set.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/modular_set.cc -------------------------------------------------------------------------------- /src/arith/pattern_match.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/pattern_match.h -------------------------------------------------------------------------------- /src/arith/presburger_set.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/presburger_set.cc -------------------------------------------------------------------------------- /src/arith/presburger_set.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/presburger_set.h -------------------------------------------------------------------------------- /src/arith/product_normal_form.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/product_normal_form.h -------------------------------------------------------------------------------- /src/arith/rewrite_simplify.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/rewrite_simplify.cc -------------------------------------------------------------------------------- /src/arith/rewrite_simplify.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/rewrite_simplify.h -------------------------------------------------------------------------------- /src/arith/scalable_expression.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/scalable_expression.cc -------------------------------------------------------------------------------- /src/arith/scalable_expression.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/scalable_expression.h -------------------------------------------------------------------------------- /src/arith/unwrap_vector_expr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/unwrap_vector_expr.cc -------------------------------------------------------------------------------- /src/arith/unwrap_vector_expr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/arith/unwrap_vector_expr.h -------------------------------------------------------------------------------- /src/contrib/msc/core/ir/graph.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/contrib/msc/core/ir/graph.cc -------------------------------------------------------------------------------- /src/contrib/msc/core/ir/graph.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/contrib/msc/core/ir/graph.h -------------------------------------------------------------------------------- /src/contrib/msc/core/ir/plugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/contrib/msc/core/ir/plugin.h -------------------------------------------------------------------------------- /src/contrib/msc/core/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/contrib/msc/core/utils.cc -------------------------------------------------------------------------------- /src/contrib/msc/core/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/contrib/msc/core/utils.h -------------------------------------------------------------------------------- /src/contrib/rust_extension.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/contrib/rust_extension.cc -------------------------------------------------------------------------------- /src/ir/analysis.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/analysis.cc -------------------------------------------------------------------------------- /src/ir/apply_pass_to_function.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/apply_pass_to_function.cc -------------------------------------------------------------------------------- /src/ir/attr_functor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/attr_functor.h -------------------------------------------------------------------------------- /src/ir/attrs.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/attrs.cc -------------------------------------------------------------------------------- /src/ir/diagnostic.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/diagnostic.cc -------------------------------------------------------------------------------- /src/ir/env_func.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/env_func.cc -------------------------------------------------------------------------------- /src/ir/expr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/expr.cc -------------------------------------------------------------------------------- /src/ir/function.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/function.cc -------------------------------------------------------------------------------- /src/ir/global_info.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/global_info.cc -------------------------------------------------------------------------------- /src/ir/global_var_supply.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/global_var_supply.cc -------------------------------------------------------------------------------- /src/ir/instrument.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/instrument.cc -------------------------------------------------------------------------------- /src/ir/module.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/module.cc -------------------------------------------------------------------------------- /src/ir/name_supply.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/name_supply.cc -------------------------------------------------------------------------------- /src/ir/op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/op.cc -------------------------------------------------------------------------------- /src/ir/replace_global_vars.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/replace_global_vars.cc -------------------------------------------------------------------------------- /src/ir/source_map.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/source_map.cc -------------------------------------------------------------------------------- /src/ir/transform.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/transform.cc -------------------------------------------------------------------------------- /src/ir/type.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/type.cc -------------------------------------------------------------------------------- /src/ir/type_functor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/ir/type_functor.cc -------------------------------------------------------------------------------- /src/meta_schedule/arg_info.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/meta_schedule/arg_info.cc -------------------------------------------------------------------------------- /src/meta_schedule/profiler.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/meta_schedule/profiler.cc -------------------------------------------------------------------------------- /src/meta_schedule/schedule/x86/.gitignore: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/meta_schedule/trace_apply.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/meta_schedule/trace_apply.cc -------------------------------------------------------------------------------- /src/meta_schedule/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/meta_schedule/utils.h -------------------------------------------------------------------------------- /src/node/attr_registry.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/node/attr_registry.h -------------------------------------------------------------------------------- /src/node/container_printing.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/node/container_printing.cc -------------------------------------------------------------------------------- /src/node/reflection.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/node/reflection.cc -------------------------------------------------------------------------------- /src/node/repr_printer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/node/repr_printer.cc -------------------------------------------------------------------------------- /src/node/script_printer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/node/script_printer.cc -------------------------------------------------------------------------------- /src/node/serialization.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/node/serialization.cc -------------------------------------------------------------------------------- /src/node/structural_equal.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/node/structural_equal.cc -------------------------------------------------------------------------------- /src/node/structural_hash.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/node/structural_hash.cc -------------------------------------------------------------------------------- /src/relax/analysis/analysis.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/analysis/analysis.cc -------------------------------------------------------------------------------- /src/relax/analysis/udchain.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/analysis/udchain.cc -------------------------------------------------------------------------------- /src/relax/ir/block_builder.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/ir/block_builder.cc -------------------------------------------------------------------------------- /src/relax/ir/emit_te.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/ir/emit_te.cc -------------------------------------------------------------------------------- /src/relax/ir/emit_te.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/ir/emit_te.h -------------------------------------------------------------------------------- /src/relax/ir/expr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/ir/expr.cc -------------------------------------------------------------------------------- /src/relax/ir/expr_functor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/ir/expr_functor.cc -------------------------------------------------------------------------------- /src/relax/ir/struct_info.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/ir/struct_info.cc -------------------------------------------------------------------------------- /src/relax/ir/tir_pattern.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/ir/tir_pattern.cc -------------------------------------------------------------------------------- /src/relax/ir/transform.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/ir/transform.cc -------------------------------------------------------------------------------- /src/relax/ir/type.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/ir/type.cc -------------------------------------------------------------------------------- /src/relax/op/ccl/ccl.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/ccl/ccl.cc -------------------------------------------------------------------------------- /src/relax/op/ccl/ccl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/ccl/ccl.h -------------------------------------------------------------------------------- /src/relax/op/distributed/nn.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/distributed/nn.cc -------------------------------------------------------------------------------- /src/relax/op/distributed/nn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/distributed/nn.h -------------------------------------------------------------------------------- /src/relax/op/distributed/op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/distributed/op.cc -------------------------------------------------------------------------------- /src/relax/op/image/resize.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/image/resize.cc -------------------------------------------------------------------------------- /src/relax/op/image/resize.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/image/resize.h -------------------------------------------------------------------------------- /src/relax/op/memory/view.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/memory/view.cc -------------------------------------------------------------------------------- /src/relax/op/memory/view.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/memory/view.h -------------------------------------------------------------------------------- /src/relax/op/nn/attention.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/nn/attention.cc -------------------------------------------------------------------------------- /src/relax/op/nn/attention.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/nn/attention.h -------------------------------------------------------------------------------- /src/relax/op/nn/convolution.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/nn/convolution.cc -------------------------------------------------------------------------------- /src/relax/op/nn/convolution.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/nn/convolution.h -------------------------------------------------------------------------------- /src/relax/op/nn/nn.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/nn/nn.cc -------------------------------------------------------------------------------- /src/relax/op/nn/nn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/nn/nn.h -------------------------------------------------------------------------------- /src/relax/op/nn/pooling.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/nn/pooling.cc -------------------------------------------------------------------------------- /src/relax/op/nn/pooling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/nn/pooling.h -------------------------------------------------------------------------------- /src/relax/op/op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/op.cc -------------------------------------------------------------------------------- /src/relax/op/op_common.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/op_common.cc -------------------------------------------------------------------------------- /src/relax/op/op_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/op_common.h -------------------------------------------------------------------------------- /src/relax/op/tensor/binary.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/binary.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/binary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/binary.h -------------------------------------------------------------------------------- /src/relax/op/tensor/create.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/create.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/create.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/create.h -------------------------------------------------------------------------------- /src/relax/op/tensor/datatype.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/datatype.h -------------------------------------------------------------------------------- /src/relax/op/tensor/grad.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/grad.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/grad.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/grad.h -------------------------------------------------------------------------------- /src/relax/op/tensor/index.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/index.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/index.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/index.h -------------------------------------------------------------------------------- /src/relax/op/tensor/inspect.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/inspect.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/inspect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/inspect.h -------------------------------------------------------------------------------- /src/relax/op/tensor/qdq.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/qdq.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/qdq.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/qdq.h -------------------------------------------------------------------------------- /src/relax/op/tensor/sampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/sampling.h -------------------------------------------------------------------------------- /src/relax/op/tensor/search.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/search.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/search.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/search.h -------------------------------------------------------------------------------- /src/relax/op/tensor/set.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/set.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/set.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/set.h -------------------------------------------------------------------------------- /src/relax/op/tensor/sorting.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/sorting.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/sorting.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/sorting.h -------------------------------------------------------------------------------- /src/relax/op/tensor/ternary.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/ternary.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/ternary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/ternary.h -------------------------------------------------------------------------------- /src/relax/op/tensor/unary.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/unary.cc -------------------------------------------------------------------------------- /src/relax/op/tensor/unary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/tensor/unary.h -------------------------------------------------------------------------------- /src/relax/op/vision/nms.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/vision/nms.cc -------------------------------------------------------------------------------- /src/relax/op/vision/nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/op/vision/nms.h -------------------------------------------------------------------------------- /src/relax/testing/transform.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/testing/transform.cc -------------------------------------------------------------------------------- /src/relax/training/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/training/utils.cc -------------------------------------------------------------------------------- /src/relax/training/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/training/utils.h -------------------------------------------------------------------------------- /src/relax/transform/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/transform/utils.cc -------------------------------------------------------------------------------- /src/relax/transform/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/transform/utils.h -------------------------------------------------------------------------------- /src/relax/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/relax/utils.cc -------------------------------------------------------------------------------- /src/runtime/builtin_fp16.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/builtin_fp16.cc -------------------------------------------------------------------------------- /src/runtime/cpu_device_api.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/cpu_device_api.cc -------------------------------------------------------------------------------- /src/runtime/cuda/cuda_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/cuda/cuda_common.h -------------------------------------------------------------------------------- /src/runtime/cuda/cuda_module.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/cuda/cuda_module.h -------------------------------------------------------------------------------- /src/runtime/device_api.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/device_api.cc -------------------------------------------------------------------------------- /src/runtime/disco/builtin.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/disco/builtin.cc -------------------------------------------------------------------------------- /src/runtime/disco/loader.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/disco/loader.cc -------------------------------------------------------------------------------- /src/runtime/disco/nccl/nccl.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/disco/nccl/nccl.cc -------------------------------------------------------------------------------- /src/runtime/disco/protocol.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/disco/protocol.h -------------------------------------------------------------------------------- /src/runtime/disco/session.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/disco/session.cc -------------------------------------------------------------------------------- /src/runtime/disco/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/disco/utils.h -------------------------------------------------------------------------------- /src/runtime/file_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/file_utils.cc -------------------------------------------------------------------------------- /src/runtime/file_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/file_utils.h -------------------------------------------------------------------------------- /src/runtime/hexagon/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/hexagon/README.md -------------------------------------------------------------------------------- /src/runtime/logging.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/logging.cc -------------------------------------------------------------------------------- /src/runtime/meta_data.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/meta_data.h -------------------------------------------------------------------------------- /src/runtime/module.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/module.cc -------------------------------------------------------------------------------- /src/runtime/nvtx.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/nvtx.cc -------------------------------------------------------------------------------- /src/runtime/pack_args.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/pack_args.h -------------------------------------------------------------------------------- /src/runtime/profiling.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/profiling.cc -------------------------------------------------------------------------------- /src/runtime/regex.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/regex.cc -------------------------------------------------------------------------------- /src/runtime/regex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/regex.h -------------------------------------------------------------------------------- /src/runtime/rocm/rocm_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/rocm/rocm_common.h -------------------------------------------------------------------------------- /src/runtime/rocm/rocm_module.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/rocm/rocm_module.h -------------------------------------------------------------------------------- /src/runtime/rpc/rpc_channel.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/rpc/rpc_channel.cc -------------------------------------------------------------------------------- /src/runtime/rpc/rpc_channel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/rpc/rpc_channel.h -------------------------------------------------------------------------------- /src/runtime/rpc/rpc_endpoint.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/rpc/rpc_endpoint.h -------------------------------------------------------------------------------- /src/runtime/rpc/rpc_module.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/rpc/rpc_module.cc -------------------------------------------------------------------------------- /src/runtime/rpc/rpc_session.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/rpc/rpc_session.cc -------------------------------------------------------------------------------- /src/runtime/rpc/rpc_session.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/rpc/rpc_session.h -------------------------------------------------------------------------------- /src/runtime/source_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/source_utils.cc -------------------------------------------------------------------------------- /src/runtime/source_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/source_utils.h -------------------------------------------------------------------------------- /src/runtime/static_library.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/static_library.cc -------------------------------------------------------------------------------- /src/runtime/static_library.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/static_library.h -------------------------------------------------------------------------------- /src/runtime/tensor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/tensor.cc -------------------------------------------------------------------------------- /src/runtime/texture.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/texture.h -------------------------------------------------------------------------------- /src/runtime/thread_map.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/thread_map.h -------------------------------------------------------------------------------- /src/runtime/thread_pool.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/thread_pool.cc -------------------------------------------------------------------------------- /src/runtime/vm/attn_backend.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/attn_backend.cc -------------------------------------------------------------------------------- /src/runtime/vm/attn_backend.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/attn_backend.h -------------------------------------------------------------------------------- /src/runtime/vm/attn_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/attn_utils.h -------------------------------------------------------------------------------- /src/runtime/vm/builtin.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/builtin.cc -------------------------------------------------------------------------------- /src/runtime/vm/bytecode.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/bytecode.cc -------------------------------------------------------------------------------- /src/runtime/vm/executable.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/executable.cc -------------------------------------------------------------------------------- /src/runtime/vm/kv_state.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/kv_state.cc -------------------------------------------------------------------------------- /src/runtime/vm/kv_state.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/kv_state.h -------------------------------------------------------------------------------- /src/runtime/vm/lm_support.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/lm_support.cc -------------------------------------------------------------------------------- /src/runtime/vm/rnn_state.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/rnn_state.cc -------------------------------------------------------------------------------- /src/runtime/vm/vm.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vm/vm.cc -------------------------------------------------------------------------------- /src/runtime/vulkan/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/vulkan/README.md -------------------------------------------------------------------------------- /src/runtime/workspace_pool.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/workspace_pool.cc -------------------------------------------------------------------------------- /src/runtime/workspace_pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/runtime/workspace_pool.h -------------------------------------------------------------------------------- /src/script/ir_builder/base.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/ir_builder/base.cc -------------------------------------------------------------------------------- /src/script/ir_builder/ir/ir.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/ir_builder/ir/ir.cc -------------------------------------------------------------------------------- /src/script/printer/doc.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/printer/doc.cc -------------------------------------------------------------------------------- /src/script/printer/ir/ir.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/printer/ir/ir.cc -------------------------------------------------------------------------------- /src/script/printer/ir/misc.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/printer/ir/misc.cc -------------------------------------------------------------------------------- /src/script/printer/ir/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/printer/ir/utils.h -------------------------------------------------------------------------------- /src/script/printer/tir/expr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/printer/tir/expr.cc -------------------------------------------------------------------------------- /src/script/printer/tir/ir.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/printer/tir/ir.cc -------------------------------------------------------------------------------- /src/script/printer/tir/stmt.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/printer/tir/stmt.cc -------------------------------------------------------------------------------- /src/script/printer/tir/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/printer/tir/utils.h -------------------------------------------------------------------------------- /src/script/printer/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/script/printer/utils.h -------------------------------------------------------------------------------- /src/support/arena.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/arena.h -------------------------------------------------------------------------------- /src/support/array.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/array.h -------------------------------------------------------------------------------- /src/support/base64.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/base64.h -------------------------------------------------------------------------------- /src/support/errno_handling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/errno_handling.h -------------------------------------------------------------------------------- /src/support/ffi_testing.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/ffi_testing.cc -------------------------------------------------------------------------------- /src/support/generic_arena.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/generic_arena.h -------------------------------------------------------------------------------- /src/support/hexdump.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/hexdump.cc -------------------------------------------------------------------------------- /src/support/hexdump.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/hexdump.h -------------------------------------------------------------------------------- /src/support/libinfo.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/libinfo.cc -------------------------------------------------------------------------------- /src/support/nd_int_set.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/nd_int_set.h -------------------------------------------------------------------------------- /src/support/ordered_map.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/ordered_map.h -------------------------------------------------------------------------------- /src/support/ordered_set.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/ordered_set.h -------------------------------------------------------------------------------- /src/support/parallel_for.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/parallel_for.cc -------------------------------------------------------------------------------- /src/support/pipe.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/pipe.h -------------------------------------------------------------------------------- /src/support/process_id.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/process_id.h -------------------------------------------------------------------------------- /src/support/ring_buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/ring_buffer.h -------------------------------------------------------------------------------- /src/support/scalars.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/scalars.cc -------------------------------------------------------------------------------- /src/support/scalars.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/scalars.h -------------------------------------------------------------------------------- /src/support/socket.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/socket.h -------------------------------------------------------------------------------- /src/support/ssize.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/ssize.h -------------------------------------------------------------------------------- /src/support/str_escape.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/str_escape.h -------------------------------------------------------------------------------- /src/support/table_printer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/table_printer.h -------------------------------------------------------------------------------- /src/support/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/support/utils.h -------------------------------------------------------------------------------- /src/target/build_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/build_common.h -------------------------------------------------------------------------------- /src/target/codegen.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/codegen.cc -------------------------------------------------------------------------------- /src/target/datatype/registry.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/datatype/registry.h -------------------------------------------------------------------------------- /src/target/intrin_rule.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/intrin_rule.cc -------------------------------------------------------------------------------- /src/target/intrin_rule.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/intrin_rule.h -------------------------------------------------------------------------------- /src/target/llvm/codegen_arm.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/llvm/codegen_arm.cc -------------------------------------------------------------------------------- /src/target/llvm/codegen_blob.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/llvm/codegen_blob.h -------------------------------------------------------------------------------- /src/target/llvm/codegen_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/llvm/codegen_cpu.cc -------------------------------------------------------------------------------- /src/target/llvm/codegen_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/llvm/codegen_cpu.h -------------------------------------------------------------------------------- /src/target/llvm/codegen_llvm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/llvm/codegen_llvm.h -------------------------------------------------------------------------------- /src/target/llvm/llvm_module.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/llvm/llvm_module.cc -------------------------------------------------------------------------------- /src/target/llvm/llvm_module.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/llvm/llvm_module.h -------------------------------------------------------------------------------- /src/target/opt/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/opt/README -------------------------------------------------------------------------------- /src/target/parsers/aprofile.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/parsers/aprofile.cc -------------------------------------------------------------------------------- /src/target/parsers/aprofile.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/parsers/aprofile.h -------------------------------------------------------------------------------- /src/target/parsers/cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/parsers/cpu.cc -------------------------------------------------------------------------------- /src/target/parsers/cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/parsers/cpu.h -------------------------------------------------------------------------------- /src/target/parsers/mprofile.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/parsers/mprofile.cc -------------------------------------------------------------------------------- /src/target/parsers/mprofile.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/parsers/mprofile.h -------------------------------------------------------------------------------- /src/target/source/codegen_c.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/source/codegen_c.cc -------------------------------------------------------------------------------- /src/target/source/codegen_c.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/source/codegen_c.h -------------------------------------------------------------------------------- /src/target/source/ptx.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/source/ptx.cc -------------------------------------------------------------------------------- /src/target/source/ptx.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/source/ptx.h -------------------------------------------------------------------------------- /src/target/spirv/ir_builder.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/spirv/ir_builder.cc -------------------------------------------------------------------------------- /src/target/spirv/ir_builder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/spirv/ir_builder.h -------------------------------------------------------------------------------- /src/target/spirv/spirv_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/spirv/spirv_utils.h -------------------------------------------------------------------------------- /src/target/tag.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/tag.cc -------------------------------------------------------------------------------- /src/target/target.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/target.cc -------------------------------------------------------------------------------- /src/target/target_info.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/target_info.cc -------------------------------------------------------------------------------- /src/target/target_kind.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/target_kind.cc -------------------------------------------------------------------------------- /src/target/virtual_device.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/target/virtual_device.cc -------------------------------------------------------------------------------- /src/te/operation/compute_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/te/operation/compute_op.cc -------------------------------------------------------------------------------- /src/te/operation/extern_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/te/operation/extern_op.cc -------------------------------------------------------------------------------- /src/te/operation/graph.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/te/operation/graph.cc -------------------------------------------------------------------------------- /src/te/operation/graph.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/te/operation/graph.h -------------------------------------------------------------------------------- /src/te/operation/scan_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/te/operation/scan_op.cc -------------------------------------------------------------------------------- /src/te/tensor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/te/tensor.cc -------------------------------------------------------------------------------- /src/tir/analysis/deep_equal.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/analysis/deep_equal.cc -------------------------------------------------------------------------------- /src/tir/analysis/var_touch.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/analysis/var_touch.cc -------------------------------------------------------------------------------- /src/tir/analysis/verify_ssa.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/analysis/verify_ssa.cc -------------------------------------------------------------------------------- /src/tir/ir/block_scope.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/block_scope.cc -------------------------------------------------------------------------------- /src/tir/ir/buffer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/buffer.cc -------------------------------------------------------------------------------- /src/tir/ir/buffer_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/buffer_common.h -------------------------------------------------------------------------------- /src/tir/ir/data_layout.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/data_layout.cc -------------------------------------------------------------------------------- /src/tir/ir/expr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/expr.cc -------------------------------------------------------------------------------- /src/tir/ir/expr_functor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/expr_functor.cc -------------------------------------------------------------------------------- /src/tir/ir/function.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/function.cc -------------------------------------------------------------------------------- /src/tir/ir/functor_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/functor_common.h -------------------------------------------------------------------------------- /src/tir/ir/index_map.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/index_map.cc -------------------------------------------------------------------------------- /src/tir/ir/py_functor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/py_functor.cc -------------------------------------------------------------------------------- /src/tir/ir/specialize.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/specialize.cc -------------------------------------------------------------------------------- /src/tir/ir/stmt.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/stmt.cc -------------------------------------------------------------------------------- /src/tir/ir/stmt_functor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/stmt_functor.cc -------------------------------------------------------------------------------- /src/tir/ir/transform.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/ir/transform.cc -------------------------------------------------------------------------------- /src/tir/op/builtin.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/op/builtin.cc -------------------------------------------------------------------------------- /src/tir/op/op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/op/op.cc -------------------------------------------------------------------------------- /src/tir/op/runtime.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/op/runtime.cc -------------------------------------------------------------------------------- /src/tir/schedule/analysis.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/schedule/analysis.h -------------------------------------------------------------------------------- /src/tir/schedule/error.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/schedule/error.cc -------------------------------------------------------------------------------- /src/tir/schedule/error.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/schedule/error.h -------------------------------------------------------------------------------- /src/tir/schedule/primitive.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/schedule/primitive.h -------------------------------------------------------------------------------- /src/tir/schedule/schedule.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/schedule/schedule.cc -------------------------------------------------------------------------------- /src/tir/schedule/state.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/schedule/state.cc -------------------------------------------------------------------------------- /src/tir/schedule/trace.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/schedule/trace.cc -------------------------------------------------------------------------------- /src/tir/schedule/transform.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/schedule/transform.cc -------------------------------------------------------------------------------- /src/tir/schedule/transform.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/schedule/transform.h -------------------------------------------------------------------------------- /src/tir/schedule/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/schedule/utils.h -------------------------------------------------------------------------------- /src/tir/transforms/ir_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/transforms/ir_utils.cc -------------------------------------------------------------------------------- /src/tir/transforms/ir_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/transforms/ir_utils.h -------------------------------------------------------------------------------- /src/tir/transforms/simplify.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/transforms/simplify.cc -------------------------------------------------------------------------------- /src/tir/transforms/simplify.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/tir/transforms/simplify.h -------------------------------------------------------------------------------- /src/topi/broadcast.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/topi/broadcast.cc -------------------------------------------------------------------------------- /src/topi/einsum.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/topi/einsum.cc -------------------------------------------------------------------------------- /src/topi/elemwise.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/topi/elemwise.cc -------------------------------------------------------------------------------- /src/topi/nn.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/topi/nn.cc -------------------------------------------------------------------------------- /src/topi/reduction.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/topi/reduction.cc -------------------------------------------------------------------------------- /src/topi/transform.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/topi/transform.cc -------------------------------------------------------------------------------- /src/topi/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/topi/utils.cc -------------------------------------------------------------------------------- /src/topi/vision.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/src/topi/vision.cc -------------------------------------------------------------------------------- /tests/cpp/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/.gitignore -------------------------------------------------------------------------------- /tests/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/README.md -------------------------------------------------------------------------------- /tests/cpp/expr_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/expr_test.cc -------------------------------------------------------------------------------- /tests/cpp/ir_functor_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/ir_functor_test.cc -------------------------------------------------------------------------------- /tests/cpp/ndarray_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/ndarray_test.cc -------------------------------------------------------------------------------- /tests/cpp/nested_msg_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/nested_msg_test.cc -------------------------------------------------------------------------------- /tests/cpp/parallel_for_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/parallel_for_test.cc -------------------------------------------------------------------------------- /tests/cpp/support_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/support_test.cc -------------------------------------------------------------------------------- /tests/cpp/target_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/target_test.cc -------------------------------------------------------------------------------- /tests/cpp/te_compute_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/te_compute_test.cc -------------------------------------------------------------------------------- /tests/cpp/topi_ewise_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/cpp/topi_ewise_test.cc -------------------------------------------------------------------------------- /tests/lint/add_asf_header.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/add_asf_header.py -------------------------------------------------------------------------------- /tests/lint/blocklint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/blocklint.sh -------------------------------------------------------------------------------- /tests/lint/check_asf_header.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/check_asf_header.sh -------------------------------------------------------------------------------- /tests/lint/check_file_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/check_file_type.py -------------------------------------------------------------------------------- /tests/lint/cppdocs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/cppdocs.sh -------------------------------------------------------------------------------- /tests/lint/cpplint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/cpplint.sh -------------------------------------------------------------------------------- /tests/lint/docker-format.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/docker-format.sh -------------------------------------------------------------------------------- /tests/lint/filter_untracked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/filter_untracked.py -------------------------------------------------------------------------------- /tests/lint/flake8.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/flake8.sh -------------------------------------------------------------------------------- /tests/lint/git-black.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/git-black.sh -------------------------------------------------------------------------------- /tests/lint/git-clang-format.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/git-clang-format.sh -------------------------------------------------------------------------------- /tests/lint/jnilint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/jnilint.sh -------------------------------------------------------------------------------- /tests/lint/pylint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/pylint.sh -------------------------------------------------------------------------------- /tests/lint/pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/pylintrc -------------------------------------------------------------------------------- /tests/lint/rat-excludes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/rat-excludes -------------------------------------------------------------------------------- /tests/lint/whitespace.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/lint/whitespace.sh -------------------------------------------------------------------------------- /tests/python/ci/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/python/ci/__init__.py -------------------------------------------------------------------------------- /tests/python/ci/test_ci.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/python/ci/test_ci.py -------------------------------------------------------------------------------- /tests/python/ci/test_tvmbot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/python/ci/test_tvmbot.py -------------------------------------------------------------------------------- /tests/python/ci/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/python/ci/test_utils.py -------------------------------------------------------------------------------- /tests/python/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/python/conftest.py -------------------------------------------------------------------------------- /tests/python/disco/test_ccl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/python/disco/test_ccl.py -------------------------------------------------------------------------------- /tests/python/ir/test_dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/python/ir/test_dtype.py -------------------------------------------------------------------------------- /tests/python/relax/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/python/relax/conftest.py -------------------------------------------------------------------------------- /tests/python/te/test_te_tag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/python/te/test_te_tag.py -------------------------------------------------------------------------------- /tests/scripts/ci.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/scripts/ci.py -------------------------------------------------------------------------------- /tests/scripts/task_build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/scripts/task_build.py -------------------------------------------------------------------------------- /tests/scripts/task_clean.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/scripts/task_clean.sh -------------------------------------------------------------------------------- /tests/scripts/task_golang.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/scripts/task_golang.sh -------------------------------------------------------------------------------- /tests/scripts/task_lint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/scripts/task_lint.sh -------------------------------------------------------------------------------- /tests/scripts/task_mypy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/scripts/task_mypy.sh -------------------------------------------------------------------------------- /tests/scripts/task_rust.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/scripts/task_rust.sh -------------------------------------------------------------------------------- /tests/scripts/task_web_wasm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/scripts/task_web_wasm.sh -------------------------------------------------------------------------------- /tests/scripts/unity/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/tests/scripts/unity/README -------------------------------------------------------------------------------- /version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/version.py -------------------------------------------------------------------------------- /web/.eslintignore: -------------------------------------------------------------------------------- 1 | dist 2 | debug 3 | tvmjs_runtime_wasi.js 4 | lib 5 | -------------------------------------------------------------------------------- /web/.eslintrc.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/.eslintrc.json -------------------------------------------------------------------------------- /web/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/.gitignore -------------------------------------------------------------------------------- /web/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/Makefile -------------------------------------------------------------------------------- /web/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/README.md -------------------------------------------------------------------------------- /web/apps/node/example.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/apps/node/example.js -------------------------------------------------------------------------------- /web/apps/node/wasi_example.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/apps/node/wasi_example.js -------------------------------------------------------------------------------- /web/emcc/decorate_as_wasi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/emcc/decorate_as_wasi.py -------------------------------------------------------------------------------- /web/emcc/preload.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/emcc/preload.js -------------------------------------------------------------------------------- /web/emcc/tvmjs_support.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/emcc/tvmjs_support.cc -------------------------------------------------------------------------------- /web/emcc/wasm_runtime.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/emcc/wasm_runtime.cc -------------------------------------------------------------------------------- /web/emcc/webgpu_runtime.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/emcc/webgpu_runtime.cc -------------------------------------------------------------------------------- /web/jest.config.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/jest.config.js -------------------------------------------------------------------------------- /web/package-lock.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/package-lock.json -------------------------------------------------------------------------------- /web/package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/package.json -------------------------------------------------------------------------------- /web/rollup.config.mjs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/rollup.config.mjs -------------------------------------------------------------------------------- /web/run_jest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/run_jest.py -------------------------------------------------------------------------------- /web/src/artifact_cache.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/artifact_cache.ts -------------------------------------------------------------------------------- /web/src/asyncify.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/asyncify.ts -------------------------------------------------------------------------------- /web/src/compact.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/compact.ts -------------------------------------------------------------------------------- /web/src/ctypes.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/ctypes.ts -------------------------------------------------------------------------------- /web/src/environment.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/environment.ts -------------------------------------------------------------------------------- /web/src/index.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/index.ts -------------------------------------------------------------------------------- /web/src/memory.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/memory.ts -------------------------------------------------------------------------------- /web/src/rpc_server.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/rpc_server.ts -------------------------------------------------------------------------------- /web/src/runtime.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/runtime.ts -------------------------------------------------------------------------------- /web/src/support.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/support.ts -------------------------------------------------------------------------------- /web/src/types.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/types.ts -------------------------------------------------------------------------------- /web/src/webgpu.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/src/webgpu.ts -------------------------------------------------------------------------------- /web/tests/node/test_object.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/tests/node/test_object.js -------------------------------------------------------------------------------- /web/tests/node/test_tensor.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/tests/node/test_tensor.js -------------------------------------------------------------------------------- /web/tests/node/test_vm.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/tests/node/test_vm.js -------------------------------------------------------------------------------- /web/tsconfig.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/tsconfig.json -------------------------------------------------------------------------------- /web/typedoc.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlc-ai/relax/HEAD/web/typedoc.json --------------------------------------------------------------------------------