tree: 05a54c3c8bdf02da537acf658d02140dcb4fe22e [path history] [tgz]
  1. backend/
  2. distributed/
  3. nvshmem/
  4. conftest.py
  5. frontend_nn_extern_module.cc
  6. test_analysis.py
  7. test_analysis_computable_at_compile_time.py
  8. test_analysis_contains_impure_call.py
  9. test_analysis_detect_recursion.py
  10. test_analysis_estimate_memory_usage.py
  11. test_analysis_struct_info_analysis.py
  12. test_analysis_suggest_layout_transforms.py
  13. test_analysis_well_formed.py
  14. test_ast_printer.py
  15. test_backend_dispatch_sampling.py
  16. test_backend_dispatch_sort_scan.py
  17. test_backend_transform_shape_lower.py
  18. test_base_py_module.py
  19. test_base_py_module_printer.py
  20. test_base_py_module_symbolic_shape.py
  21. test_bind_params.py
  22. test_bind_symbolic_vars.py
  23. test_binding_rewrite.py
  24. test_blockbuilder_core.py
  25. test_blockbuilder_emit_te.py
  26. test_codegen_coreml.py
  27. test_codegen_cublas.py
  28. test_codegen_cudnn.py
  29. test_codegen_cutlass.py
  30. test_codegen_dnnl.py
  31. test_codegen_hipblas.py
  32. test_codegen_tensorrt.py
  33. test_contrib_vllm.py
  34. test_dataflow_inplace.py
  35. test_dataflow_pattern.py
  36. test_dataflow_rewriter.py
  37. test_dlpack_integration.py
  38. test_e2e_op_dynamic.py
  39. test_eliminate_pad_branch_using_buffer_assumption.py
  40. test_expr.py
  41. test_expr_args_converter.py
  42. test_expr_functor.py
  43. test_fast_math_transform.py
  44. test_frontend_common.py
  45. test_frontend_dynamo.py
  46. test_frontend_from_exported_program.py
  47. test_frontend_from_fx.py
  48. test_frontend_nn_debug.py
  49. test_frontend_nn_exporter.py
  50. test_frontend_nn_extern_module.py
  51. test_frontend_nn_jit.py
  52. test_frontend_nn_modules.py
  53. test_frontend_nn_mutator.py
  54. test_frontend_nn_op.py
  55. test_frontend_nn_packing.py
  56. test_frontend_nn_subroutines.py
  57. test_frontend_nn_tensor.py
  58. test_frontend_onnx.py
  59. test_frontend_stablehlo.py
  60. test_group_gemm_flashinfer.py
  61. test_inline_functions.py
  62. test_kill_after_last_use.py
  63. test_lower_alloc_tensor.py
  64. test_meta_schedule_relax_integration.py
  65. test_op_binary.py
  66. test_op_ccl.py
  67. test_op_create.py
  68. test_op_datatype.py
  69. test_op_distributed.py
  70. test_op_grad.py
  71. test_op_gradient_numeric.py
  72. test_op_image.py
  73. test_op_index.py
  74. test_op_inspect.py
  75. test_op_linear_algebra.py
  76. test_op_manipulate.py
  77. test_op_misc.py
  78. test_op_nn.py
  79. test_op_nn_convolution.py
  80. test_op_nn_pooling.py
  81. test_op_qdq.py
  82. test_op_sampling.py
  83. test_op_search.py
  84. test_op_set.py
  85. test_op_sort.py
  86. test_op_statistical.py
  87. test_op_take.py
  88. test_op_ternary.py
  89. test_op_unary.py
  90. test_op_view.py
  91. test_op_vision.py
  92. test_optimize_layout_transform.py
  93. test_pipeline.py
  94. test_pytorch_integration.py
  95. test_relax_operators.py
  96. test_relax_to_pyfunc_converter.py
  97. test_remove_redundant_reshape.py
  98. test_runtime_builtin.py
  99. test_runtime_builtin_paged_attention_kv_cache_cpu.py
  100. test_runtime_builtin_paged_attention_kv_cache_flashinfer.py
  101. test_runtime_builtin_paged_attention_kv_cache_mla_flashinfer.py
  102. test_runtime_builtin_paged_attention_kv_cache_mla_tir.py
  103. test_runtime_builtin_paged_attention_kv_cache_tir.py
  104. test_runtime_builtin_rnn_state.py
  105. test_runtime_sampling_flashinfer.py
  106. test_struct_info.py
  107. test_testing_nn.py
  108. test_tir_call_source_kernel.py
  109. test_training_append_loss.py
  110. test_training_loss.py
  111. test_training_optimizer.py
  112. test_training_optimizer_numeric.py
  113. test_training_setup_trainer.py
  114. test_training_trainer_numeric.py
  115. test_transform.py
  116. test_transform_adjust_matmul_order.py
  117. test_transform_allocate_workspace.py
  118. test_transform_alter_op_impl.py
  119. test_transform_annotate_tir_op_pattern.py
  120. test_transform_attach_attr_layout_free_buffers.py
  121. test_transform_attach_global_symbol.py
  122. test_transform_bind_params.py
  123. test_transform_bind_symbolic_vars.py
  124. test_transform_bundle_model_params.py
  125. test_transform_canonicalize_bindings.py
  126. test_transform_codegen_pass.py
  127. test_transform_combine_parallel_matmul.py
  128. test_transform_compute_prim_value.py
  129. test_transform_convert_dataflow.py
  130. test_transform_convert_layout.py
  131. test_transform_cse.py
  132. test_transform_dead_code_elimination.py
  133. test_transform_decompose_ops.py
  134. test_transform_expand_matmul_of_sum.py
  135. test_transform_expand_tuple_args.py
  136. test_transform_few_shot_tuning.py
  137. test_transform_fold_batch_norm_to_conv2d.py
  138. test_transform_fold_constant.py
  139. test_transform_fuse_ops.py
  140. test_transform_fuse_ops_by_pattern.py
  141. test_transform_fuse_tir.py
  142. test_transform_fuse_transpose_matmul.py
  143. test_transform_gradient.py
  144. test_transform_gradient_checkpoint.py
  145. test_transform_gradient_numeric.py
  146. test_transform_gradient_te_register.py
  147. test_transform_inline_private_functions.py
  148. test_transform_ipc_allreduce_rewrite.py
  149. test_transform_lambda_lift.py
  150. test_transform_lazy_transform_params.py
  151. test_transform_legalize_ops.py
  152. test_transform_legalize_ops_binary.py
  153. test_transform_legalize_ops_ccl.py
  154. test_transform_legalize_ops_create_datatype.py
  155. test_transform_legalize_ops_distributed.py
  156. test_transform_legalize_ops_grad.py
  157. test_transform_legalize_ops_image.py
  158. test_transform_legalize_ops_index_linear_algebra.py
  159. test_transform_legalize_ops_manipulate.py
  160. test_transform_legalize_ops_nn.py
  161. test_transform_legalize_ops_qdq.py
  162. test_transform_legalize_ops_search_statistical.py
  163. test_transform_legalize_ops_unary.py
  164. test_transform_lift_transform_params.py
  165. test_transform_lower_gpu_ipc_alloc_storage.py
  166. test_transform_merge_composite_functions.py
  167. test_transform_meta_schedule_apply_database.py
  168. test_transform_meta_schedule_tuning.py
  169. test_transform_normalize.py
  170. test_transform_normalize_global_var.py
  171. test_transform_operator_specific_normalization.py
  172. test_transform_realize_vdevice.py
  173. test_transform_remove_unused_outputs.py
  174. test_transform_remove_unused_parameters.py
  175. test_transform_reorder_permute_dims_after_concat.py
  176. test_transform_reorder_take_after_matmul.py
  177. test_transform_rewrite_cuda_graph.py
  178. test_transform_rewrite_dataflow_reshape.py
  179. test_transform_split_layout_rewrite_preproc.py
  180. test_transform_static_plan_block_memory.py
  181. test_transform_to_mixed_precision.py
  182. test_transform_topological_sort.py
  183. test_transform_update_param_struct_info.py
  184. test_transform_update_vdevice.py
  185. test_tvmscript_ir_builder.py
  186. test_tvmscript_parser.py
  187. test_tvmscript_parser_op_arith_cmp.py
  188. test_tvmscript_parser_op_create.py
  189. test_tvmscript_parser_op_datatype.py
  190. test_tvmscript_parser_op_grad.py
  191. test_tvmscript_parser_op_image.py
  192. test_tvmscript_parser_op_index.py
  193. test_tvmscript_parser_op_linear_algebra.py
  194. test_tvmscript_parser_op_manipulate.py
  195. test_tvmscript_parser_op_nn.py
  196. test_tvmscript_parser_op_search.py
  197. test_tvmscript_parser_op_set.py
  198. test_tvmscript_parser_op_sort.py
  199. test_tvmscript_parser_op_statistical.py
  200. test_tvmscript_parser_op_vision.py
  201. test_tvmscript_printer_relax.py
  202. test_tvmscript_pyfunc.py
  203. test_utils.py
  204. test_vm_alloc_storage_with_scope.py
  205. test_vm_build.py
  206. test_vm_builtin.py
  207. test_vm_builtin_lower.py
  208. test_vm_callback_function.py
  209. test_vm_codegen_only.py
  210. test_vm_codegen_tir.py
  211. test_vm_cuda_graph.py
  212. test_vm_execbuilder.py
  213. test_vm_instrument.py
  214. test_vm_multi_device.py
  215. test_vm_profiler.py