Sign in
apache
/
tvm
/
HEAD
/
.
/
tests
/
python
/
relax
tree: 05a54c3c8bdf02da537acf658d02140dcb4fe22e [
path history
]
[
tgz
]
backend/
distributed/
nvshmem/
conftest.py
frontend_nn_extern_module.cc
test_analysis.py
test_analysis_computable_at_compile_time.py
test_analysis_contains_impure_call.py
test_analysis_detect_recursion.py
test_analysis_estimate_memory_usage.py
test_analysis_struct_info_analysis.py
test_analysis_suggest_layout_transforms.py
test_analysis_well_formed.py
test_ast_printer.py
test_backend_dispatch_sampling.py
test_backend_dispatch_sort_scan.py
test_backend_transform_shape_lower.py
test_base_py_module.py
test_base_py_module_printer.py
test_base_py_module_symbolic_shape.py
test_bind_params.py
test_bind_symbolic_vars.py
test_binding_rewrite.py
test_blockbuilder_core.py
test_blockbuilder_emit_te.py
test_codegen_coreml.py
test_codegen_cublas.py
test_codegen_cudnn.py
test_codegen_cutlass.py
test_codegen_dnnl.py
test_codegen_hipblas.py
test_codegen_tensorrt.py
test_contrib_vllm.py
test_dataflow_inplace.py
test_dataflow_pattern.py
test_dataflow_rewriter.py
test_dlpack_integration.py
test_e2e_op_dynamic.py
test_eliminate_pad_branch_using_buffer_assumption.py
test_expr.py
test_expr_args_converter.py
test_expr_functor.py
test_fast_math_transform.py
test_frontend_common.py
test_frontend_dynamo.py
test_frontend_from_exported_program.py
test_frontend_from_fx.py
test_frontend_nn_debug.py
test_frontend_nn_exporter.py
test_frontend_nn_extern_module.py
test_frontend_nn_jit.py
test_frontend_nn_modules.py
test_frontend_nn_mutator.py
test_frontend_nn_op.py
test_frontend_nn_packing.py
test_frontend_nn_subroutines.py
test_frontend_nn_tensor.py
test_frontend_onnx.py
test_frontend_stablehlo.py
test_group_gemm_flashinfer.py
test_inline_functions.py
test_kill_after_last_use.py
test_lower_alloc_tensor.py
test_meta_schedule_relax_integration.py
test_op_binary.py
test_op_ccl.py
test_op_create.py
test_op_datatype.py
test_op_distributed.py
test_op_grad.py
test_op_gradient_numeric.py
test_op_image.py
test_op_index.py
test_op_inspect.py
test_op_linear_algebra.py
test_op_manipulate.py
test_op_misc.py
test_op_nn.py
test_op_nn_convolution.py
test_op_nn_pooling.py
test_op_qdq.py
test_op_sampling.py
test_op_search.py
test_op_set.py
test_op_sort.py
test_op_statistical.py
test_op_take.py
test_op_ternary.py
test_op_unary.py
test_op_view.py
test_op_vision.py
test_optimize_layout_transform.py
test_pipeline.py
test_pytorch_integration.py
test_relax_operators.py
test_relax_to_pyfunc_converter.py
test_remove_redundant_reshape.py
test_runtime_builtin.py
test_runtime_builtin_paged_attention_kv_cache_cpu.py
test_runtime_builtin_paged_attention_kv_cache_flashinfer.py
test_runtime_builtin_paged_attention_kv_cache_mla_flashinfer.py
test_runtime_builtin_paged_attention_kv_cache_mla_tir.py
test_runtime_builtin_paged_attention_kv_cache_tir.py
test_runtime_builtin_rnn_state.py
test_runtime_sampling_flashinfer.py
test_struct_info.py
test_testing_nn.py
test_tir_call_source_kernel.py
test_training_append_loss.py
test_training_loss.py
test_training_optimizer.py
test_training_optimizer_numeric.py
test_training_setup_trainer.py
test_training_trainer_numeric.py
test_transform.py
test_transform_adjust_matmul_order.py
test_transform_allocate_workspace.py
test_transform_alter_op_impl.py
test_transform_annotate_tir_op_pattern.py
test_transform_attach_attr_layout_free_buffers.py
test_transform_attach_global_symbol.py
test_transform_bind_params.py
test_transform_bind_symbolic_vars.py
test_transform_bundle_model_params.py
test_transform_canonicalize_bindings.py
test_transform_codegen_pass.py
test_transform_combine_parallel_matmul.py
test_transform_compute_prim_value.py
test_transform_convert_dataflow.py
test_transform_convert_layout.py
test_transform_cse.py
test_transform_dead_code_elimination.py
test_transform_decompose_ops.py
test_transform_expand_matmul_of_sum.py
test_transform_expand_tuple_args.py
test_transform_few_shot_tuning.py
test_transform_fold_batch_norm_to_conv2d.py
test_transform_fold_constant.py
test_transform_fuse_ops.py
test_transform_fuse_ops_by_pattern.py
test_transform_fuse_tir.py
test_transform_fuse_transpose_matmul.py
test_transform_gradient.py
test_transform_gradient_checkpoint.py
test_transform_gradient_numeric.py
test_transform_gradient_te_register.py
test_transform_inline_private_functions.py
test_transform_ipc_allreduce_rewrite.py
test_transform_lambda_lift.py
test_transform_lazy_transform_params.py
test_transform_legalize_ops.py
test_transform_legalize_ops_binary.py
test_transform_legalize_ops_ccl.py
test_transform_legalize_ops_create_datatype.py
test_transform_legalize_ops_distributed.py
test_transform_legalize_ops_grad.py
test_transform_legalize_ops_image.py
test_transform_legalize_ops_index_linear_algebra.py
test_transform_legalize_ops_manipulate.py
test_transform_legalize_ops_nn.py
test_transform_legalize_ops_qdq.py
test_transform_legalize_ops_search_statistical.py
test_transform_legalize_ops_unary.py
test_transform_lift_transform_params.py
test_transform_lower_gpu_ipc_alloc_storage.py
test_transform_merge_composite_functions.py
test_transform_meta_schedule_apply_database.py
test_transform_meta_schedule_tuning.py
test_transform_normalize.py
test_transform_normalize_global_var.py
test_transform_operator_specific_normalization.py
test_transform_realize_vdevice.py
test_transform_remove_unused_outputs.py
test_transform_remove_unused_parameters.py
test_transform_reorder_permute_dims_after_concat.py
test_transform_reorder_take_after_matmul.py
test_transform_rewrite_cuda_graph.py
test_transform_rewrite_dataflow_reshape.py
test_transform_split_layout_rewrite_preproc.py
test_transform_static_plan_block_memory.py
test_transform_to_mixed_precision.py
test_transform_topological_sort.py
test_transform_update_param_struct_info.py
test_transform_update_vdevice.py
test_tvmscript_ir_builder.py
test_tvmscript_parser.py
test_tvmscript_parser_op_arith_cmp.py
test_tvmscript_parser_op_create.py
test_tvmscript_parser_op_datatype.py
test_tvmscript_parser_op_grad.py
test_tvmscript_parser_op_image.py
test_tvmscript_parser_op_index.py
test_tvmscript_parser_op_linear_algebra.py
test_tvmscript_parser_op_manipulate.py
test_tvmscript_parser_op_nn.py
test_tvmscript_parser_op_search.py
test_tvmscript_parser_op_set.py
test_tvmscript_parser_op_sort.py
test_tvmscript_parser_op_statistical.py
test_tvmscript_parser_op_vision.py
test_tvmscript_printer_relax.py
test_tvmscript_pyfunc.py
test_utils.py
test_vm_alloc_storage_with_scope.py
test_vm_build.py
test_vm_builtin.py
test_vm_builtin_lower.py
test_vm_callback_function.py
test_vm_codegen_only.py
test_vm_codegen_tir.py
test_vm_cuda_graph.py
test_vm_execbuilder.py
test_vm_instrument.py
test_vm_multi_device.py
test_vm_profiler.py