├── .github ├── Dockerfile ├── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md ├── actions │ ├── build-base-image │ │ └── action.yaml │ ├── build-docs │ │ └── action.yaml │ ├── build-wheel │ │ └── action.yaml │ ├── install-hidet │ │ └── action.yaml │ └── run-tests │ │ └── action.yaml ├── base-image │ └── Dockerfile ├── copy-pr-bot.yaml ├── requirements-ci.txt ├── scripts │ ├── __init__.py │ ├── bench │ │ └── __init__.py │ ├── build_wheel.sh │ ├── cuda_cleanup.py │ ├── db_utils.py │ ├── set_gpu_types.sh │ ├── set_test_matrix.py │ ├── start_instances.py │ ├── stop_instances.py │ └── upload_results.py └── workflows │ ├── check-pr-title.yaml │ ├── lint.yaml │ ├── regression.yaml │ ├── release.yaml │ ├── sync.yaml │ └── tests.yaml ├── .gitignore ├── CMakeLists.txt ├── CODE_OF_CONDUCT.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── apps └── compile_server │ ├── Dockerfile │ ├── README.md │ ├── app.py │ ├── requirements.txt │ ├── resources │ ├── __init__.py │ ├── auth.py │ ├── compilation.py │ ├── compile_worker.py │ ├── status.py │ ├── user.py │ └── utils.py │ ├── run.py │ └── run.sh ├── config.cmake ├── docs ├── Makefile ├── make.bat ├── requirements.txt └── source │ ├── _static │ ├── custom.css │ ├── favicon.svg │ ├── img │ │ ├── resolve-example-conv2d.svg │ │ ├── resolve-example-matmul.svg │ │ └── subgraph-rewrite-example.svg │ └── logo.svg │ ├── clear-hidet-cache.py │ ├── conf.py │ ├── developer-guides │ └── contributing.rst │ ├── genindex.rst │ ├── getting-started │ ├── build-from-source.rst │ └── install.rst │ ├── hidet-script │ ├── examples │ │ └── index.rst │ ├── index.rst │ └── reference │ │ ├── 1-type-system.rst │ │ ├── 2-expression.rst │ │ ├── 3-statement.rst │ │ ├── 4-function.rst │ │ ├── 5-module.rst │ │ ├── 6-cuda-specific.rst │ │ ├── 7-cpu-specific.rst │ │ └── index.rst │ ├── how-to-guides │ └── add-new-operator │ │ └── index.rst │ ├── index.rst │ ├── notes │ └── operator-cache.rst │ ├── python_api │ ├── cuda.rst │ ├── data_types.rst │ ├── drivers.rst │ ├── ffi │ │ └── index.rst │ ├── graph │ │ ├── frontend │ │ │ ├── index.rst │ │ │ ├── onnx.rst │ │ │ └── torch.rst │ │ ├── index.rst │ │ └── transforms │ │ │ ├── index.rst │ │ │ ├── resolve_variant.rst │ │ │ └── subgraph_rewrite.rst │ ├── index.rst │ ├── ir │ │ ├── compute.rst │ │ ├── expr.rst │ │ ├── func.rst │ │ ├── index.rst │ │ ├── stmt.rst │ │ ├── task.rst │ │ └── type.rst │ ├── ops │ │ └── index.rst │ ├── option.rst │ ├── root.rst │ ├── runtime │ │ └── index.rst │ ├── tensor.rst │ ├── testing │ │ └── index.rst │ └── utils │ │ └── index.rst │ └── references.bib ├── examples ├── README.md ├── distributed │ └── test.py └── quantization │ ├── gpt2.py │ └── gpt2_performance.py ├── gallery ├── README.rst ├── developer-guides │ ├── README.rst │ ├── add-new-operator-compute-definition.py │ ├── add-new-operator-rule-based.py │ ├── add-new-operator-template-based.py │ ├── add-operator-resolve-rule.py │ ├── add-subgraph-rewrite-rule.py │ ├── add-torch-operator-mapping.py │ └── hidet-script-dynamic-kernel.py ├── getting-started │ ├── README.rst │ └── quick-start.py ├── hidet-script │ ├── 0-hello-world.py │ ├── 1-scalar-addition.py │ ├── 2-vector-addition.py │ ├── 3-kernel-functions.py │ ├── 4-naive-matmul.py │ ├── 5-efficient-matmul.py │ └── README.rst ├── how-to-guides │ ├── README.rst │ └── visualize-flow-graph.py └── tutorials │ ├── README.rst │ ├── optimize-onnx-model.py.backup │ └── optimize-pytorch-model.py.backup ├── include └── hidet │ ├── runtime.h │ └── runtime │ ├── callbacks.h │ ├── common.h │ ├── context.h │ ├── cpu │ ├── bfloat16.h │ ├── complex.h │ ├── context.h │ ├── float16.h │ ├── float32.h │ └── vector_types.h │ ├── cuda │ ├── complex.h │ ├── context.h │ ├── cublas.h │ ├── cuda.h │ ├── cudnn.h │ ├── float8_e4m3.h │ └── float8_e5m2.h │ ├── hip │ ├── context.h │ └── f16_utils.h │ ├── int_fastdiv.h │ ├── logging.h │ ├── memory_planner.h │ ├── symbols.h │ └── torch │ └── stream.h ├── pyproject.toml ├── python └── hidet │ ├── __init__.py │ ├── apps │ └── compile_server │ │ ├── __init__.py │ │ ├── auth.py │ │ ├── compilation.py │ │ ├── core.py │ │ └── user.py │ ├── backend │ ├── __init__.py │ ├── build.py │ └── codegen.py │ ├── cli │ ├── __init__.py │ ├── bench │ │ ├── __init__.py │ │ ├── bench.py │ │ ├── bench_all.py │ │ ├── bench_common.py │ │ ├── model.py │ │ ├── nlp │ │ │ ├── __init__.py │ │ │ ├── models.py │ │ │ └── nlp_model.py │ │ └── vision │ │ │ ├── __init__.py │ │ │ ├── inception_v3.py │ │ │ ├── mobilenet_v2.py │ │ │ ├── resnet.py │ │ │ ├── resnext.py │ │ │ └── vision_model.py │ ├── cache │ │ ├── __init__.py │ │ ├── clear.py │ │ ├── entry.py │ │ ├── status.py │ │ └── utils.py │ └── main.py │ ├── cuda │ ├── __init__.py │ ├── capability.py │ ├── cublas │ │ ├── __init__.py │ │ ├── ffi.py │ │ ├── kernels.py │ │ └── utils.py │ ├── cudnn │ │ ├── __init__.py │ │ ├── benchmark.py │ │ ├── ffi.py │ │ ├── kernels.py │ │ └── utils.py │ ├── device.py │ ├── event.py │ ├── graph.py │ ├── memory.py │ ├── nccl │ │ ├── __init__.py │ │ ├── comm.py │ │ ├── ffi.py │ │ └── libinfo.py │ └── stream.py │ ├── distributed │ ├── __init__.py │ ├── distributed.py │ ├── group.py │ └── store.py │ ├── drivers │ ├── __init__.py │ ├── build_graph.py │ ├── build_module.py │ ├── build_task.py │ └── utils.py │ ├── ffi │ ├── __init__.py │ ├── array.py │ ├── callbacks.py │ ├── convert.py │ ├── crt.py │ ├── ffi.py │ ├── runtime_api.py │ ├── shared_lib.py │ └── utils.py │ ├── graph │ ├── __init__.py │ ├── common.py │ ├── flow_graph.py │ ├── frontend │ │ ├── __init__.py │ │ ├── onnx │ │ │ ├── __init__.py │ │ │ ├── availability.py │ │ │ ├── onnx.py │ │ │ └── utils.py │ │ └── torch │ │ │ ├── __init__.py │ │ │ ├── availability.py │ │ │ ├── dynamo_backends.py │ │ │ ├── dynamo_config.py │ │ │ ├── flow_graph_cache.py │ │ │ ├── interpreter.py │ │ │ ├── register_functions.py │ │ │ ├── register_methods.py │ │ │ ├── register_modules.py │ │ │ ├── registry.py │ │ │ └── utils.py │ ├── graph_utils │ │ ├── __init__.py │ │ ├── functors.py │ │ └── instruments │ │ │ ├── __init__.py │ │ │ ├── benchmark_instrument.py │ │ │ └── debug_instrument.py │ ├── impl │ │ ├── __init__.py │ │ ├── dlpack.py │ │ └── graph_impl.py │ ├── nn │ │ ├── __init__.py │ │ ├── activations.py │ │ ├── attention.py │ │ ├── container.py │ │ ├── convolutions.py │ │ ├── identity.py │ │ ├── linear.py │ │ ├── module.py │ │ ├── norms.py │ │ ├── poolings.py │ │ └── transforms.py │ ├── operator.py │ ├── ops │ │ ├── __init__.py │ │ ├── activation.py │ │ ├── arithmetic.py │ │ ├── arithmetic_resolve.py │ │ ├── attention │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ └── attention_mask.py │ │ ├── compare.py │ │ ├── complex.py │ │ ├── conv1d │ │ │ ├── __init__.py │ │ │ ├── conv1d.py │ │ │ ├── conv1d_gemm.py │ │ │ ├── resolve.py │ │ │ └── utils.py │ │ ├── conv1d_transpose │ │ │ ├── __init__.py │ │ │ └── conv1d_transpose.py │ │ ├── conv2d │ │ │ ├── __init__.py │ │ │ ├── conv2d.py │ │ │ ├── conv2d_gemm.py │ │ │ ├── conv2d_winograd.py │ │ │ ├── resolve.py │ │ │ └── utils.py │ │ ├── conv2d_transpose │ │ │ ├── __init__.py │ │ │ ├── conv2d_transpose.py │ │ │ ├── conv2d_transpose_gemm.py │ │ │ └── resolve.py │ │ ├── conv3d │ │ │ ├── __init__.py │ │ │ ├── conv3d.py │ │ │ ├── conv3d_gemm.py │ │ │ ├── resolve.py │ │ │ └── utils.py │ │ ├── conv3d_transpose │ │ │ ├── __init__.py │ │ │ └── conv3d_transpose.py │ │ ├── create.py │ │ ├── cumulative.py │ │ ├── distributed.py │ │ ├── einops.py │ │ ├── embedding_bag.py │ │ ├── fusion │ │ │ ├── __init__.py │ │ │ ├── apply_prologue_epilogue.py │ │ │ └── fused_operator.py │ │ ├── image.py │ │ ├── linear.py │ │ ├── matmul │ │ │ ├── __init__.py │ │ │ ├── cuda_batch_matmul.py │ │ │ ├── hip_batch_matmul.py │ │ │ ├── matmul.py │ │ │ ├── matmul_cublas.py │ │ │ ├── matmul_f16.py │ │ │ ├── matmul_f16_cute.py │ │ │ ├── matmul_f16_cute_experimental.py │ │ │ ├── matmul_f16_sm90.py │ │ │ ├── matmul_f32_x86.py │ │ │ ├── matmul_f8.py │ │ │ └── resolve.py │ │ ├── normalize │ │ │ ├── __init__.py │ │ │ ├── layers.py │ │ │ ├── lp.py │ │ │ ├── norm.py │ │ │ └── resolve.py │ │ ├── opaque.py │ │ ├── pool.py │ │ ├── quant │ │ │ ├── __init__.py │ │ │ ├── matmul.py │ │ │ ├── matmul_f16_i8.py │ │ │ ├── matmul_f16_i8_atomic.py │ │ │ ├── resolve.py │ │ │ └── symmetric.py │ │ ├── reduce │ │ │ ├── __init__.py │ │ │ ├── reduce.py │ │ │ └── resolve.py │ │ ├── scatter.py │ │ ├── softmax.py │ │ ├── special.py │ │ ├── transfer.py │ │ ├── transform.py │ │ ├── transpose2d.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── schedule_utils.py │ │ │ └── tensor_utils.py │ ├── tensor.py │ └── transforms │ │ ├── __init__.py │ │ ├── base.py │ │ ├── conv_channel_last.py │ │ ├── eliminate_barrier.py │ │ ├── fuse_operator.py │ │ ├── graph_patterns │ │ ├── __init__.py │ │ ├── arithmetic_patterns.py │ │ ├── attn_patterns.py │ │ ├── base.py │ │ ├── conv2d_patterns.py │ │ ├── matmul_patterns.py │ │ ├── quant │ │ │ ├── __init__.py │ │ │ ├── embedding.py │ │ │ └── linear.py │ │ ├── register_all_patterns.py │ │ └── transform_patterns.py │ │ ├── instruments │ │ ├── __init__.py │ │ ├── base.py │ │ ├── convert_flowgraph_to_vgpu.py │ │ ├── profile_instrument.py │ │ └── save_graph_instrument.py │ │ ├── resolve_variant.py │ │ ├── selective_quantize.py │ │ ├── subgraph_rewrite.py │ │ └── utils.py │ ├── hip │ ├── __init__.py │ ├── capability.py │ ├── device.py │ ├── event.py │ ├── ffi.py │ ├── graph.py │ ├── memory.py │ └── stream.py │ ├── ir │ ├── __init__.py │ ├── analyzers │ │ ├── __init__.py │ │ └── bound_analyzer.py │ ├── builders │ │ ├── __init__.py │ │ ├── func_builder.py │ │ └── stmt_builder.py │ ├── compute │ │ ├── __init__.py │ │ ├── cops │ │ │ ├── __init__.py │ │ │ ├── matmul.py │ │ │ ├── pad.py │ │ │ └── reduce.py │ │ ├── primitives.py │ │ └── reduce_operations.py │ ├── cute │ │ ├── __init__.py │ │ ├── algorithm.py │ │ ├── collective │ │ │ ├── __init__.py │ │ │ └── copy.py │ │ ├── contexts.py │ │ ├── expr.py │ │ ├── int_tuple.py │ │ ├── layout.py │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── algorithm.py │ │ │ ├── arithmetic.py │ │ │ ├── copy.py │ │ │ ├── misc.py │ │ │ ├── mma.py │ │ │ ├── partition.py │ │ │ ├── rearrange.py │ │ │ ├── reduce.py │ │ │ ├── subtensor.py │ │ │ └── tensor.py │ │ ├── swizzle.py │ │ ├── type.py │ │ └── typing.py │ ├── dialects │ │ ├── __init__.py │ │ └── pattern.py │ ├── dtypes │ │ ├── __init__.py │ │ ├── boolean.py │ │ ├── complex.py │ │ ├── floats.py │ │ ├── integer.py │ │ ├── integer_subbyte.py │ │ ├── promotion.py │ │ ├── utils.py │ │ └── vector.py │ ├── expr.py │ ├── func.py │ ├── functors │ │ ├── __init__.py │ │ ├── base_functor.py │ │ ├── compute_functor.py │ │ ├── cute_functor.py │ │ ├── expr_functor.py │ │ ├── ir_functor.py │ │ ├── layout_functor.py │ │ ├── mapping_functor.py │ │ ├── module_functor.py │ │ ├── stmt_functor.py │ │ └── type_functor.py │ ├── layout.py │ ├── library │ │ ├── __init__.py │ │ ├── cuda │ │ │ ├── __init__.py │ │ │ ├── cublas │ │ │ │ ├── __init__.py │ │ │ │ ├── kernels.py │ │ │ │ └── regs.py │ │ │ └── matmul │ │ │ │ ├── __init__.py │ │ │ │ └── simt.py │ │ ├── tune.py │ │ └── utils.py │ ├── mapping.py │ ├── module.py │ ├── node.py │ ├── polinomial.py │ ├── primitives │ │ ├── __init__.py │ │ ├── complex.py │ │ ├── cpu │ │ │ ├── __init__.py │ │ │ ├── atomic.py │ │ │ ├── avx.py │ │ │ ├── avx_helper.py │ │ │ └── math │ │ │ │ ├── __init__.py │ │ │ │ ├── bfloat16.py │ │ │ │ ├── float16.py │ │ │ │ ├── float32.py │ │ │ │ ├── float64.py │ │ │ │ ├── int32.py │ │ │ │ └── int64.py │ │ ├── cuda │ │ │ ├── __init__.py │ │ │ ├── atomic.py │ │ │ ├── barrier.py │ │ │ ├── cluster.py │ │ │ ├── copy_tma.py │ │ │ ├── cp_async.py │ │ │ ├── cvt.py │ │ │ ├── cvta.py │ │ │ ├── errchk.py │ │ │ ├── fastintdiv.py │ │ │ ├── funcs.py │ │ │ ├── half.py │ │ │ ├── ldst.py │ │ │ ├── lop3.py │ │ │ ├── math │ │ │ │ ├── __init__.py │ │ │ │ ├── bfloat16.py │ │ │ │ ├── complex128.py │ │ │ │ ├── complex64.py │ │ │ │ ├── float16.py │ │ │ │ ├── float16x2.py │ │ │ │ ├── float32.py │ │ │ │ ├── float64.py │ │ │ │ ├── float8e4m3.py │ │ │ │ ├── float8e5m2.py │ │ │ │ ├── int32.py │ │ │ │ └── int64.py │ │ │ ├── memcpy.py │ │ │ ├── mma.py │ │ │ ├── mutex.py │ │ │ ├── nccl.py │ │ │ ├── prmt.py │ │ │ ├── setmaxnreg.py │ │ │ ├── shfl.py │ │ │ ├── smem.py │ │ │ ├── sync.py │ │ │ ├── tcgen05_cp.py │ │ │ ├── tcgen05_ldst.py │ │ │ ├── tensor_map.py │ │ │ ├── time.py │ │ │ ├── tmem.py │ │ │ ├── vars.py │ │ │ ├── wgmma.py │ │ │ └── wmma.py │ │ ├── debug.py │ │ ├── func.py │ │ ├── hip │ │ │ ├── __init__.py │ │ │ ├── buffer_addr.py │ │ │ ├── errchk.py │ │ │ ├── lds_sync.py │ │ │ ├── math │ │ │ │ ├── __init__.py │ │ │ │ ├── float16.py │ │ │ │ ├── float32.py │ │ │ │ └── int32.py │ │ │ ├── mfma.py │ │ │ └── vars.py │ │ ├── math.py │ │ ├── runtime.py │ │ └── vars.py │ ├── schedulers │ │ ├── __init__.py │ │ ├── base.py │ │ ├── cpu │ │ │ ├── __init__.py │ │ │ └── scheduler.py │ │ └── cuda │ │ │ ├── __init__.py │ │ │ └── scheduler.py │ ├── stmt.py │ ├── target.py │ ├── task.py │ ├── tools │ │ ├── __init__.py │ │ ├── free_var_collector.py │ │ ├── hasher.py │ │ ├── ir_dumper.py │ │ ├── printer.py │ │ ├── renamer.py │ │ ├── rewriter.py │ │ ├── simplifier.py │ │ ├── type_infer.py │ │ └── util_functors.py │ ├── type.py │ └── utils │ │ ├── __init__.py │ │ ├── broadcast_utils.py │ │ ├── call_graph.py │ │ ├── expr_utils.py │ │ ├── hash_sum.py │ │ ├── index_transform.py │ │ └── type_utils.py │ ├── kernels │ └── __init__.py │ ├── lang │ ├── __init__.py │ ├── attrs.py │ ├── attrs │ │ ├── __init__.py │ │ ├── cuda.py │ │ └── hip.py │ ├── constructs │ │ ├── __init__.py │ │ ├── context.py │ │ ├── declare.py │ │ ├── loops.py │ │ └── meta.py │ ├── cpu.py │ ├── cuda │ │ ├── __init__.py │ │ └── contexts.py │ ├── hip.py │ ├── layout.py │ ├── mapping.py │ ├── runtime.py │ ├── script.py │ ├── transpiler.py │ └── types.py │ ├── libinfo.py │ ├── logging.py │ ├── option.py │ ├── runtime │ ├── __init__.py │ ├── compiled_graph.py │ ├── compiled_module.py │ ├── compiled_task.py │ ├── device.py │ ├── storage.py │ └── utils │ │ └── dispatch_table.py │ ├── testing │ ├── __init__.py │ ├── capture_stdout.py │ ├── models │ │ ├── __init__.py │ │ ├── gemma.py │ │ ├── gpt2.py │ │ ├── llama.py │ │ └── resnet.py │ ├── onnx_models.py │ ├── onnx_utils.py │ ├── torch_utils.py │ └── utils.py │ ├── transforms │ ├── __init__.py │ ├── add_explicit_cast.py │ ├── add_hints.py │ ├── annotate_header_and_libs.py │ ├── attach_hash_to_signature.py │ ├── base.py │ ├── check_launch_configuration.py │ ├── convert_div_to_fastintdiv.py │ ├── cute │ │ ├── __init__.py │ │ ├── analysis │ │ │ ├── __init__.py │ │ │ └── tensor_alias_analysis.py │ │ ├── cuda │ │ │ ├── __init__.py │ │ │ ├── annotate_mbarrier.py │ │ │ ├── cost_model.py │ │ │ ├── instantiate_auto_annotation.py │ │ │ ├── instruction_selection.py │ │ │ ├── lower_cute_dialect.py │ │ │ ├── lower_ops │ │ │ │ ├── __init__.py │ │ │ │ ├── algorithm.py │ │ │ │ ├── arithmetic.py │ │ │ │ ├── collective.py │ │ │ │ ├── copy.py │ │ │ │ ├── mbarrier.py │ │ │ │ ├── misc.py │ │ │ │ ├── mma.py │ │ │ │ ├── partition.py │ │ │ │ ├── rearrange.py │ │ │ │ ├── reduce.py │ │ │ │ ├── registry.py │ │ │ │ ├── subtensor.py │ │ │ │ └── tensor.py │ │ │ ├── resolve_bank_conflict.py │ │ │ ├── shared_memory_allocation.py │ │ │ ├── tma_fallback_copy.py │ │ │ ├── tma_layout_utils.py │ │ │ └── vectorize_elementwise.py │ │ └── generic │ │ │ ├── __init__.py │ │ │ ├── canonicalize.py │ │ │ ├── canonicalize_arithmetic_expression.py │ │ │ └── deadcode_elimination.py │ ├── declare_to_let.py │ ├── expand_let_expr.py │ ├── expand_repeat.py │ ├── explicit_unroll.py │ ├── flatten_tensor_index.py │ ├── flatten_tensor_slice.py │ ├── generate_launch_func.py │ ├── import_primitive_functions.py │ ├── inline_function.py │ ├── inline_let_stmt.py │ ├── instantiate_symbols.py │ ├── instruments │ │ ├── __init__.py │ │ ├── base.py │ │ ├── profile_instrument.py │ │ └── save_ir_instrument.py │ ├── lower_integer_subbyte.py │ ├── lower_protect_access.py │ ├── lower_special_cast.py │ ├── lower_task_mapping.py │ ├── normalize_const_tensor.py │ ├── propagate_launch_bound.py │ ├── resolve_generic_primitive_function.py │ ├── rule_based_simplifier.py │ ├── simplify_addition_chain.py │ ├── simplify_stmt.py │ ├── spatial_simplification.py │ ├── task_mapping_bound_check.py │ └── unify_global_objects.py │ ├── utils │ ├── __init__.py │ ├── benchmark │ │ ├── __init__.py │ │ ├── bench.py │ │ ├── gpu_freq.py │ │ └── utils.py │ ├── cache_utils.py │ ├── counters.py │ ├── cuda_sanitizer.py │ ├── dataclass.py │ ├── doc.py │ ├── exiting.py │ ├── fault_handler.py │ ├── files.py │ ├── folder_lock.py │ ├── gc.py │ ├── git_utils.py │ ├── multiprocess.py │ ├── namer.py │ ├── ncu_utils.py │ ├── net_utils.py │ ├── netron.py │ ├── nsys_utils.py │ ├── omniperf_utils.py │ ├── ort_utils.py │ ├── overrides.py │ ├── py.py │ ├── stack_limit.py │ ├── structure.py │ ├── torch_utils.py │ ├── trace_utils.py │ └── transformers_utils.py │ └── version.py ├── scripts ├── bench │ ├── README.md │ ├── benchmark.py │ ├── requirements.txt │ └── run.py ├── lint │ ├── .clang-format │ ├── _format.py │ ├── add_copyright.py │ ├── format.sh │ ├── lint.sh │ └── pylintrc ├── nightly-builds │ ├── README.md │ ├── add-crontab-record.sh │ └── update-nightly.sh ├── regression │ ├── __init__.py │ ├── email_sender.py │ ├── model_performance.py │ ├── op_performance.py │ ├── regression_data.json │ ├── requirements.txt │ ├── result_entry.py │ └── run.py └── wheel │ ├── build_wheel.sh │ ├── build_wheel_manylinux_2_28_x86_64.sh │ ├── current_version.py │ ├── dockerfiles │ └── manylinux_2_28_x86_64 │ │ └── Dockerfile │ ├── update_nightly.sh │ └── upload_wheel_to_pypi.sh ├── setup.py ├── src └── hidet │ ├── empty.cpp │ └── runtime │ ├── callbacks.cpp │ ├── cpu │ └── context.cpp │ ├── cuda │ ├── context.cpp │ ├── cublas.cpp │ ├── cuda.cpp │ ├── cudnn.cpp │ └── utils.h │ ├── hip │ └── context.cpp │ ├── int_fastdiv.cpp │ ├── logging.cpp │ └── symbols.cpp └── tests ├── README.md ├── benchmarks ├── bench_dynamic.py ├── bench_op.py ├── bench_op_torch_api.py ├── bench_task.py ├── bench_transformer.py ├── bench_transformer_comptime.py ├── bench_vision.py ├── run_configs.json ├── run_configs_full.json └── run_tests.py ├── conftest.py ├── cuda ├── test_cublas.py ├── test_cuda_graph.py └── test_cudnn.py ├── cute_fusion ├── fusion_bench_utils.py ├── test_epilogue_fusion.py ├── test_issue436.py ├── test_matmul_bias_cast.py ├── test_matmul_cast_dynamic.py ├── test_matmul_residual_bias.py ├── test_matmul_residual_permute.py ├── test_matmul_riskfuel.py └── test_matmul_standalone.py ├── cute_kernel ├── gemm_quant_nbit.py ├── quant_utils.py ├── quantized_linear.py ├── test_attention_bwd.py ├── test_cute_attention.py ├── test_flash_decoding.py ├── test_gemm_quant_nbit.py ├── test_gemv.py ├── test_hopper_gemm.py ├── test_hopper_gemm_e2e.py ├── test_mlp.py ├── test_moe_align.py ├── test_quant_linear.py ├── test_rmsnorm.py └── test_weight_quantization.py ├── cute_ops ├── __init__.py ├── test_arithmetic2.py ├── test_collective.py ├── test_ldst.py ├── test_mma.py ├── test_rearrange.py ├── test_reduce.py └── test_tensor.py ├── distributed ├── test_file_store.py ├── test_op.py ├── test_runtime.py ├── test_tcp_store.py └── utils.py ├── flowgraph └── test_graph_visualization.py ├── frontends ├── onnx │ └── test_onnx_slice.py └── torch │ ├── models │ ├── test_torch_bert.py │ ├── test_torch_densenet121.py │ ├── test_torch_pegasus.py │ └── test_torch_resnet50.py │ ├── subgraphs │ ├── test_torch_graph_flatten.py │ ├── test_torch_graph_pad.py │ └── test_torch_graph_to.py │ ├── test_einops_rearrange.py │ ├── test_flow_graph_cache.py │ ├── test_torch_activation.py │ ├── test_torch_arithmetic.py │ ├── test_torch_conv1d.py │ ├── test_torch_conv1d_transpose.py │ ├── test_torch_conv2d.py │ ├── test_torch_conv2d_transpose.py │ ├── test_torch_conv3d.py │ ├── test_torch_conv3d_transpose.py │ ├── test_torch_creation.py │ ├── test_torch_dyn_shape.py │ ├── test_torch_fxgraph.py │ ├── test_torch_image.py │ ├── test_torch_inplace.py │ ├── test_torch_interoperability.py │ ├── test_torch_mix_cuda_cpu.py │ ├── test_torch_mul.py │ ├── test_torch_norm.py │ ├── test_torch_pooling.py │ ├── test_torch_reduce.py │ ├── test_torch_sdpa.py │ ├── test_torch_split.py │ ├── test_torch_stream.py │ └── test_torch_to.py ├── hip └── test_hip_graph.py ├── ir ├── dialects │ └── test_pattern.py ├── dtypes │ ├── test_fp8e4m3.py │ └── test_fp8e5m2.py ├── functors │ └── test_persistence.py ├── parser │ └── test_parser.py ├── primitives │ ├── cuda │ │ ├── test_barrier.py │ │ ├── test_cluster.py │ │ ├── test_copy_tma.py │ │ ├── test_exp2.py │ │ ├── test_half.py │ │ ├── test_lop3.py │ │ ├── test_mma.py │ │ ├── test_prmt.py │ │ ├── test_tcgen05_cp.py │ │ ├── test_tcgen05_ldst.py │ │ ├── test_tmem_alloc.py │ │ └── test_wgmma.py │ └── hip │ │ ├── test_buffer_addr.py │ │ └── test_mfma.py ├── test_expr_const_fold.py ├── test_int_subbyte.py ├── test_layout.py ├── test_primitives.py └── test_symbol_var.py ├── minimal └── test_add.py ├── models ├── test_gemma.py ├── test_gpt2.py ├── test_llama.py ├── test_llama_graph.py └── test_quant_gpt2.py ├── multiprocessing ├── lazy_init_sample.py └── test_lazy_initialization.py ├── operators_core ├── test_activation.py ├── test_arithmetic.py ├── test_attention.py ├── test_compare.py ├── test_complex.py ├── test_conv1d_transpose.py ├── test_create.py ├── test_fusion.py ├── test_identity.py ├── test_inplace_operator.py ├── test_matmul.py ├── test_norm.py ├── test_opaque.py ├── test_operator.py ├── test_quantization.py ├── test_reduce.py ├── test_softmax.py ├── test_symmetric_quant.py ├── test_tensor.py ├── test_transform.py └── test_tri.py ├── operators_vision ├── test_conv1d.py ├── test_conv2d.py ├── test_conv2d_transpose.py ├── test_conv3d.py ├── test_conv3d_transpose.py ├── test_image.py └── test_pool.py ├── runtime ├── test_dispatch_table.py └── test_try_catch.py ├── script ├── test_assignment.py ├── test_comprehension.py ├── test_constant.py ├── test_context.py ├── test_cpu_kernel.py ├── test_for_loops.py ├── test_func_import.py ├── test_lambda.py ├── test_meta.py ├── test_parallel.py ├── test_return_type.py └── test_unroll.py ├── transforms ├── test_graph_rewrites.py ├── test_inline_let_stmt.py ├── test_rule_based_simplifier.py └── test_simplify_addition_chain.py ├── unit_tests ├── check_import_time.py ├── option │ ├── main.py │ └── test_option_env.py ├── test_compiled_model.py ├── test_dynamic_shape.py ├── test_frontend_onnx.py ├── test_import_time.py ├── test_save_lower_ir.py └── test_vllm_hidet_compile.py └── utils ├── benchmark └── example.py ├── test_cuda_sanitizer.py ├── test_ncu_utils.py └── test_nsys_utils.py /.github/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/Dockerfile -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/actions/build-base-image/action.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/actions/build-base-image/action.yaml -------------------------------------------------------------------------------- /.github/actions/build-docs/action.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/actions/build-docs/action.yaml -------------------------------------------------------------------------------- /.github/actions/build-wheel/action.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/actions/build-wheel/action.yaml -------------------------------------------------------------------------------- /.github/actions/install-hidet/action.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/actions/install-hidet/action.yaml -------------------------------------------------------------------------------- /.github/actions/run-tests/action.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/actions/run-tests/action.yaml -------------------------------------------------------------------------------- /.github/base-image/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/base-image/Dockerfile -------------------------------------------------------------------------------- /.github/copy-pr-bot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/copy-pr-bot.yaml -------------------------------------------------------------------------------- /.github/requirements-ci.txt: -------------------------------------------------------------------------------- 1 | mysql-connector-python -------------------------------------------------------------------------------- /.github/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.github/scripts/bench/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.github/scripts/build_wheel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/scripts/build_wheel.sh -------------------------------------------------------------------------------- /.github/scripts/cuda_cleanup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/scripts/cuda_cleanup.py -------------------------------------------------------------------------------- /.github/scripts/db_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/scripts/db_utils.py -------------------------------------------------------------------------------- /.github/scripts/set_gpu_types.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/scripts/set_gpu_types.sh -------------------------------------------------------------------------------- /.github/scripts/set_test_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/scripts/set_test_matrix.py -------------------------------------------------------------------------------- /.github/scripts/start_instances.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/scripts/start_instances.py -------------------------------------------------------------------------------- /.github/scripts/stop_instances.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/scripts/stop_instances.py -------------------------------------------------------------------------------- /.github/scripts/upload_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/scripts/upload_results.py -------------------------------------------------------------------------------- /.github/workflows/check-pr-title.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/workflows/check-pr-title.yaml -------------------------------------------------------------------------------- /.github/workflows/lint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/workflows/lint.yaml -------------------------------------------------------------------------------- /.github/workflows/regression.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/workflows/regression.yaml -------------------------------------------------------------------------------- /.github/workflows/release.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/workflows/release.yaml -------------------------------------------------------------------------------- /.github/workflows/sync.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/workflows/sync.yaml -------------------------------------------------------------------------------- /.github/workflows/tests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.github/workflows/tests.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/README.md -------------------------------------------------------------------------------- /apps/compile_server/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/Dockerfile -------------------------------------------------------------------------------- /apps/compile_server/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/README.md -------------------------------------------------------------------------------- /apps/compile_server/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/app.py -------------------------------------------------------------------------------- /apps/compile_server/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/requirements.txt -------------------------------------------------------------------------------- /apps/compile_server/resources/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/resources/__init__.py -------------------------------------------------------------------------------- /apps/compile_server/resources/auth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/resources/auth.py -------------------------------------------------------------------------------- /apps/compile_server/resources/compilation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/resources/compilation.py -------------------------------------------------------------------------------- /apps/compile_server/resources/compile_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/resources/compile_worker.py -------------------------------------------------------------------------------- /apps/compile_server/resources/status.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/resources/status.py -------------------------------------------------------------------------------- /apps/compile_server/resources/user.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/resources/user.py -------------------------------------------------------------------------------- /apps/compile_server/resources/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/resources/utils.py -------------------------------------------------------------------------------- /apps/compile_server/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/run.py -------------------------------------------------------------------------------- /apps/compile_server/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/apps/compile_server/run.sh -------------------------------------------------------------------------------- /config.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/config.cmake -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/_static/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/_static/custom.css -------------------------------------------------------------------------------- /docs/source/_static/favicon.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/_static/favicon.svg -------------------------------------------------------------------------------- /docs/source/_static/img/resolve-example-conv2d.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/_static/img/resolve-example-conv2d.svg -------------------------------------------------------------------------------- /docs/source/_static/img/resolve-example-matmul.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/_static/img/resolve-example-matmul.svg -------------------------------------------------------------------------------- /docs/source/_static/img/subgraph-rewrite-example.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/_static/img/subgraph-rewrite-example.svg -------------------------------------------------------------------------------- /docs/source/_static/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/_static/logo.svg -------------------------------------------------------------------------------- /docs/source/clear-hidet-cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/clear-hidet-cache.py -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/developer-guides/contributing.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/developer-guides/contributing.rst -------------------------------------------------------------------------------- /docs/source/genindex.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/genindex.rst -------------------------------------------------------------------------------- /docs/source/getting-started/build-from-source.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/getting-started/build-from-source.rst -------------------------------------------------------------------------------- /docs/source/getting-started/install.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/getting-started/install.rst -------------------------------------------------------------------------------- /docs/source/hidet-script/examples/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/hidet-script/examples/index.rst -------------------------------------------------------------------------------- /docs/source/hidet-script/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/hidet-script/index.rst -------------------------------------------------------------------------------- /docs/source/hidet-script/reference/1-type-system.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/hidet-script/reference/1-type-system.rst -------------------------------------------------------------------------------- /docs/source/hidet-script/reference/2-expression.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/hidet-script/reference/2-expression.rst -------------------------------------------------------------------------------- /docs/source/hidet-script/reference/3-statement.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/hidet-script/reference/3-statement.rst -------------------------------------------------------------------------------- /docs/source/hidet-script/reference/4-function.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/hidet-script/reference/4-function.rst -------------------------------------------------------------------------------- /docs/source/hidet-script/reference/5-module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/hidet-script/reference/5-module.rst -------------------------------------------------------------------------------- /docs/source/hidet-script/reference/7-cpu-specific.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/hidet-script/reference/7-cpu-specific.rst -------------------------------------------------------------------------------- /docs/source/hidet-script/reference/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/hidet-script/reference/index.rst -------------------------------------------------------------------------------- /docs/source/how-to-guides/add-new-operator/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/how-to-guides/add-new-operator/index.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/notes/operator-cache.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/notes/operator-cache.rst -------------------------------------------------------------------------------- /docs/source/python_api/cuda.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/cuda.rst -------------------------------------------------------------------------------- /docs/source/python_api/data_types.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/data_types.rst -------------------------------------------------------------------------------- /docs/source/python_api/drivers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/drivers.rst -------------------------------------------------------------------------------- /docs/source/python_api/ffi/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/ffi/index.rst -------------------------------------------------------------------------------- /docs/source/python_api/graph/frontend/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/graph/frontend/index.rst -------------------------------------------------------------------------------- /docs/source/python_api/graph/frontend/onnx.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/graph/frontend/onnx.rst -------------------------------------------------------------------------------- /docs/source/python_api/graph/frontend/torch.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/graph/frontend/torch.rst -------------------------------------------------------------------------------- /docs/source/python_api/graph/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/graph/index.rst -------------------------------------------------------------------------------- /docs/source/python_api/graph/transforms/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/graph/transforms/index.rst -------------------------------------------------------------------------------- /docs/source/python_api/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/index.rst -------------------------------------------------------------------------------- /docs/source/python_api/ir/compute.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/ir/compute.rst -------------------------------------------------------------------------------- /docs/source/python_api/ir/expr.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/ir/expr.rst -------------------------------------------------------------------------------- /docs/source/python_api/ir/func.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/ir/func.rst -------------------------------------------------------------------------------- /docs/source/python_api/ir/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/ir/index.rst -------------------------------------------------------------------------------- /docs/source/python_api/ir/stmt.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/ir/stmt.rst -------------------------------------------------------------------------------- /docs/source/python_api/ir/task.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/ir/task.rst -------------------------------------------------------------------------------- /docs/source/python_api/ir/type.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/ir/type.rst -------------------------------------------------------------------------------- /docs/source/python_api/ops/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/ops/index.rst -------------------------------------------------------------------------------- /docs/source/python_api/option.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/option.rst -------------------------------------------------------------------------------- /docs/source/python_api/root.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/root.rst -------------------------------------------------------------------------------- /docs/source/python_api/runtime/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/runtime/index.rst -------------------------------------------------------------------------------- /docs/source/python_api/tensor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/tensor.rst -------------------------------------------------------------------------------- /docs/source/python_api/testing/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/testing/index.rst -------------------------------------------------------------------------------- /docs/source/python_api/utils/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/python_api/utils/index.rst -------------------------------------------------------------------------------- /docs/source/references.bib: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/docs/source/references.bib -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/distributed/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/examples/distributed/test.py -------------------------------------------------------------------------------- /examples/quantization/gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/examples/quantization/gpt2.py -------------------------------------------------------------------------------- /examples/quantization/gpt2_performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/examples/quantization/gpt2_performance.py -------------------------------------------------------------------------------- /gallery/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/README.rst -------------------------------------------------------------------------------- /gallery/developer-guides/README.rst: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gallery/developer-guides/add-operator-resolve-rule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/developer-guides/add-operator-resolve-rule.py -------------------------------------------------------------------------------- /gallery/developer-guides/add-subgraph-rewrite-rule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/developer-guides/add-subgraph-rewrite-rule.py -------------------------------------------------------------------------------- /gallery/getting-started/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/getting-started/README.rst -------------------------------------------------------------------------------- /gallery/getting-started/quick-start.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/getting-started/quick-start.py -------------------------------------------------------------------------------- /gallery/hidet-script/0-hello-world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/hidet-script/0-hello-world.py -------------------------------------------------------------------------------- /gallery/hidet-script/1-scalar-addition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/hidet-script/1-scalar-addition.py -------------------------------------------------------------------------------- /gallery/hidet-script/2-vector-addition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/hidet-script/2-vector-addition.py -------------------------------------------------------------------------------- /gallery/hidet-script/3-kernel-functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/hidet-script/3-kernel-functions.py -------------------------------------------------------------------------------- /gallery/hidet-script/4-naive-matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/hidet-script/4-naive-matmul.py -------------------------------------------------------------------------------- /gallery/hidet-script/5-efficient-matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/hidet-script/5-efficient-matmul.py -------------------------------------------------------------------------------- /gallery/hidet-script/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/hidet-script/README.rst -------------------------------------------------------------------------------- /gallery/how-to-guides/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/how-to-guides/README.rst -------------------------------------------------------------------------------- /gallery/how-to-guides/visualize-flow-graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/how-to-guides/visualize-flow-graph.py -------------------------------------------------------------------------------- /gallery/tutorials/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/tutorials/README.rst -------------------------------------------------------------------------------- /gallery/tutorials/optimize-onnx-model.py.backup: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/tutorials/optimize-onnx-model.py.backup -------------------------------------------------------------------------------- /gallery/tutorials/optimize-pytorch-model.py.backup: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/gallery/tutorials/optimize-pytorch-model.py.backup -------------------------------------------------------------------------------- /include/hidet/runtime.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime.h -------------------------------------------------------------------------------- /include/hidet/runtime/callbacks.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/callbacks.h -------------------------------------------------------------------------------- /include/hidet/runtime/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/common.h -------------------------------------------------------------------------------- /include/hidet/runtime/context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/context.h -------------------------------------------------------------------------------- /include/hidet/runtime/cpu/bfloat16.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cpu/bfloat16.h -------------------------------------------------------------------------------- /include/hidet/runtime/cpu/complex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cpu/complex.h -------------------------------------------------------------------------------- /include/hidet/runtime/cpu/context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cpu/context.h -------------------------------------------------------------------------------- /include/hidet/runtime/cpu/float16.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cpu/float16.h -------------------------------------------------------------------------------- /include/hidet/runtime/cpu/float32.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cpu/float32.h -------------------------------------------------------------------------------- /include/hidet/runtime/cpu/vector_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cpu/vector_types.h -------------------------------------------------------------------------------- /include/hidet/runtime/cuda/complex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cuda/complex.h -------------------------------------------------------------------------------- /include/hidet/runtime/cuda/context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cuda/context.h -------------------------------------------------------------------------------- /include/hidet/runtime/cuda/cublas.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cuda/cublas.h -------------------------------------------------------------------------------- /include/hidet/runtime/cuda/cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cuda/cuda.h -------------------------------------------------------------------------------- /include/hidet/runtime/cuda/cudnn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cuda/cudnn.h -------------------------------------------------------------------------------- /include/hidet/runtime/cuda/float8_e4m3.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cuda/float8_e4m3.h -------------------------------------------------------------------------------- /include/hidet/runtime/cuda/float8_e5m2.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/cuda/float8_e5m2.h -------------------------------------------------------------------------------- /include/hidet/runtime/hip/context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/hip/context.h -------------------------------------------------------------------------------- /include/hidet/runtime/hip/f16_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/hip/f16_utils.h -------------------------------------------------------------------------------- /include/hidet/runtime/int_fastdiv.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/int_fastdiv.h -------------------------------------------------------------------------------- /include/hidet/runtime/logging.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/logging.h -------------------------------------------------------------------------------- /include/hidet/runtime/memory_planner.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/memory_planner.h -------------------------------------------------------------------------------- /include/hidet/runtime/symbols.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/symbols.h -------------------------------------------------------------------------------- /include/hidet/runtime/torch/stream.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/include/hidet/runtime/torch/stream.h -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/pyproject.toml -------------------------------------------------------------------------------- /python/hidet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/__init__.py -------------------------------------------------------------------------------- /python/hidet/apps/compile_server/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/apps/compile_server/__init__.py -------------------------------------------------------------------------------- /python/hidet/apps/compile_server/auth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/apps/compile_server/auth.py -------------------------------------------------------------------------------- /python/hidet/apps/compile_server/compilation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/apps/compile_server/compilation.py -------------------------------------------------------------------------------- /python/hidet/apps/compile_server/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/apps/compile_server/core.py -------------------------------------------------------------------------------- /python/hidet/apps/compile_server/user.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/apps/compile_server/user.py -------------------------------------------------------------------------------- /python/hidet/backend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/backend/__init__.py -------------------------------------------------------------------------------- /python/hidet/backend/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/backend/build.py -------------------------------------------------------------------------------- /python/hidet/backend/codegen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/backend/codegen.py -------------------------------------------------------------------------------- /python/hidet/cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/__init__.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/__init__.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/bench.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/bench_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/bench_all.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/bench_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/bench_common.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/model.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/nlp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/nlp/__init__.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/nlp/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/nlp/models.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/nlp/nlp_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/nlp/nlp_model.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/vision/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/vision/__init__.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/vision/inception_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/vision/inception_v3.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/vision/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/vision/mobilenet_v2.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/vision/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/vision/resnet.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/vision/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/vision/resnext.py -------------------------------------------------------------------------------- /python/hidet/cli/bench/vision/vision_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/bench/vision/vision_model.py -------------------------------------------------------------------------------- /python/hidet/cli/cache/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/cache/__init__.py -------------------------------------------------------------------------------- /python/hidet/cli/cache/clear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/cache/clear.py -------------------------------------------------------------------------------- /python/hidet/cli/cache/entry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/cache/entry.py -------------------------------------------------------------------------------- /python/hidet/cli/cache/status.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/cache/status.py -------------------------------------------------------------------------------- /python/hidet/cli/cache/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/cache/utils.py -------------------------------------------------------------------------------- /python/hidet/cli/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cli/main.py -------------------------------------------------------------------------------- /python/hidet/cuda/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/__init__.py -------------------------------------------------------------------------------- /python/hidet/cuda/capability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/capability.py -------------------------------------------------------------------------------- /python/hidet/cuda/cublas/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/cublas/__init__.py -------------------------------------------------------------------------------- /python/hidet/cuda/cublas/ffi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/cublas/ffi.py -------------------------------------------------------------------------------- /python/hidet/cuda/cublas/kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/cublas/kernels.py -------------------------------------------------------------------------------- /python/hidet/cuda/cublas/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/cublas/utils.py -------------------------------------------------------------------------------- /python/hidet/cuda/cudnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/cudnn/__init__.py -------------------------------------------------------------------------------- /python/hidet/cuda/cudnn/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/cudnn/benchmark.py -------------------------------------------------------------------------------- /python/hidet/cuda/cudnn/ffi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/cudnn/ffi.py -------------------------------------------------------------------------------- /python/hidet/cuda/cudnn/kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/cudnn/kernels.py -------------------------------------------------------------------------------- /python/hidet/cuda/cudnn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/cudnn/utils.py -------------------------------------------------------------------------------- /python/hidet/cuda/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/device.py -------------------------------------------------------------------------------- /python/hidet/cuda/event.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/event.py -------------------------------------------------------------------------------- /python/hidet/cuda/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/graph.py -------------------------------------------------------------------------------- /python/hidet/cuda/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/memory.py -------------------------------------------------------------------------------- /python/hidet/cuda/nccl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/nccl/__init__.py -------------------------------------------------------------------------------- /python/hidet/cuda/nccl/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/nccl/comm.py -------------------------------------------------------------------------------- /python/hidet/cuda/nccl/ffi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/nccl/ffi.py -------------------------------------------------------------------------------- /python/hidet/cuda/nccl/libinfo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/nccl/libinfo.py -------------------------------------------------------------------------------- /python/hidet/cuda/stream.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/cuda/stream.py -------------------------------------------------------------------------------- /python/hidet/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/distributed/__init__.py -------------------------------------------------------------------------------- /python/hidet/distributed/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/distributed/distributed.py -------------------------------------------------------------------------------- /python/hidet/distributed/group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/distributed/group.py -------------------------------------------------------------------------------- /python/hidet/distributed/store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/distributed/store.py -------------------------------------------------------------------------------- /python/hidet/drivers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/drivers/__init__.py -------------------------------------------------------------------------------- /python/hidet/drivers/build_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/drivers/build_graph.py -------------------------------------------------------------------------------- /python/hidet/drivers/build_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/drivers/build_module.py -------------------------------------------------------------------------------- /python/hidet/drivers/build_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/drivers/build_task.py -------------------------------------------------------------------------------- /python/hidet/drivers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/drivers/utils.py -------------------------------------------------------------------------------- /python/hidet/ffi/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ffi/__init__.py -------------------------------------------------------------------------------- /python/hidet/ffi/array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ffi/array.py -------------------------------------------------------------------------------- /python/hidet/ffi/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ffi/callbacks.py -------------------------------------------------------------------------------- /python/hidet/ffi/convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ffi/convert.py -------------------------------------------------------------------------------- /python/hidet/ffi/crt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ffi/crt.py -------------------------------------------------------------------------------- /python/hidet/ffi/ffi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ffi/ffi.py -------------------------------------------------------------------------------- /python/hidet/ffi/runtime_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ffi/runtime_api.py -------------------------------------------------------------------------------- /python/hidet/ffi/shared_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ffi/shared_lib.py -------------------------------------------------------------------------------- /python/hidet/ffi/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ffi/utils.py -------------------------------------------------------------------------------- /python/hidet/graph/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/common.py -------------------------------------------------------------------------------- /python/hidet/graph/flow_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/flow_graph.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/onnx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/onnx/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/onnx/availability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/onnx/availability.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/onnx/onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/onnx/onnx.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/onnx/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/onnx/utils.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/torch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/torch/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/torch/availability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/torch/availability.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/torch/dynamo_backends.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/torch/dynamo_backends.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/torch/dynamo_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/torch/dynamo_config.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/torch/flow_graph_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/torch/flow_graph_cache.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/torch/interpreter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/torch/interpreter.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/torch/register_methods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/torch/register_methods.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/torch/register_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/torch/register_modules.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/torch/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/torch/registry.py -------------------------------------------------------------------------------- /python/hidet/graph/frontend/torch/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/frontend/torch/utils.py -------------------------------------------------------------------------------- /python/hidet/graph/graph_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/graph_utils/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/graph_utils/functors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/graph_utils/functors.py -------------------------------------------------------------------------------- /python/hidet/graph/impl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/impl/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/impl/dlpack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/impl/dlpack.py -------------------------------------------------------------------------------- /python/hidet/graph/impl/graph_impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/impl/graph_impl.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/activations.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/attention.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/container.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/convolutions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/convolutions.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/identity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/identity.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/linear.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/module.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/norms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/norms.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/poolings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/poolings.py -------------------------------------------------------------------------------- /python/hidet/graph/nn/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/nn/transforms.py -------------------------------------------------------------------------------- /python/hidet/graph/operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/operator.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/activation.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/arithmetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/arithmetic.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/arithmetic_resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/arithmetic_resolve.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/attention/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/attention/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/attention/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/attention/attention.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/attention/attention_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/attention/attention_mask.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/compare.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/complex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/complex.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv1d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv1d/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv1d/conv1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv1d/conv1d.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv1d/conv1d_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv1d/conv1d_gemm.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv1d/resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv1d/resolve.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv1d/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv1d/utils.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv1d_transpose/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv1d_transpose/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv2d/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv2d/conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv2d/conv2d.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv2d/conv2d_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv2d/conv2d_gemm.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv2d/conv2d_winograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv2d/conv2d_winograd.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv2d/resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv2d/resolve.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv2d/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv2d/utils.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv2d_transpose/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv2d_transpose/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv2d_transpose/resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv2d_transpose/resolve.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv3d/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv3d/conv3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv3d/conv3d.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv3d/conv3d_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv3d/conv3d_gemm.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv3d/resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv3d/resolve.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv3d/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv3d/utils.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/conv3d_transpose/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/conv3d_transpose/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/create.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/create.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/cumulative.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/cumulative.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/distributed.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/einops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/einops.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/embedding_bag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/embedding_bag.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/fusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/fusion/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/fusion/fused_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/fusion/fused_operator.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/image.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/linear.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/cuda_batch_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/cuda_batch_matmul.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/hip_batch_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/hip_batch_matmul.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/matmul.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/matmul_cublas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/matmul_cublas.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/matmul_f16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/matmul_f16.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/matmul_f16_cute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/matmul_f16_cute.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/matmul_f16_sm90.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/matmul_f16_sm90.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/matmul_f32_x86.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/matmul_f32_x86.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/matmul_f8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/matmul_f8.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/matmul/resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/matmul/resolve.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/normalize/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/normalize/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/normalize/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/normalize/layers.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/normalize/lp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/normalize/lp.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/normalize/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/normalize/norm.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/normalize/resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/normalize/resolve.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/opaque.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/opaque.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/pool.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/quant/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/quant/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/quant/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/quant/matmul.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/quant/matmul_f16_i8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/quant/matmul_f16_i8.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/quant/matmul_f16_i8_atomic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/quant/matmul_f16_i8_atomic.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/quant/resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/quant/resolve.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/quant/symmetric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/quant/symmetric.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/reduce/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/reduce/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/reduce/reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/reduce/reduce.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/reduce/resolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/reduce/resolve.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/scatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/scatter.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/softmax.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/special.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/special.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/transfer.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/transform.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/transpose2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/transpose2d.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/utils/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/utils/schedule_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/utils/schedule_utils.py -------------------------------------------------------------------------------- /python/hidet/graph/ops/utils/tensor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/ops/utils/tensor_utils.py -------------------------------------------------------------------------------- /python/hidet/graph/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/tensor.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/base.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/conv_channel_last.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/conv_channel_last.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/eliminate_barrier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/eliminate_barrier.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/fuse_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/fuse_operator.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/graph_patterns/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/graph_patterns/base.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/instruments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/instruments/__init__.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/instruments/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/instruments/base.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/resolve_variant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/resolve_variant.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/selective_quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/selective_quantize.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/subgraph_rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/subgraph_rewrite.py -------------------------------------------------------------------------------- /python/hidet/graph/transforms/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/graph/transforms/utils.py -------------------------------------------------------------------------------- /python/hidet/hip/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/hip/__init__.py -------------------------------------------------------------------------------- /python/hidet/hip/capability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/hip/capability.py -------------------------------------------------------------------------------- /python/hidet/hip/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/hip/device.py -------------------------------------------------------------------------------- /python/hidet/hip/event.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/hip/event.py -------------------------------------------------------------------------------- /python/hidet/hip/ffi.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/hidet/hip/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/hip/graph.py -------------------------------------------------------------------------------- /python/hidet/hip/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/hip/memory.py -------------------------------------------------------------------------------- /python/hidet/hip/stream.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/hip/stream.py -------------------------------------------------------------------------------- /python/hidet/ir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/analyzers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/analyzers/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/analyzers/bound_analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/analyzers/bound_analyzer.py -------------------------------------------------------------------------------- /python/hidet/ir/builders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/builders/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/builders/func_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/builders/func_builder.py -------------------------------------------------------------------------------- /python/hidet/ir/builders/stmt_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/builders/stmt_builder.py -------------------------------------------------------------------------------- /python/hidet/ir/compute/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/compute/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/compute/cops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/compute/cops/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/compute/cops/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/compute/cops/matmul.py -------------------------------------------------------------------------------- /python/hidet/ir/compute/cops/pad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/compute/cops/pad.py -------------------------------------------------------------------------------- /python/hidet/ir/compute/cops/reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/compute/cops/reduce.py -------------------------------------------------------------------------------- /python/hidet/ir/compute/primitives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/compute/primitives.py -------------------------------------------------------------------------------- /python/hidet/ir/compute/reduce_operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/compute/reduce_operations.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/algorithm.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/collective/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/collective/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/collective/copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/collective/copy.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/contexts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/contexts.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/expr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/expr.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/int_tuple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/int_tuple.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/layout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/layout.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/algorithm.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/arithmetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/arithmetic.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/copy.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/misc.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/mma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/mma.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/partition.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/rearrange.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/rearrange.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/reduce.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/subtensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/subtensor.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/ops/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/ops/tensor.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/swizzle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/swizzle.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/type.py -------------------------------------------------------------------------------- /python/hidet/ir/cute/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/cute/typing.py -------------------------------------------------------------------------------- /python/hidet/ir/dialects/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dialects/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/dialects/pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dialects/pattern.py -------------------------------------------------------------------------------- /python/hidet/ir/dtypes/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dtypes/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/dtypes/boolean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dtypes/boolean.py -------------------------------------------------------------------------------- /python/hidet/ir/dtypes/complex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dtypes/complex.py -------------------------------------------------------------------------------- /python/hidet/ir/dtypes/floats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dtypes/floats.py -------------------------------------------------------------------------------- /python/hidet/ir/dtypes/integer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dtypes/integer.py -------------------------------------------------------------------------------- /python/hidet/ir/dtypes/integer_subbyte.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dtypes/integer_subbyte.py -------------------------------------------------------------------------------- /python/hidet/ir/dtypes/promotion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dtypes/promotion.py -------------------------------------------------------------------------------- /python/hidet/ir/dtypes/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dtypes/utils.py -------------------------------------------------------------------------------- /python/hidet/ir/dtypes/vector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/dtypes/vector.py -------------------------------------------------------------------------------- /python/hidet/ir/expr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/expr.py -------------------------------------------------------------------------------- /python/hidet/ir/func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/func.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/base_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/base_functor.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/compute_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/compute_functor.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/cute_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/cute_functor.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/expr_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/expr_functor.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/ir_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/ir_functor.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/layout_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/layout_functor.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/mapping_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/mapping_functor.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/module_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/module_functor.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/stmt_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/stmt_functor.py -------------------------------------------------------------------------------- /python/hidet/ir/functors/type_functor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/functors/type_functor.py -------------------------------------------------------------------------------- /python/hidet/ir/layout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/layout.py -------------------------------------------------------------------------------- /python/hidet/ir/library/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/library/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/library/cuda/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/library/cuda/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/library/cuda/cublas/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/library/cuda/cublas/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/library/cuda/cublas/kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/library/cuda/cublas/kernels.py -------------------------------------------------------------------------------- /python/hidet/ir/library/cuda/cublas/regs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/library/cuda/cublas/regs.py -------------------------------------------------------------------------------- /python/hidet/ir/library/cuda/matmul/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/library/cuda/matmul/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/library/cuda/matmul/simt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/library/cuda/matmul/simt.py -------------------------------------------------------------------------------- /python/hidet/ir/library/tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/library/tune.py -------------------------------------------------------------------------------- /python/hidet/ir/library/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/library/utils.py -------------------------------------------------------------------------------- /python/hidet/ir/mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/mapping.py -------------------------------------------------------------------------------- /python/hidet/ir/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/module.py -------------------------------------------------------------------------------- /python/hidet/ir/node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/node.py -------------------------------------------------------------------------------- /python/hidet/ir/polinomial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/polinomial.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/complex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/complex.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/atomic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/atomic.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/avx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/avx.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/avx_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/avx_helper.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/math/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/math/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/math/bfloat16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/math/bfloat16.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/math/float16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/math/float16.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/math/float32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/math/float32.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/math/float64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/math/float64.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/math/int32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/math/int32.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cpu/math/int64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cpu/math/int64.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/atomic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/atomic.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/barrier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/barrier.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/cluster.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/copy_tma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/copy_tma.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/cp_async.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/cp_async.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/cvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/cvt.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/cvta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/cvta.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/errchk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/errchk.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/fastintdiv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/fastintdiv.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/funcs.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/half.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/half.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/ldst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/ldst.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/lop3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/lop3.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/bfloat16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/bfloat16.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/complex128.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/complex128.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/complex64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/complex64.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/float16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/float16.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/float16x2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/float16x2.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/float32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/float32.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/float64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/float64.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/float8e4m3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/float8e4m3.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/float8e5m2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/float8e5m2.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/int32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/int32.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/math/int64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/math/int64.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/memcpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/memcpy.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/mma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/mma.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/mutex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/mutex.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/nccl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/nccl.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/prmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/prmt.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/setmaxnreg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/setmaxnreg.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/shfl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/shfl.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/smem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/smem.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/sync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/sync.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/tcgen05_cp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/tcgen05_cp.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/tcgen05_ldst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/tcgen05_ldst.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/tensor_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/tensor_map.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/time.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/tmem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/tmem.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/vars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/vars.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/wgmma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/wgmma.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/cuda/wmma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/cuda/wmma.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/debug.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/func.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/hip/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/hip/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/hip/buffer_addr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/hip/buffer_addr.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/hip/errchk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/hip/errchk.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/hip/lds_sync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/hip/lds_sync.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/hip/math/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/hip/math/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/hip/math/float16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/hip/math/float16.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/hip/math/float32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/hip/math/float32.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/hip/math/int32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/hip/math/int32.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/hip/mfma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/hip/mfma.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/hip/vars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/hip/vars.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/math.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/runtime.py -------------------------------------------------------------------------------- /python/hidet/ir/primitives/vars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/primitives/vars.py -------------------------------------------------------------------------------- /python/hidet/ir/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/schedulers/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/schedulers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/schedulers/base.py -------------------------------------------------------------------------------- /python/hidet/ir/schedulers/cpu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/schedulers/cpu/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/schedulers/cpu/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/schedulers/cpu/scheduler.py -------------------------------------------------------------------------------- /python/hidet/ir/schedulers/cuda/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/schedulers/cuda/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/schedulers/cuda/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/schedulers/cuda/scheduler.py -------------------------------------------------------------------------------- /python/hidet/ir/stmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/stmt.py -------------------------------------------------------------------------------- /python/hidet/ir/target.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/target.py -------------------------------------------------------------------------------- /python/hidet/ir/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/task.py -------------------------------------------------------------------------------- /python/hidet/ir/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/tools/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/tools/free_var_collector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/tools/free_var_collector.py -------------------------------------------------------------------------------- /python/hidet/ir/tools/hasher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/tools/hasher.py -------------------------------------------------------------------------------- /python/hidet/ir/tools/ir_dumper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/tools/ir_dumper.py -------------------------------------------------------------------------------- /python/hidet/ir/tools/printer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/tools/printer.py -------------------------------------------------------------------------------- /python/hidet/ir/tools/renamer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/tools/renamer.py -------------------------------------------------------------------------------- /python/hidet/ir/tools/rewriter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/tools/rewriter.py -------------------------------------------------------------------------------- /python/hidet/ir/tools/simplifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/tools/simplifier.py -------------------------------------------------------------------------------- /python/hidet/ir/tools/type_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/tools/type_infer.py -------------------------------------------------------------------------------- /python/hidet/ir/tools/util_functors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/tools/util_functors.py -------------------------------------------------------------------------------- /python/hidet/ir/type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/type.py -------------------------------------------------------------------------------- /python/hidet/ir/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/utils/__init__.py -------------------------------------------------------------------------------- /python/hidet/ir/utils/broadcast_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/utils/broadcast_utils.py -------------------------------------------------------------------------------- /python/hidet/ir/utils/call_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/utils/call_graph.py -------------------------------------------------------------------------------- /python/hidet/ir/utils/expr_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/utils/expr_utils.py -------------------------------------------------------------------------------- /python/hidet/ir/utils/hash_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/utils/hash_sum.py -------------------------------------------------------------------------------- /python/hidet/ir/utils/index_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/utils/index_transform.py -------------------------------------------------------------------------------- /python/hidet/ir/utils/type_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/ir/utils/type_utils.py -------------------------------------------------------------------------------- /python/hidet/kernels/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/kernels/__init__.py -------------------------------------------------------------------------------- /python/hidet/lang/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/__init__.py -------------------------------------------------------------------------------- /python/hidet/lang/attrs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/attrs.py -------------------------------------------------------------------------------- /python/hidet/lang/attrs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/attrs/__init__.py -------------------------------------------------------------------------------- /python/hidet/lang/attrs/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/attrs/cuda.py -------------------------------------------------------------------------------- /python/hidet/lang/attrs/hip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/attrs/hip.py -------------------------------------------------------------------------------- /python/hidet/lang/constructs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/constructs/__init__.py -------------------------------------------------------------------------------- /python/hidet/lang/constructs/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/constructs/context.py -------------------------------------------------------------------------------- /python/hidet/lang/constructs/declare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/constructs/declare.py -------------------------------------------------------------------------------- /python/hidet/lang/constructs/loops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/constructs/loops.py -------------------------------------------------------------------------------- /python/hidet/lang/constructs/meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/constructs/meta.py -------------------------------------------------------------------------------- /python/hidet/lang/cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/cpu.py -------------------------------------------------------------------------------- /python/hidet/lang/cuda/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/cuda/__init__.py -------------------------------------------------------------------------------- /python/hidet/lang/cuda/contexts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/cuda/contexts.py -------------------------------------------------------------------------------- /python/hidet/lang/hip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/hip.py -------------------------------------------------------------------------------- /python/hidet/lang/layout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/layout.py -------------------------------------------------------------------------------- /python/hidet/lang/mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/mapping.py -------------------------------------------------------------------------------- /python/hidet/lang/runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/runtime.py -------------------------------------------------------------------------------- /python/hidet/lang/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/script.py -------------------------------------------------------------------------------- /python/hidet/lang/transpiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/transpiler.py -------------------------------------------------------------------------------- /python/hidet/lang/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/lang/types.py -------------------------------------------------------------------------------- /python/hidet/libinfo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/libinfo.py -------------------------------------------------------------------------------- /python/hidet/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/logging.py -------------------------------------------------------------------------------- /python/hidet/option.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/option.py -------------------------------------------------------------------------------- /python/hidet/runtime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/runtime/__init__.py -------------------------------------------------------------------------------- /python/hidet/runtime/compiled_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/runtime/compiled_graph.py -------------------------------------------------------------------------------- /python/hidet/runtime/compiled_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/runtime/compiled_module.py -------------------------------------------------------------------------------- /python/hidet/runtime/compiled_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/runtime/compiled_task.py -------------------------------------------------------------------------------- /python/hidet/runtime/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/runtime/device.py -------------------------------------------------------------------------------- /python/hidet/runtime/storage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/runtime/storage.py -------------------------------------------------------------------------------- /python/hidet/runtime/utils/dispatch_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/runtime/utils/dispatch_table.py -------------------------------------------------------------------------------- /python/hidet/testing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/__init__.py -------------------------------------------------------------------------------- /python/hidet/testing/capture_stdout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/capture_stdout.py -------------------------------------------------------------------------------- /python/hidet/testing/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/models/__init__.py -------------------------------------------------------------------------------- /python/hidet/testing/models/gemma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/models/gemma.py -------------------------------------------------------------------------------- /python/hidet/testing/models/gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/models/gpt2.py -------------------------------------------------------------------------------- /python/hidet/testing/models/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/models/llama.py -------------------------------------------------------------------------------- /python/hidet/testing/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/models/resnet.py -------------------------------------------------------------------------------- /python/hidet/testing/onnx_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/onnx_models.py -------------------------------------------------------------------------------- /python/hidet/testing/onnx_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/onnx_utils.py -------------------------------------------------------------------------------- /python/hidet/testing/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/torch_utils.py -------------------------------------------------------------------------------- /python/hidet/testing/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/testing/utils.py -------------------------------------------------------------------------------- /python/hidet/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/__init__.py -------------------------------------------------------------------------------- /python/hidet/transforms/add_explicit_cast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/add_explicit_cast.py -------------------------------------------------------------------------------- /python/hidet/transforms/add_hints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/add_hints.py -------------------------------------------------------------------------------- /python/hidet/transforms/annotate_header_and_libs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/annotate_header_and_libs.py -------------------------------------------------------------------------------- /python/hidet/transforms/attach_hash_to_signature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/attach_hash_to_signature.py -------------------------------------------------------------------------------- /python/hidet/transforms/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/base.py -------------------------------------------------------------------------------- /python/hidet/transforms/check_launch_configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/check_launch_configuration.py -------------------------------------------------------------------------------- /python/hidet/transforms/convert_div_to_fastintdiv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/convert_div_to_fastintdiv.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/__init__.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/analysis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/analysis/__init__.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/cuda/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/cuda/__init__.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/cuda/cost_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/cuda/cost_model.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/cuda/lower_ops/copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/cuda/lower_ops/copy.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/cuda/lower_ops/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/cuda/lower_ops/misc.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/cuda/lower_ops/mma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/cuda/lower_ops/mma.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/cuda/lower_ops/reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/cuda/lower_ops/reduce.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/cuda/lower_ops/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/cuda/lower_ops/tensor.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/cuda/tma_layout_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/cuda/tma_layout_utils.py -------------------------------------------------------------------------------- /python/hidet/transforms/cute/generic/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/cute/generic/__init__.py -------------------------------------------------------------------------------- /python/hidet/transforms/declare_to_let.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/declare_to_let.py -------------------------------------------------------------------------------- /python/hidet/transforms/expand_let_expr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/expand_let_expr.py -------------------------------------------------------------------------------- /python/hidet/transforms/expand_repeat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/expand_repeat.py -------------------------------------------------------------------------------- /python/hidet/transforms/explicit_unroll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/explicit_unroll.py -------------------------------------------------------------------------------- /python/hidet/transforms/flatten_tensor_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/flatten_tensor_index.py -------------------------------------------------------------------------------- /python/hidet/transforms/flatten_tensor_slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/flatten_tensor_slice.py -------------------------------------------------------------------------------- /python/hidet/transforms/generate_launch_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/generate_launch_func.py -------------------------------------------------------------------------------- /python/hidet/transforms/inline_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/inline_function.py -------------------------------------------------------------------------------- /python/hidet/transforms/inline_let_stmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/inline_let_stmt.py -------------------------------------------------------------------------------- /python/hidet/transforms/instantiate_symbols.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/instantiate_symbols.py -------------------------------------------------------------------------------- /python/hidet/transforms/instruments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/instruments/__init__.py -------------------------------------------------------------------------------- /python/hidet/transforms/instruments/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/instruments/base.py -------------------------------------------------------------------------------- /python/hidet/transforms/lower_integer_subbyte.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/lower_integer_subbyte.py -------------------------------------------------------------------------------- /python/hidet/transforms/lower_protect_access.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/lower_protect_access.py -------------------------------------------------------------------------------- /python/hidet/transforms/lower_special_cast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/lower_special_cast.py -------------------------------------------------------------------------------- /python/hidet/transforms/lower_task_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/lower_task_mapping.py -------------------------------------------------------------------------------- /python/hidet/transforms/normalize_const_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/normalize_const_tensor.py -------------------------------------------------------------------------------- /python/hidet/transforms/propagate_launch_bound.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/propagate_launch_bound.py -------------------------------------------------------------------------------- /python/hidet/transforms/rule_based_simplifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/rule_based_simplifier.py -------------------------------------------------------------------------------- /python/hidet/transforms/simplify_addition_chain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/simplify_addition_chain.py -------------------------------------------------------------------------------- /python/hidet/transforms/simplify_stmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/simplify_stmt.py -------------------------------------------------------------------------------- /python/hidet/transforms/spatial_simplification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/spatial_simplification.py -------------------------------------------------------------------------------- /python/hidet/transforms/task_mapping_bound_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/task_mapping_bound_check.py -------------------------------------------------------------------------------- /python/hidet/transforms/unify_global_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/transforms/unify_global_objects.py -------------------------------------------------------------------------------- /python/hidet/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/__init__.py -------------------------------------------------------------------------------- /python/hidet/utils/benchmark/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/benchmark/__init__.py -------------------------------------------------------------------------------- /python/hidet/utils/benchmark/bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/benchmark/bench.py -------------------------------------------------------------------------------- /python/hidet/utils/benchmark/gpu_freq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/benchmark/gpu_freq.py -------------------------------------------------------------------------------- /python/hidet/utils/benchmark/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/benchmark/utils.py -------------------------------------------------------------------------------- /python/hidet/utils/cache_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/cache_utils.py -------------------------------------------------------------------------------- /python/hidet/utils/counters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/counters.py -------------------------------------------------------------------------------- /python/hidet/utils/cuda_sanitizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/cuda_sanitizer.py -------------------------------------------------------------------------------- /python/hidet/utils/dataclass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/dataclass.py -------------------------------------------------------------------------------- /python/hidet/utils/doc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/doc.py -------------------------------------------------------------------------------- /python/hidet/utils/exiting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/exiting.py -------------------------------------------------------------------------------- /python/hidet/utils/fault_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/fault_handler.py -------------------------------------------------------------------------------- /python/hidet/utils/files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/files.py -------------------------------------------------------------------------------- /python/hidet/utils/folder_lock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/folder_lock.py -------------------------------------------------------------------------------- /python/hidet/utils/gc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/gc.py -------------------------------------------------------------------------------- /python/hidet/utils/git_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/git_utils.py -------------------------------------------------------------------------------- /python/hidet/utils/multiprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/multiprocess.py -------------------------------------------------------------------------------- /python/hidet/utils/namer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/namer.py -------------------------------------------------------------------------------- /python/hidet/utils/ncu_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/ncu_utils.py -------------------------------------------------------------------------------- /python/hidet/utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/net_utils.py -------------------------------------------------------------------------------- /python/hidet/utils/netron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/netron.py -------------------------------------------------------------------------------- /python/hidet/utils/nsys_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/nsys_utils.py -------------------------------------------------------------------------------- /python/hidet/utils/omniperf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/omniperf_utils.py -------------------------------------------------------------------------------- /python/hidet/utils/ort_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/ort_utils.py -------------------------------------------------------------------------------- /python/hidet/utils/overrides.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/overrides.py -------------------------------------------------------------------------------- /python/hidet/utils/py.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/py.py -------------------------------------------------------------------------------- /python/hidet/utils/stack_limit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/stack_limit.py -------------------------------------------------------------------------------- /python/hidet/utils/structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/structure.py -------------------------------------------------------------------------------- /python/hidet/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/torch_utils.py -------------------------------------------------------------------------------- /python/hidet/utils/trace_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/trace_utils.py -------------------------------------------------------------------------------- /python/hidet/utils/transformers_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/utils/transformers_utils.py -------------------------------------------------------------------------------- /python/hidet/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/python/hidet/version.py -------------------------------------------------------------------------------- /scripts/bench/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/bench/README.md -------------------------------------------------------------------------------- /scripts/bench/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/bench/benchmark.py -------------------------------------------------------------------------------- /scripts/bench/requirements.txt: -------------------------------------------------------------------------------- 1 | schedule 2 | importlib_metadata 3 | -------------------------------------------------------------------------------- /scripts/bench/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/bench/run.py -------------------------------------------------------------------------------- /scripts/lint/.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/lint/.clang-format -------------------------------------------------------------------------------- /scripts/lint/_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/lint/_format.py -------------------------------------------------------------------------------- /scripts/lint/add_copyright.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/lint/add_copyright.py -------------------------------------------------------------------------------- /scripts/lint/format.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/lint/format.sh -------------------------------------------------------------------------------- /scripts/lint/lint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/lint/lint.sh -------------------------------------------------------------------------------- /scripts/lint/pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/lint/pylintrc -------------------------------------------------------------------------------- /scripts/nightly-builds/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/nightly-builds/README.md -------------------------------------------------------------------------------- /scripts/nightly-builds/add-crontab-record.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/nightly-builds/add-crontab-record.sh -------------------------------------------------------------------------------- /scripts/nightly-builds/update-nightly.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/nightly-builds/update-nightly.sh -------------------------------------------------------------------------------- /scripts/regression/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/regression/email_sender.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/regression/email_sender.py -------------------------------------------------------------------------------- /scripts/regression/model_performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/regression/model_performance.py -------------------------------------------------------------------------------- /scripts/regression/op_performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/regression/op_performance.py -------------------------------------------------------------------------------- /scripts/regression/regression_data.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/regression/regression_data.json -------------------------------------------------------------------------------- /scripts/regression/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/regression/requirements.txt -------------------------------------------------------------------------------- /scripts/regression/result_entry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/regression/result_entry.py -------------------------------------------------------------------------------- /scripts/regression/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/regression/run.py -------------------------------------------------------------------------------- /scripts/wheel/build_wheel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/wheel/build_wheel.sh -------------------------------------------------------------------------------- /scripts/wheel/build_wheel_manylinux_2_28_x86_64.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/wheel/build_wheel_manylinux_2_28_x86_64.sh -------------------------------------------------------------------------------- /scripts/wheel/current_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/wheel/current_version.py -------------------------------------------------------------------------------- /scripts/wheel/update_nightly.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/wheel/update_nightly.sh -------------------------------------------------------------------------------- /scripts/wheel/upload_wheel_to_pypi.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/scripts/wheel/upload_wheel_to_pypi.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/setup.py -------------------------------------------------------------------------------- /src/hidet/empty.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/empty.cpp -------------------------------------------------------------------------------- /src/hidet/runtime/callbacks.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/callbacks.cpp -------------------------------------------------------------------------------- /src/hidet/runtime/cpu/context.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/cpu/context.cpp -------------------------------------------------------------------------------- /src/hidet/runtime/cuda/context.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/cuda/context.cpp -------------------------------------------------------------------------------- /src/hidet/runtime/cuda/cublas.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/cuda/cublas.cpp -------------------------------------------------------------------------------- /src/hidet/runtime/cuda/cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/cuda/cuda.cpp -------------------------------------------------------------------------------- /src/hidet/runtime/cuda/cudnn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/cuda/cudnn.cpp -------------------------------------------------------------------------------- /src/hidet/runtime/cuda/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/cuda/utils.h -------------------------------------------------------------------------------- /src/hidet/runtime/hip/context.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/hip/context.cpp -------------------------------------------------------------------------------- /src/hidet/runtime/int_fastdiv.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/int_fastdiv.cpp -------------------------------------------------------------------------------- /src/hidet/runtime/logging.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/logging.cpp -------------------------------------------------------------------------------- /src/hidet/runtime/symbols.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/src/hidet/runtime/symbols.cpp -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/README.md -------------------------------------------------------------------------------- /tests/benchmarks/bench_dynamic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/benchmarks/bench_dynamic.py -------------------------------------------------------------------------------- /tests/benchmarks/bench_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/benchmarks/bench_op.py -------------------------------------------------------------------------------- /tests/benchmarks/bench_op_torch_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/benchmarks/bench_op_torch_api.py -------------------------------------------------------------------------------- /tests/benchmarks/bench_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/benchmarks/bench_task.py -------------------------------------------------------------------------------- /tests/benchmarks/bench_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/benchmarks/bench_transformer.py -------------------------------------------------------------------------------- /tests/benchmarks/bench_transformer_comptime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/benchmarks/bench_transformer_comptime.py -------------------------------------------------------------------------------- /tests/benchmarks/bench_vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/benchmarks/bench_vision.py -------------------------------------------------------------------------------- /tests/benchmarks/run_configs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/benchmarks/run_configs.json -------------------------------------------------------------------------------- /tests/benchmarks/run_configs_full.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/benchmarks/run_configs_full.json -------------------------------------------------------------------------------- /tests/benchmarks/run_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/benchmarks/run_tests.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/cuda/test_cublas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cuda/test_cublas.py -------------------------------------------------------------------------------- /tests/cuda/test_cuda_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cuda/test_cuda_graph.py -------------------------------------------------------------------------------- /tests/cuda/test_cudnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cuda/test_cudnn.py -------------------------------------------------------------------------------- /tests/cute_fusion/fusion_bench_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_fusion/fusion_bench_utils.py -------------------------------------------------------------------------------- /tests/cute_fusion/test_epilogue_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_fusion/test_epilogue_fusion.py -------------------------------------------------------------------------------- /tests/cute_fusion/test_issue436.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_fusion/test_issue436.py -------------------------------------------------------------------------------- /tests/cute_fusion/test_matmul_bias_cast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_fusion/test_matmul_bias_cast.py -------------------------------------------------------------------------------- /tests/cute_fusion/test_matmul_cast_dynamic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_fusion/test_matmul_cast_dynamic.py -------------------------------------------------------------------------------- /tests/cute_fusion/test_matmul_residual_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_fusion/test_matmul_residual_bias.py -------------------------------------------------------------------------------- /tests/cute_fusion/test_matmul_residual_permute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_fusion/test_matmul_residual_permute.py -------------------------------------------------------------------------------- /tests/cute_fusion/test_matmul_riskfuel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_fusion/test_matmul_riskfuel.py -------------------------------------------------------------------------------- /tests/cute_fusion/test_matmul_standalone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_fusion/test_matmul_standalone.py -------------------------------------------------------------------------------- /tests/cute_kernel/gemm_quant_nbit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/gemm_quant_nbit.py -------------------------------------------------------------------------------- /tests/cute_kernel/quant_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/quant_utils.py -------------------------------------------------------------------------------- /tests/cute_kernel/quantized_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/quantized_linear.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_attention_bwd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_attention_bwd.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_cute_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_cute_attention.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_flash_decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_flash_decoding.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_gemm_quant_nbit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_gemm_quant_nbit.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_gemv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_gemv.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_hopper_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_hopper_gemm.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_hopper_gemm_e2e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_hopper_gemm_e2e.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_mlp.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_moe_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_moe_align.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_quant_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_quant_linear.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_rmsnorm.py -------------------------------------------------------------------------------- /tests/cute_kernel/test_weight_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_kernel/test_weight_quantization.py -------------------------------------------------------------------------------- /tests/cute_ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/cute_ops/test_arithmetic2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_ops/test_arithmetic2.py -------------------------------------------------------------------------------- /tests/cute_ops/test_collective.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_ops/test_collective.py -------------------------------------------------------------------------------- /tests/cute_ops/test_ldst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_ops/test_ldst.py -------------------------------------------------------------------------------- /tests/cute_ops/test_mma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_ops/test_mma.py -------------------------------------------------------------------------------- /tests/cute_ops/test_rearrange.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_ops/test_rearrange.py -------------------------------------------------------------------------------- /tests/cute_ops/test_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_ops/test_reduce.py -------------------------------------------------------------------------------- /tests/cute_ops/test_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/cute_ops/test_tensor.py -------------------------------------------------------------------------------- /tests/distributed/test_file_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/distributed/test_file_store.py -------------------------------------------------------------------------------- /tests/distributed/test_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/distributed/test_op.py -------------------------------------------------------------------------------- /tests/distributed/test_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/distributed/test_runtime.py -------------------------------------------------------------------------------- /tests/distributed/test_tcp_store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/distributed/test_tcp_store.py -------------------------------------------------------------------------------- /tests/distributed/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/distributed/utils.py -------------------------------------------------------------------------------- /tests/flowgraph/test_graph_visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/flowgraph/test_graph_visualization.py -------------------------------------------------------------------------------- /tests/frontends/onnx/test_onnx_slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/onnx/test_onnx_slice.py -------------------------------------------------------------------------------- /tests/frontends/torch/models/test_torch_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/models/test_torch_bert.py -------------------------------------------------------------------------------- /tests/frontends/torch/models/test_torch_pegasus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/models/test_torch_pegasus.py -------------------------------------------------------------------------------- /tests/frontends/torch/models/test_torch_resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/models/test_torch_resnet50.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_einops_rearrange.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_einops_rearrange.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_flow_graph_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_flow_graph_cache.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_activation.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_arithmetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_arithmetic.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_conv1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_conv1d.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_conv2d.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_conv3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_conv3d.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_creation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_creation.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_dyn_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_dyn_shape.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_fxgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_fxgraph.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_image.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_inplace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_inplace.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_mix_cuda_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_mix_cuda_cpu.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_mul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_mul.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_norm.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_pooling.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_reduce.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_sdpa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_sdpa.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_split.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_stream.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_stream.py -------------------------------------------------------------------------------- /tests/frontends/torch/test_torch_to.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/frontends/torch/test_torch_to.py -------------------------------------------------------------------------------- /tests/hip/test_hip_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/hip/test_hip_graph.py -------------------------------------------------------------------------------- /tests/ir/dialects/test_pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/dialects/test_pattern.py -------------------------------------------------------------------------------- /tests/ir/dtypes/test_fp8e4m3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/dtypes/test_fp8e4m3.py -------------------------------------------------------------------------------- /tests/ir/dtypes/test_fp8e5m2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/dtypes/test_fp8e5m2.py -------------------------------------------------------------------------------- /tests/ir/functors/test_persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/functors/test_persistence.py -------------------------------------------------------------------------------- /tests/ir/parser/test_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/parser/test_parser.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_barrier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_barrier.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_cluster.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_copy_tma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_copy_tma.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_exp2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_exp2.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_half.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_half.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_lop3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_lop3.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_mma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_mma.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_prmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_prmt.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_tcgen05_cp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_tcgen05_cp.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_tcgen05_ldst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_tcgen05_ldst.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_tmem_alloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_tmem_alloc.py -------------------------------------------------------------------------------- /tests/ir/primitives/cuda/test_wgmma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/cuda/test_wgmma.py -------------------------------------------------------------------------------- /tests/ir/primitives/hip/test_buffer_addr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/hip/test_buffer_addr.py -------------------------------------------------------------------------------- /tests/ir/primitives/hip/test_mfma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/primitives/hip/test_mfma.py -------------------------------------------------------------------------------- /tests/ir/test_expr_const_fold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/test_expr_const_fold.py -------------------------------------------------------------------------------- /tests/ir/test_int_subbyte.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/test_int_subbyte.py -------------------------------------------------------------------------------- /tests/ir/test_layout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/test_layout.py -------------------------------------------------------------------------------- /tests/ir/test_primitives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/test_primitives.py -------------------------------------------------------------------------------- /tests/ir/test_symbol_var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/ir/test_symbol_var.py -------------------------------------------------------------------------------- /tests/minimal/test_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/minimal/test_add.py -------------------------------------------------------------------------------- /tests/models/test_gemma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/models/test_gemma.py -------------------------------------------------------------------------------- /tests/models/test_gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/models/test_gpt2.py -------------------------------------------------------------------------------- /tests/models/test_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/models/test_llama.py -------------------------------------------------------------------------------- /tests/models/test_llama_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/models/test_llama_graph.py -------------------------------------------------------------------------------- /tests/models/test_quant_gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/models/test_quant_gpt2.py -------------------------------------------------------------------------------- /tests/multiprocessing/lazy_init_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/multiprocessing/lazy_init_sample.py -------------------------------------------------------------------------------- /tests/multiprocessing/test_lazy_initialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/multiprocessing/test_lazy_initialization.py -------------------------------------------------------------------------------- /tests/operators_core/test_activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_activation.py -------------------------------------------------------------------------------- /tests/operators_core/test_arithmetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_arithmetic.py -------------------------------------------------------------------------------- /tests/operators_core/test_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_attention.py -------------------------------------------------------------------------------- /tests/operators_core/test_compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_compare.py -------------------------------------------------------------------------------- /tests/operators_core/test_complex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_complex.py -------------------------------------------------------------------------------- /tests/operators_core/test_conv1d_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_conv1d_transpose.py -------------------------------------------------------------------------------- /tests/operators_core/test_create.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_create.py -------------------------------------------------------------------------------- /tests/operators_core/test_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_fusion.py -------------------------------------------------------------------------------- /tests/operators_core/test_identity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_identity.py -------------------------------------------------------------------------------- /tests/operators_core/test_inplace_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_inplace_operator.py -------------------------------------------------------------------------------- /tests/operators_core/test_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_matmul.py -------------------------------------------------------------------------------- /tests/operators_core/test_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_norm.py -------------------------------------------------------------------------------- /tests/operators_core/test_opaque.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_opaque.py -------------------------------------------------------------------------------- /tests/operators_core/test_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_operator.py -------------------------------------------------------------------------------- /tests/operators_core/test_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_quantization.py -------------------------------------------------------------------------------- /tests/operators_core/test_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_reduce.py -------------------------------------------------------------------------------- /tests/operators_core/test_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_softmax.py -------------------------------------------------------------------------------- /tests/operators_core/test_symmetric_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_symmetric_quant.py -------------------------------------------------------------------------------- /tests/operators_core/test_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_tensor.py -------------------------------------------------------------------------------- /tests/operators_core/test_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_transform.py -------------------------------------------------------------------------------- /tests/operators_core/test_tri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_core/test_tri.py -------------------------------------------------------------------------------- /tests/operators_vision/test_conv1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_vision/test_conv1d.py -------------------------------------------------------------------------------- /tests/operators_vision/test_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_vision/test_conv2d.py -------------------------------------------------------------------------------- /tests/operators_vision/test_conv2d_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_vision/test_conv2d_transpose.py -------------------------------------------------------------------------------- /tests/operators_vision/test_conv3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_vision/test_conv3d.py -------------------------------------------------------------------------------- /tests/operators_vision/test_conv3d_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_vision/test_conv3d_transpose.py -------------------------------------------------------------------------------- /tests/operators_vision/test_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_vision/test_image.py -------------------------------------------------------------------------------- /tests/operators_vision/test_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/operators_vision/test_pool.py -------------------------------------------------------------------------------- /tests/runtime/test_dispatch_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/runtime/test_dispatch_table.py -------------------------------------------------------------------------------- /tests/runtime/test_try_catch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/runtime/test_try_catch.py -------------------------------------------------------------------------------- /tests/script/test_assignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_assignment.py -------------------------------------------------------------------------------- /tests/script/test_comprehension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_comprehension.py -------------------------------------------------------------------------------- /tests/script/test_constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_constant.py -------------------------------------------------------------------------------- /tests/script/test_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_context.py -------------------------------------------------------------------------------- /tests/script/test_cpu_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_cpu_kernel.py -------------------------------------------------------------------------------- /tests/script/test_for_loops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_for_loops.py -------------------------------------------------------------------------------- /tests/script/test_func_import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_func_import.py -------------------------------------------------------------------------------- /tests/script/test_lambda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_lambda.py -------------------------------------------------------------------------------- /tests/script/test_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_meta.py -------------------------------------------------------------------------------- /tests/script/test_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_parallel.py -------------------------------------------------------------------------------- /tests/script/test_return_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_return_type.py -------------------------------------------------------------------------------- /tests/script/test_unroll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/script/test_unroll.py -------------------------------------------------------------------------------- /tests/transforms/test_graph_rewrites.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/transforms/test_graph_rewrites.py -------------------------------------------------------------------------------- /tests/transforms/test_inline_let_stmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/transforms/test_inline_let_stmt.py -------------------------------------------------------------------------------- /tests/transforms/test_rule_based_simplifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/transforms/test_rule_based_simplifier.py -------------------------------------------------------------------------------- /tests/transforms/test_simplify_addition_chain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/transforms/test_simplify_addition_chain.py -------------------------------------------------------------------------------- /tests/unit_tests/check_import_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/unit_tests/check_import_time.py -------------------------------------------------------------------------------- /tests/unit_tests/option/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/unit_tests/option/main.py -------------------------------------------------------------------------------- /tests/unit_tests/option/test_option_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/unit_tests/option/test_option_env.py -------------------------------------------------------------------------------- /tests/unit_tests/test_compiled_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/unit_tests/test_compiled_model.py -------------------------------------------------------------------------------- /tests/unit_tests/test_dynamic_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/unit_tests/test_dynamic_shape.py -------------------------------------------------------------------------------- /tests/unit_tests/test_frontend_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/unit_tests/test_frontend_onnx.py -------------------------------------------------------------------------------- /tests/unit_tests/test_import_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/unit_tests/test_import_time.py -------------------------------------------------------------------------------- /tests/unit_tests/test_save_lower_ir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/unit_tests/test_save_lower_ir.py -------------------------------------------------------------------------------- /tests/unit_tests/test_vllm_hidet_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/unit_tests/test_vllm_hidet_compile.py -------------------------------------------------------------------------------- /tests/utils/benchmark/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/utils/benchmark/example.py -------------------------------------------------------------------------------- /tests/utils/test_cuda_sanitizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/utils/test_cuda_sanitizer.py -------------------------------------------------------------------------------- /tests/utils/test_ncu_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/utils/test_ncu_utils.py -------------------------------------------------------------------------------- /tests/utils/test_nsys_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hidet-org/hidet/HEAD/tests/utils/test_nsys_utils.py --------------------------------------------------------------------------------