├── .clang-format ├── .dockerignore ├── .editorconfig ├── .github ├── codeql │ └── codeql-config.yml └── workflows │ ├── codeql.yml │ ├── doc.yml │ ├── format.yml │ └── main.yml ├── .gitignore ├── .gitmodules ├── .style.yapf ├── .yapfignore ├── 3rd-party ├── README.md └── antlr │ └── antlr-4.11.1-complete.jar ├── CMakeLists.txt ├── Doxyfile ├── LICENSE ├── README.md ├── ci-script ├── prepare-python-environment.sh └── with-spack-mkl.toml ├── clang-minimal-dev.Dockerfile ├── clang-mkl-dev.Dockerfile ├── cuda-mkl-dev.Dockerfile ├── cuda-mkl-pytorch-dev.Dockerfile ├── docker.Makefile ├── docs ├── about │ ├── contrib.md │ └── pub.md ├── api.md ├── extra.css ├── guide │ ├── ad.md │ ├── build-and-run.md │ ├── first-program.md │ ├── gpu.md │ ├── hint.md │ ├── index.md │ └── schedules.md ├── index.md ├── javascripts │ └── mathjax.js └── resource │ ├── logo-dark.png │ └── logo-light.png ├── ffi ├── analyze.cc ├── array.cc ├── ast.cc ├── autograd.cc ├── buffer.cc ├── codegen.cc ├── config.cc ├── debug.cc ├── device.cc ├── driver.cc ├── except.cc ├── expr.cc ├── ffi.cc ├── ffi.h ├── for_property.cc ├── frontend.cc ├── func.cc ├── mangle.cc ├── metadata.cc ├── parallel_scope.cc ├── pass.cc ├── reduce_op.cc ├── schedule.cc ├── selector.cc ├── stmt.cc ├── tensor.cc └── tensor_types.cc ├── gcc-minimal-dev.Dockerfile ├── grammar ├── ast_lexer.g ├── ast_parser.g ├── pb_lexer.g ├── pb_parser.g ├── selector_lexer.g └── selector_parser.g ├── include ├── allocator.h ├── analyze │ ├── all_defs.h │ ├── all_side_effect_intrinsics.h │ ├── all_uses.h │ ├── analyze_linear.h │ ├── as_dnf.h │ ├── check_all_defined.h │ ├── check_not_modified.h │ ├── comp_access_bound.h │ ├── comp_transient_bounds.h │ ├── comp_unique_bounds.h │ ├── comp_unique_bounds_combination.h │ ├── comp_unique_bounds_pb.h │ ├── count_contig_access_loops.h │ ├── deps.h │ ├── find_indexing_loops.h │ ├── find_loop_variance.h │ ├── find_stmt.h │ ├── merge_no_deps_hint.h │ ├── normalize_conditional_expr.h │ ├── structural_feature.h │ ├── symbol_table.h │ └── track_stmt.h ├── ast.h ├── autograd │ ├── all_no_reuse_defs.h │ ├── analyze_version.h │ ├── clear_mark_version.h │ ├── dedup_tape_names.h │ ├── derivative.h │ ├── find_tape_or_recomp_stmts.h │ ├── grad.h │ ├── invert_from_store.h │ ├── invert_stmts.h │ ├── merge_tape_input.h │ ├── output_intermediates.h │ ├── propagate_defs_need_grad.h │ ├── replace_by_saved.h │ ├── tape_strategy.h │ └── user_grad.h ├── buffer.h ├── codegen │ ├── code_gen.h │ ├── code_gen_c.h │ ├── code_gen_cpu.h │ ├── code_gen_cuda.h │ └── native_code.h ├── config.h ├── container_utils.h ├── cutlass_micro_kernel_property.h ├── data_type_infer.h ├── debug.h ├── debug │ ├── check_conflict_id.h │ ├── logger.h │ ├── match_ast.h │ └── profile.h ├── disjoint_set.h ├── driver.h ├── driver │ ├── array.h │ ├── device.h │ ├── gpu.h │ ├── target.h │ └── target_type.h ├── except.h ├── expr.h ├── for_property.h ├── frontend │ ├── frontend_var.h │ └── inlined_invoke.h ├── func.h ├── func_utils.h ├── get_new_name.h ├── hash.h ├── hash_combine.h ├── id.h ├── lazy.h ├── lower.h ├── math │ ├── bounds.h │ ├── gen_pb_expr.h │ ├── linear.h │ ├── min_max.h │ ├── parse_pb_expr.h │ ├── presburger.h │ ├── rational.h │ └── utils.h ├── maybe_void.h ├── metadata.h ├── mutator.h ├── omp_utils.h ├── parallel_scope.h ├── pass │ ├── annotate_conds.h │ ├── const_fold.h │ ├── cpu │ │ └── lower_parallel_reduction.h │ ├── flatten_stmt_seq.h │ ├── float_simplify.h │ ├── gpu │ │ ├── lower_parallel_reduction.h │ │ ├── lower_vector.h │ │ ├── make_sync.h │ │ ├── multiplex_buffers.h │ │ ├── normalize_thread_dims.h │ │ ├── normalize_threads.h │ │ ├── normalize_var_in_kernel.h │ │ └── simplex_buffers.h │ ├── hoist_return_vars.h │ ├── hoist_var_over_stmt_seq.h │ ├── make_heap_alloc.h │ ├── make_nested_loops.h │ ├── make_parallel_reduction.h │ ├── make_reduction.h │ ├── merge_and_hoist_if.h │ ├── move_out_first_or_last_iter.h │ ├── normalize_loops.h │ ├── prop_one_time_use.h │ ├── refine_sign_data_type.h │ ├── remove_cyclic_assign.h │ ├── remove_dead_var.h │ ├── remove_writes.h │ ├── rename_var.h │ ├── replace_iter.h │ ├── replace_uses.h │ ├── scalar_prop_const.h │ ├── shrink_for.h │ ├── shrink_linear_indices.h │ ├── shrink_var.h │ ├── simplify.h │ ├── sink_var.h │ ├── tensor_prop_const.h │ ├── undo_make_reduction.h │ ├── use_builtin_div.h │ └── z3_simplify.h ├── probability │ ├── rand_cond.h │ ├── rand_ctx.h │ └── rand_var.h ├── random.h ├── reduce_op.h ├── ref.h ├── schedule.h ├── schedule │ ├── as_matmul.h │ ├── blend.h │ ├── cache.h │ ├── check_loop_order.h │ ├── check_not_in_lib.h │ ├── check_var_cross_parallel.h │ ├── fission.h │ ├── fuse.h │ ├── hoist_selected_var.h │ ├── inlining.h │ ├── lower_cutlass_micro_block.h │ ├── memoized_schedules.h │ ├── merge.h │ ├── parallelize.h │ ├── parallelize_as.h │ ├── permute.h │ ├── pluto.h │ ├── reorder.h │ ├── schedule_log.h │ ├── separate_tail.h │ ├── set_mem_type.h │ ├── split.h │ ├── swap.h │ ├── unroll.h │ ├── var_merge.h │ ├── var_reorder.h │ ├── var_split.h │ ├── var_squeeze.h │ ├── var_unsqueeze.h │ └── vectorize.h ├── selector.h ├── serialize │ ├── ast_parser_base.h │ ├── load_ast.h │ ├── load_driver.h │ ├── mangle.h │ ├── print_ast.h │ ├── print_driver.h │ ├── stream_utils.h │ └── to_string.h ├── shared_linked_list.h ├── stmt.h ├── sub_tree.h ├── sync_func.h ├── tensor.h ├── timeout.h ├── type │ ├── access_type.h │ ├── data_type.h │ └── mem_type.h └── visitor.h ├── mkdocs.yml ├── pyproject.toml ├── pytest.ini ├── python └── freetensor │ ├── __init__.py │ ├── core │ ├── __init__.py │ ├── analyze.py │ ├── autograd.py │ ├── codegen.py │ ├── config.py │ ├── context.py │ ├── driver.py │ ├── enable_attach_backward.py │ ├── expr.py │ ├── frontend.py │ ├── func.py │ ├── jit.py │ ├── meta.py │ ├── optimize.py │ ├── param_ret_dict.py │ ├── passes.py │ ├── return_values_pack.py │ ├── schedule.py │ ├── serialize.py │ ├── staging.py │ ├── stmt.py │ ├── transform.py │ └── utils.py │ ├── debug.py │ ├── ffi.py │ └── libop │ ├── __init__.py │ ├── assign.py │ ├── concat.py │ ├── constant.py │ ├── conv.py │ ├── conv_shape_utils.py │ ├── element_wise.py │ ├── logsumexp.py │ ├── matmul.py │ ├── pooling.py │ ├── reduction.py │ ├── reshape.py │ ├── shape_utils.py │ ├── softmax.py │ ├── transpose.py │ └── utils.py ├── requirements.txt ├── runtime ├── README.md ├── context.h ├── cpu_context.h ├── cpu_runtime.h ├── gpu_context.h ├── gpu_runtime.h ├── mdspan.h ├── micro_kernel │ └── matmul │ │ └── cutlass │ │ ├── gemm.h │ │ └── gemm_sm80.h └── unchecked_opt.h ├── src ├── allocator.cc ├── analyze │ ├── all_uses.cc │ ├── analyze_linear.cc │ ├── as_dnf.cc │ ├── check_not_modified.cc │ ├── comp_access_bound.cc │ ├── comp_unique_bounds.cc │ ├── comp_unique_bounds_combination.cc │ ├── comp_unique_bounds_pb.cc │ ├── count_contig_access_loops.cc │ ├── deps.cc │ ├── find_indexing_loops.cc │ ├── find_loop_variance.cc │ ├── find_stmt.cc │ ├── merge_no_deps_hint.cc │ ├── normalize_conditional_expr.cc │ └── structural_feature.cc ├── ast.cc ├── autograd │ ├── all_no_reuse_defs.cc │ ├── analyze_version.cc │ ├── dedup_tape_names.cc │ ├── derivative.cc │ ├── find_tape_or_recomp_stmts.cc │ ├── grad.cc │ ├── invert_from_store.cc │ ├── invert_stmts.cc │ ├── merge_tape_input.cc │ ├── output_intermediates.cc │ ├── propagate_defs_need_grad.cc │ ├── replace_by_saved.cc │ ├── tape_strategy.cc │ └── user_grad.cc ├── buffer.cc ├── codegen │ ├── code_gen.cc │ ├── code_gen_cpu.cc │ ├── code_gen_cuda.cc │ ├── detail │ │ ├── code_gen.h │ │ └── code_gen_c.h │ └── native_code.cc ├── config.cc ├── container_utils.cc ├── cutlass_micro_kernel_property.cc ├── data_type_infer.cc ├── debug │ ├── check_conflict_id.cc │ ├── logger.cc │ ├── match_ast.cc │ └── profile.cc ├── driver.cc ├── driver │ ├── array.cc │ ├── device.cc │ └── target.cc ├── except.cc ├── expr.cc ├── for_property.cc ├── frontend │ ├── frontend_var.cc │ └── inlined_invoke.cc ├── func.cc ├── get_new_name.cc ├── hash.cc ├── hash_combine.cc ├── id.cc ├── math │ ├── bounds.cc │ ├── gen_pb_expr.cc │ ├── min_max.cc │ ├── parse_pb_expr.cc │ └── presburger.cc ├── metadata.cc ├── mutator.cc ├── pass │ ├── annotate_conds.cc │ ├── const_fold.cc │ ├── cpu │ │ └── lower_parallel_reduction.cc │ ├── flatten_stmt_seq.cc │ ├── float_simplify.cc │ ├── gpu │ │ ├── lower_parallel_reduction.cc │ │ ├── lower_vector.cc │ │ ├── make_sync.cc │ │ ├── multiplex_buffers.cc │ │ ├── normalize_thread_dims.cc │ │ ├── normalize_threads.cc │ │ ├── normalize_var_in_kernel.cc │ │ └── simplex_buffers.cc │ ├── hoist_return_vars.cc │ ├── hoist_var_over_stmt_seq.cc │ ├── make_heap_alloc.cc │ ├── make_parallel_reduction.cc │ ├── make_reduction.cc │ ├── merge_and_hoist_if.cc │ ├── move_out_first_or_last_iter.cc │ ├── normalize_loops.cc │ ├── prop_one_time_use.cc │ ├── refine_sign_data_type.cc │ ├── remove_cyclic_assign.cc │ ├── remove_dead_var.cc │ ├── remove_writes.cc │ ├── rename_var.cc │ ├── replace_uses.cc │ ├── scalar_prop_const.cc │ ├── shrink_for.cc │ ├── shrink_linear_indices.cc │ ├── shrink_var.cc │ ├── simplify.cc │ ├── sink_var.cc │ ├── tensor_prop_const.cc │ ├── undo_make_reduction.cc │ ├── use_builtin_div.cc │ └── z3_simplify.cc ├── probability │ ├── rand_ctx.cc │ └── rand_var.cc ├── random.cc ├── reduce_op.cc ├── schedule.cc ├── schedule │ ├── as_matmul.cc │ ├── auto_fission_fuse.cc │ ├── auto_inline.cc │ ├── auto_mem_layout.cc │ ├── auto_parallelize.cc │ ├── auto_pluto.cc │ ├── auto_reorder.cc │ ├── auto_set_mem_type.cc │ ├── auto_swap.cc │ ├── auto_unroll.cc │ ├── auto_use_lib.cc │ ├── blend.cc │ ├── cache.cc │ ├── check_loop_order.cc │ ├── check_var_cross_parallel.cc │ ├── fission.cc │ ├── fuse.cc │ ├── hoist_selected_var.cc │ ├── inlining.cc │ ├── lower_cutlass_micro_block.cc │ ├── merge.cc │ ├── move_to.cc │ ├── parallelize.cc │ ├── parallelize_as.cc │ ├── permute.cc │ ├── pluto.cc │ ├── reorder.cc │ ├── separate_tail.cc │ ├── set_mem_type.cc │ ├── split.cc │ ├── swap.cc │ ├── unroll.cc │ ├── var_merge.cc │ ├── var_reorder.cc │ ├── var_split.cc │ ├── var_squeeze.cc │ ├── var_unsqueeze.cc │ └── vectorize.cc ├── selector.cc ├── serialize │ ├── load_ast.cc │ ├── load_driver.cc │ ├── mangle.cc │ ├── print_ast.cc │ └── print_driver.cc ├── stmt.cc ├── sub_tree.cc ├── tensor.cc ├── timeout.cc ├── type │ └── data_type.cc └── visitor.cc ├── test ├── 00.hello_world │ ├── test_basic.py │ ├── test_doc_examples.py │ ├── test_io.py │ ├── test_serialize_ast.py │ └── test_serialize_driver.py ├── 10.analyze │ ├── test_find_stmt.py │ └── test_structural_feature.py ├── 20.pass │ ├── test_float_simplify.py │ ├── test_hoist_var_over_stmt_seq.py │ ├── test_lower.py │ ├── test_make_heap_alloc.py │ ├── test_make_reduction.py │ ├── test_merge_and_hoist_if.py │ ├── test_move_out_first_or_later_iter.py │ ├── test_prop_one_time_use.py │ ├── test_refine_sign_data_type.py │ ├── test_remove_cyclic_assign.py │ ├── test_remove_dead_var.py │ ├── test_remove_writes.py │ ├── test_scalar_prop_const.py │ ├── test_shrink_for.py │ ├── test_shrink_var.py │ ├── test_simplify.py │ ├── test_sink_var.py │ ├── test_tensor_prop_const.py │ └── test_use_builtin_div.py ├── 21.autograd │ ├── test_grad.py │ ├── test_invert.py │ ├── test_jacrev.py │ ├── test_output_intermediates.py │ ├── test_pytorch_integration.py │ ├── test_tape.py │ └── test_user_grad.py ├── 30.schedule │ ├── test_as_matmul.py │ ├── test_blend.py │ ├── test_cache.py │ ├── test_cache_reduction.py │ ├── test_fission.py │ ├── test_fuse.py │ ├── test_inline.py │ ├── test_memoized_schedules.py │ ├── test_merge.py │ ├── test_move_to.py │ ├── test_parallelize.py │ ├── test_parallelize_as.py │ ├── test_permute.py │ ├── test_pluto.py │ ├── test_reorder.py │ ├── test_separate_tail.py │ ├── test_set_mem_type.py │ ├── test_split.py │ ├── test_swap.py │ ├── test_unroll.py │ ├── test_var_merge.py │ ├── test_var_reorder.py │ ├── test_var_split.py │ ├── test_var_squeeze.py │ ├── test_var_unsqueeze.py │ └── test_vectorize.py ├── 31.auto_schedule │ ├── test_auto_fission_fuse.py │ ├── test_auto_inline.py │ ├── test_auto_mem_layout.py │ ├── test_auto_parallelize.py │ ├── test_auto_pluto.py │ ├── test_auto_reorder.py │ ├── test_auto_set_mem_type.py │ ├── test_auto_swap.py │ └── test_auto_use_lib.py ├── 40.codegen │ ├── cpu │ │ ├── test_cpu.py │ │ ├── test_cpu_reduce.py │ │ └── test_mkl.py │ └── gpu │ │ ├── test_gpu.py │ │ ├── test_gpu_cublas.py │ │ ├── test_gpu_cutlass.py │ │ ├── test_gpu_normalize_buffer_shapes.py │ │ ├── test_gpu_reduce.py │ │ ├── test_gpu_sync.py │ │ ├── test_gpu_vector.py │ │ └── test_gpu_warpshuffle.py ├── 50.frontend │ ├── test_ast_inline.py │ ├── test_error_report.py │ ├── test_jit.py │ ├── test_ops.py │ ├── test_staging_inline.py │ ├── test_transformer_basic.py │ └── test_var.py ├── 60.libop │ ├── test_assign.py │ ├── test_augment_assign.py │ ├── test_binary.py │ ├── test_concat.py │ ├── test_conv.py │ ├── test_einsum.py │ ├── test_expand.py │ ├── test_flatten_onnx.py │ ├── test_flatten_pytorch.py │ ├── test_gemm.py │ ├── test_logsumexp.py │ ├── test_matmul.py │ ├── test_ones.py │ ├── test_pooling.py │ ├── test_reduce.py │ ├── test_reshape.py │ ├── test_softmax.py │ ├── test_squeeze.py │ ├── test_stack.py │ ├── test_transpose.py │ ├── test_unary.py │ ├── test_unsqueeze.py │ └── test_zeros.py └── 70.program │ ├── test_common_transforms.py │ ├── test_gpu_conv2d.py │ ├── test_gpu_softmax.py │ └── test_program_with_micro_kernel.py ├── with-cuda.toml ├── with-mkl.toml └── with-pytorch.toml /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.clang-format -------------------------------------------------------------------------------- /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.dockerignore -------------------------------------------------------------------------------- /.editorconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.editorconfig -------------------------------------------------------------------------------- /.github/codeql/codeql-config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.github/codeql/codeql-config.yml -------------------------------------------------------------------------------- /.github/workflows/codeql.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.github/workflows/codeql.yml -------------------------------------------------------------------------------- /.github/workflows/doc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.github/workflows/doc.yml -------------------------------------------------------------------------------- /.github/workflows/format.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.github/workflows/format.yml -------------------------------------------------------------------------------- /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.gitmodules -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/.style.yapf -------------------------------------------------------------------------------- /.yapfignore: -------------------------------------------------------------------------------- 1 | **/3rd-party/** 2 | **/experiments/** 3 | -------------------------------------------------------------------------------- /3rd-party/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/3rd-party/README.md -------------------------------------------------------------------------------- /3rd-party/antlr/antlr-4.11.1-complete.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/3rd-party/antlr/antlr-4.11.1-complete.jar -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /Doxyfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/Doxyfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/README.md -------------------------------------------------------------------------------- /ci-script/prepare-python-environment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ci-script/prepare-python-environment.sh -------------------------------------------------------------------------------- /ci-script/with-spack-mkl.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ci-script/with-spack-mkl.toml -------------------------------------------------------------------------------- /clang-minimal-dev.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/clang-minimal-dev.Dockerfile -------------------------------------------------------------------------------- /clang-mkl-dev.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/clang-mkl-dev.Dockerfile -------------------------------------------------------------------------------- /cuda-mkl-dev.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/cuda-mkl-dev.Dockerfile -------------------------------------------------------------------------------- /cuda-mkl-pytorch-dev.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/cuda-mkl-pytorch-dev.Dockerfile -------------------------------------------------------------------------------- /docker.Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docker.Makefile -------------------------------------------------------------------------------- /docs/about/contrib.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/about/contrib.md -------------------------------------------------------------------------------- /docs/about/pub.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/about/pub.md -------------------------------------------------------------------------------- /docs/api.md: -------------------------------------------------------------------------------- 1 | # Python API 2 | 3 | ::: freetensor 4 | -------------------------------------------------------------------------------- /docs/extra.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/extra.css -------------------------------------------------------------------------------- /docs/guide/ad.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/guide/ad.md -------------------------------------------------------------------------------- /docs/guide/build-and-run.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/guide/build-and-run.md -------------------------------------------------------------------------------- /docs/guide/first-program.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/guide/first-program.md -------------------------------------------------------------------------------- /docs/guide/gpu.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/guide/gpu.md -------------------------------------------------------------------------------- /docs/guide/hint.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/guide/hint.md -------------------------------------------------------------------------------- /docs/guide/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/guide/index.md -------------------------------------------------------------------------------- /docs/guide/schedules.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/guide/schedules.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/javascripts/mathjax.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/javascripts/mathjax.js -------------------------------------------------------------------------------- /docs/resource/logo-dark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/resource/logo-dark.png -------------------------------------------------------------------------------- /docs/resource/logo-light.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/docs/resource/logo-light.png -------------------------------------------------------------------------------- /ffi/analyze.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/analyze.cc -------------------------------------------------------------------------------- /ffi/array.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/array.cc -------------------------------------------------------------------------------- /ffi/ast.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/ast.cc -------------------------------------------------------------------------------- /ffi/autograd.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/autograd.cc -------------------------------------------------------------------------------- /ffi/buffer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/buffer.cc -------------------------------------------------------------------------------- /ffi/codegen.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/codegen.cc -------------------------------------------------------------------------------- /ffi/config.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/config.cc -------------------------------------------------------------------------------- /ffi/debug.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/debug.cc -------------------------------------------------------------------------------- /ffi/device.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/device.cc -------------------------------------------------------------------------------- /ffi/driver.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/driver.cc -------------------------------------------------------------------------------- /ffi/except.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/except.cc -------------------------------------------------------------------------------- /ffi/expr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/expr.cc -------------------------------------------------------------------------------- /ffi/ffi.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/ffi.cc -------------------------------------------------------------------------------- /ffi/ffi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/ffi.h -------------------------------------------------------------------------------- /ffi/for_property.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/for_property.cc -------------------------------------------------------------------------------- /ffi/frontend.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/frontend.cc -------------------------------------------------------------------------------- /ffi/func.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/func.cc -------------------------------------------------------------------------------- /ffi/mangle.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/mangle.cc -------------------------------------------------------------------------------- /ffi/metadata.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/metadata.cc -------------------------------------------------------------------------------- /ffi/parallel_scope.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/parallel_scope.cc -------------------------------------------------------------------------------- /ffi/pass.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/pass.cc -------------------------------------------------------------------------------- /ffi/reduce_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/reduce_op.cc -------------------------------------------------------------------------------- /ffi/schedule.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/schedule.cc -------------------------------------------------------------------------------- /ffi/selector.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/selector.cc -------------------------------------------------------------------------------- /ffi/stmt.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/stmt.cc -------------------------------------------------------------------------------- /ffi/tensor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/tensor.cc -------------------------------------------------------------------------------- /ffi/tensor_types.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/ffi/tensor_types.cc -------------------------------------------------------------------------------- /gcc-minimal-dev.Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/gcc-minimal-dev.Dockerfile -------------------------------------------------------------------------------- /grammar/ast_lexer.g: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/grammar/ast_lexer.g -------------------------------------------------------------------------------- /grammar/ast_parser.g: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/grammar/ast_parser.g -------------------------------------------------------------------------------- /grammar/pb_lexer.g: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/grammar/pb_lexer.g -------------------------------------------------------------------------------- /grammar/pb_parser.g: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/grammar/pb_parser.g -------------------------------------------------------------------------------- /grammar/selector_lexer.g: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/grammar/selector_lexer.g -------------------------------------------------------------------------------- /grammar/selector_parser.g: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/grammar/selector_parser.g -------------------------------------------------------------------------------- /include/allocator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/allocator.h -------------------------------------------------------------------------------- /include/analyze/all_defs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/all_defs.h -------------------------------------------------------------------------------- /include/analyze/all_side_effect_intrinsics.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/all_side_effect_intrinsics.h -------------------------------------------------------------------------------- /include/analyze/all_uses.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/all_uses.h -------------------------------------------------------------------------------- /include/analyze/analyze_linear.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/analyze_linear.h -------------------------------------------------------------------------------- /include/analyze/as_dnf.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/as_dnf.h -------------------------------------------------------------------------------- /include/analyze/check_all_defined.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/check_all_defined.h -------------------------------------------------------------------------------- /include/analyze/check_not_modified.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/check_not_modified.h -------------------------------------------------------------------------------- /include/analyze/comp_access_bound.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/comp_access_bound.h -------------------------------------------------------------------------------- /include/analyze/comp_transient_bounds.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/comp_transient_bounds.h -------------------------------------------------------------------------------- /include/analyze/comp_unique_bounds.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/comp_unique_bounds.h -------------------------------------------------------------------------------- /include/analyze/comp_unique_bounds_combination.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/comp_unique_bounds_combination.h -------------------------------------------------------------------------------- /include/analyze/comp_unique_bounds_pb.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/comp_unique_bounds_pb.h -------------------------------------------------------------------------------- /include/analyze/count_contig_access_loops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/count_contig_access_loops.h -------------------------------------------------------------------------------- /include/analyze/deps.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/deps.h -------------------------------------------------------------------------------- /include/analyze/find_indexing_loops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/find_indexing_loops.h -------------------------------------------------------------------------------- /include/analyze/find_loop_variance.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/find_loop_variance.h -------------------------------------------------------------------------------- /include/analyze/find_stmt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/find_stmt.h -------------------------------------------------------------------------------- /include/analyze/merge_no_deps_hint.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/merge_no_deps_hint.h -------------------------------------------------------------------------------- /include/analyze/normalize_conditional_expr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/normalize_conditional_expr.h -------------------------------------------------------------------------------- /include/analyze/structural_feature.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/structural_feature.h -------------------------------------------------------------------------------- /include/analyze/symbol_table.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/symbol_table.h -------------------------------------------------------------------------------- /include/analyze/track_stmt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/analyze/track_stmt.h -------------------------------------------------------------------------------- /include/ast.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/ast.h -------------------------------------------------------------------------------- /include/autograd/all_no_reuse_defs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/all_no_reuse_defs.h -------------------------------------------------------------------------------- /include/autograd/analyze_version.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/analyze_version.h -------------------------------------------------------------------------------- /include/autograd/clear_mark_version.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/clear_mark_version.h -------------------------------------------------------------------------------- /include/autograd/dedup_tape_names.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/dedup_tape_names.h -------------------------------------------------------------------------------- /include/autograd/derivative.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/derivative.h -------------------------------------------------------------------------------- /include/autograd/find_tape_or_recomp_stmts.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/find_tape_or_recomp_stmts.h -------------------------------------------------------------------------------- /include/autograd/grad.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/grad.h -------------------------------------------------------------------------------- /include/autograd/invert_from_store.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/invert_from_store.h -------------------------------------------------------------------------------- /include/autograd/invert_stmts.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/invert_stmts.h -------------------------------------------------------------------------------- /include/autograd/merge_tape_input.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/merge_tape_input.h -------------------------------------------------------------------------------- /include/autograd/output_intermediates.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/output_intermediates.h -------------------------------------------------------------------------------- /include/autograd/propagate_defs_need_grad.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/propagate_defs_need_grad.h -------------------------------------------------------------------------------- /include/autograd/replace_by_saved.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/replace_by_saved.h -------------------------------------------------------------------------------- /include/autograd/tape_strategy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/tape_strategy.h -------------------------------------------------------------------------------- /include/autograd/user_grad.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/autograd/user_grad.h -------------------------------------------------------------------------------- /include/buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/buffer.h -------------------------------------------------------------------------------- /include/codegen/code_gen.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/codegen/code_gen.h -------------------------------------------------------------------------------- /include/codegen/code_gen_c.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/codegen/code_gen_c.h -------------------------------------------------------------------------------- /include/codegen/code_gen_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/codegen/code_gen_cpu.h -------------------------------------------------------------------------------- /include/codegen/code_gen_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/codegen/code_gen_cuda.h -------------------------------------------------------------------------------- /include/codegen/native_code.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/codegen/native_code.h -------------------------------------------------------------------------------- /include/config.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/config.h -------------------------------------------------------------------------------- /include/container_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/container_utils.h -------------------------------------------------------------------------------- /include/cutlass_micro_kernel_property.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/cutlass_micro_kernel_property.h -------------------------------------------------------------------------------- /include/data_type_infer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/data_type_infer.h -------------------------------------------------------------------------------- /include/debug.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/debug.h -------------------------------------------------------------------------------- /include/debug/check_conflict_id.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/debug/check_conflict_id.h -------------------------------------------------------------------------------- /include/debug/logger.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/debug/logger.h -------------------------------------------------------------------------------- /include/debug/match_ast.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/debug/match_ast.h -------------------------------------------------------------------------------- /include/debug/profile.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/debug/profile.h -------------------------------------------------------------------------------- /include/disjoint_set.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/disjoint_set.h -------------------------------------------------------------------------------- /include/driver.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/driver.h -------------------------------------------------------------------------------- /include/driver/array.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/driver/array.h -------------------------------------------------------------------------------- /include/driver/device.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/driver/device.h -------------------------------------------------------------------------------- /include/driver/gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/driver/gpu.h -------------------------------------------------------------------------------- /include/driver/target.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/driver/target.h -------------------------------------------------------------------------------- /include/driver/target_type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/driver/target_type.h -------------------------------------------------------------------------------- /include/except.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/except.h -------------------------------------------------------------------------------- /include/expr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/expr.h -------------------------------------------------------------------------------- /include/for_property.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/for_property.h -------------------------------------------------------------------------------- /include/frontend/frontend_var.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/frontend/frontend_var.h -------------------------------------------------------------------------------- /include/frontend/inlined_invoke.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/frontend/inlined_invoke.h -------------------------------------------------------------------------------- /include/func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/func.h -------------------------------------------------------------------------------- /include/func_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/func_utils.h -------------------------------------------------------------------------------- /include/get_new_name.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/get_new_name.h -------------------------------------------------------------------------------- /include/hash.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/hash.h -------------------------------------------------------------------------------- /include/hash_combine.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/hash_combine.h -------------------------------------------------------------------------------- /include/id.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/id.h -------------------------------------------------------------------------------- /include/lazy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/lazy.h -------------------------------------------------------------------------------- /include/lower.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/lower.h -------------------------------------------------------------------------------- /include/math/bounds.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/math/bounds.h -------------------------------------------------------------------------------- /include/math/gen_pb_expr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/math/gen_pb_expr.h -------------------------------------------------------------------------------- /include/math/linear.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/math/linear.h -------------------------------------------------------------------------------- /include/math/min_max.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/math/min_max.h -------------------------------------------------------------------------------- /include/math/parse_pb_expr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/math/parse_pb_expr.h -------------------------------------------------------------------------------- /include/math/presburger.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/math/presburger.h -------------------------------------------------------------------------------- /include/math/rational.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/math/rational.h -------------------------------------------------------------------------------- /include/math/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/math/utils.h -------------------------------------------------------------------------------- /include/maybe_void.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/maybe_void.h -------------------------------------------------------------------------------- /include/metadata.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/metadata.h -------------------------------------------------------------------------------- /include/mutator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/mutator.h -------------------------------------------------------------------------------- /include/omp_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/omp_utils.h -------------------------------------------------------------------------------- /include/parallel_scope.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/parallel_scope.h -------------------------------------------------------------------------------- /include/pass/annotate_conds.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/annotate_conds.h -------------------------------------------------------------------------------- /include/pass/const_fold.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/const_fold.h -------------------------------------------------------------------------------- /include/pass/cpu/lower_parallel_reduction.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/cpu/lower_parallel_reduction.h -------------------------------------------------------------------------------- /include/pass/flatten_stmt_seq.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/flatten_stmt_seq.h -------------------------------------------------------------------------------- /include/pass/float_simplify.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/float_simplify.h -------------------------------------------------------------------------------- /include/pass/gpu/lower_parallel_reduction.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/gpu/lower_parallel_reduction.h -------------------------------------------------------------------------------- /include/pass/gpu/lower_vector.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/gpu/lower_vector.h -------------------------------------------------------------------------------- /include/pass/gpu/make_sync.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/gpu/make_sync.h -------------------------------------------------------------------------------- /include/pass/gpu/multiplex_buffers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/gpu/multiplex_buffers.h -------------------------------------------------------------------------------- /include/pass/gpu/normalize_thread_dims.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/gpu/normalize_thread_dims.h -------------------------------------------------------------------------------- /include/pass/gpu/normalize_threads.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/gpu/normalize_threads.h -------------------------------------------------------------------------------- /include/pass/gpu/normalize_var_in_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/gpu/normalize_var_in_kernel.h -------------------------------------------------------------------------------- /include/pass/gpu/simplex_buffers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/gpu/simplex_buffers.h -------------------------------------------------------------------------------- /include/pass/hoist_return_vars.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/hoist_return_vars.h -------------------------------------------------------------------------------- /include/pass/hoist_var_over_stmt_seq.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/hoist_var_over_stmt_seq.h -------------------------------------------------------------------------------- /include/pass/make_heap_alloc.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/make_heap_alloc.h -------------------------------------------------------------------------------- /include/pass/make_nested_loops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/make_nested_loops.h -------------------------------------------------------------------------------- /include/pass/make_parallel_reduction.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/make_parallel_reduction.h -------------------------------------------------------------------------------- /include/pass/make_reduction.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/make_reduction.h -------------------------------------------------------------------------------- /include/pass/merge_and_hoist_if.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/merge_and_hoist_if.h -------------------------------------------------------------------------------- /include/pass/move_out_first_or_last_iter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/move_out_first_or_last_iter.h -------------------------------------------------------------------------------- /include/pass/normalize_loops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/normalize_loops.h -------------------------------------------------------------------------------- /include/pass/prop_one_time_use.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/prop_one_time_use.h -------------------------------------------------------------------------------- /include/pass/refine_sign_data_type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/refine_sign_data_type.h -------------------------------------------------------------------------------- /include/pass/remove_cyclic_assign.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/remove_cyclic_assign.h -------------------------------------------------------------------------------- /include/pass/remove_dead_var.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/remove_dead_var.h -------------------------------------------------------------------------------- /include/pass/remove_writes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/remove_writes.h -------------------------------------------------------------------------------- /include/pass/rename_var.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/rename_var.h -------------------------------------------------------------------------------- /include/pass/replace_iter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/replace_iter.h -------------------------------------------------------------------------------- /include/pass/replace_uses.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/replace_uses.h -------------------------------------------------------------------------------- /include/pass/scalar_prop_const.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/scalar_prop_const.h -------------------------------------------------------------------------------- /include/pass/shrink_for.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/shrink_for.h -------------------------------------------------------------------------------- /include/pass/shrink_linear_indices.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/shrink_linear_indices.h -------------------------------------------------------------------------------- /include/pass/shrink_var.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/shrink_var.h -------------------------------------------------------------------------------- /include/pass/simplify.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/simplify.h -------------------------------------------------------------------------------- /include/pass/sink_var.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/sink_var.h -------------------------------------------------------------------------------- /include/pass/tensor_prop_const.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/tensor_prop_const.h -------------------------------------------------------------------------------- /include/pass/undo_make_reduction.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/undo_make_reduction.h -------------------------------------------------------------------------------- /include/pass/use_builtin_div.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/use_builtin_div.h -------------------------------------------------------------------------------- /include/pass/z3_simplify.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/pass/z3_simplify.h -------------------------------------------------------------------------------- /include/probability/rand_cond.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/probability/rand_cond.h -------------------------------------------------------------------------------- /include/probability/rand_ctx.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/probability/rand_ctx.h -------------------------------------------------------------------------------- /include/probability/rand_var.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/probability/rand_var.h -------------------------------------------------------------------------------- /include/random.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/random.h -------------------------------------------------------------------------------- /include/reduce_op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/reduce_op.h -------------------------------------------------------------------------------- /include/ref.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/ref.h -------------------------------------------------------------------------------- /include/schedule.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule.h -------------------------------------------------------------------------------- /include/schedule/as_matmul.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/as_matmul.h -------------------------------------------------------------------------------- /include/schedule/blend.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/blend.h -------------------------------------------------------------------------------- /include/schedule/cache.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/cache.h -------------------------------------------------------------------------------- /include/schedule/check_loop_order.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/check_loop_order.h -------------------------------------------------------------------------------- /include/schedule/check_not_in_lib.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/check_not_in_lib.h -------------------------------------------------------------------------------- /include/schedule/check_var_cross_parallel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/check_var_cross_parallel.h -------------------------------------------------------------------------------- /include/schedule/fission.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/fission.h -------------------------------------------------------------------------------- /include/schedule/fuse.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/fuse.h -------------------------------------------------------------------------------- /include/schedule/hoist_selected_var.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/hoist_selected_var.h -------------------------------------------------------------------------------- /include/schedule/inlining.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/inlining.h -------------------------------------------------------------------------------- /include/schedule/lower_cutlass_micro_block.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/lower_cutlass_micro_block.h -------------------------------------------------------------------------------- /include/schedule/memoized_schedules.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/memoized_schedules.h -------------------------------------------------------------------------------- /include/schedule/merge.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/merge.h -------------------------------------------------------------------------------- /include/schedule/parallelize.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/parallelize.h -------------------------------------------------------------------------------- /include/schedule/parallelize_as.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/parallelize_as.h -------------------------------------------------------------------------------- /include/schedule/permute.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/permute.h -------------------------------------------------------------------------------- /include/schedule/pluto.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/pluto.h -------------------------------------------------------------------------------- /include/schedule/reorder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/reorder.h -------------------------------------------------------------------------------- /include/schedule/schedule_log.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/schedule_log.h -------------------------------------------------------------------------------- /include/schedule/separate_tail.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/separate_tail.h -------------------------------------------------------------------------------- /include/schedule/set_mem_type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/set_mem_type.h -------------------------------------------------------------------------------- /include/schedule/split.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/split.h -------------------------------------------------------------------------------- /include/schedule/swap.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/swap.h -------------------------------------------------------------------------------- /include/schedule/unroll.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/unroll.h -------------------------------------------------------------------------------- /include/schedule/var_merge.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/var_merge.h -------------------------------------------------------------------------------- /include/schedule/var_reorder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/var_reorder.h -------------------------------------------------------------------------------- /include/schedule/var_split.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/var_split.h -------------------------------------------------------------------------------- /include/schedule/var_squeeze.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/var_squeeze.h -------------------------------------------------------------------------------- /include/schedule/var_unsqueeze.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/var_unsqueeze.h -------------------------------------------------------------------------------- /include/schedule/vectorize.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/schedule/vectorize.h -------------------------------------------------------------------------------- /include/selector.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/selector.h -------------------------------------------------------------------------------- /include/serialize/ast_parser_base.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/serialize/ast_parser_base.h -------------------------------------------------------------------------------- /include/serialize/load_ast.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/serialize/load_ast.h -------------------------------------------------------------------------------- /include/serialize/load_driver.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/serialize/load_driver.h -------------------------------------------------------------------------------- /include/serialize/mangle.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/serialize/mangle.h -------------------------------------------------------------------------------- /include/serialize/print_ast.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/serialize/print_ast.h -------------------------------------------------------------------------------- /include/serialize/print_driver.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/serialize/print_driver.h -------------------------------------------------------------------------------- /include/serialize/stream_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/serialize/stream_utils.h -------------------------------------------------------------------------------- /include/serialize/to_string.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/serialize/to_string.h -------------------------------------------------------------------------------- /include/shared_linked_list.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/shared_linked_list.h -------------------------------------------------------------------------------- /include/stmt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/stmt.h -------------------------------------------------------------------------------- /include/sub_tree.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/sub_tree.h -------------------------------------------------------------------------------- /include/sync_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/sync_func.h -------------------------------------------------------------------------------- /include/tensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/tensor.h -------------------------------------------------------------------------------- /include/timeout.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/timeout.h -------------------------------------------------------------------------------- /include/type/access_type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/type/access_type.h -------------------------------------------------------------------------------- /include/type/data_type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/type/data_type.h -------------------------------------------------------------------------------- /include/type/mem_type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/type/mem_type.h -------------------------------------------------------------------------------- /include/visitor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/include/visitor.h -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/pytest.ini -------------------------------------------------------------------------------- /python/freetensor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/__init__.py -------------------------------------------------------------------------------- /python/freetensor/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/__init__.py -------------------------------------------------------------------------------- /python/freetensor/core/analyze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/analyze.py -------------------------------------------------------------------------------- /python/freetensor/core/autograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/autograd.py -------------------------------------------------------------------------------- /python/freetensor/core/codegen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/codegen.py -------------------------------------------------------------------------------- /python/freetensor/core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/config.py -------------------------------------------------------------------------------- /python/freetensor/core/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/context.py -------------------------------------------------------------------------------- /python/freetensor/core/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/driver.py -------------------------------------------------------------------------------- /python/freetensor/core/enable_attach_backward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/enable_attach_backward.py -------------------------------------------------------------------------------- /python/freetensor/core/expr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/expr.py -------------------------------------------------------------------------------- /python/freetensor/core/frontend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/frontend.py -------------------------------------------------------------------------------- /python/freetensor/core/func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/func.py -------------------------------------------------------------------------------- /python/freetensor/core/jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/jit.py -------------------------------------------------------------------------------- /python/freetensor/core/meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/meta.py -------------------------------------------------------------------------------- /python/freetensor/core/optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/optimize.py -------------------------------------------------------------------------------- /python/freetensor/core/param_ret_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/param_ret_dict.py -------------------------------------------------------------------------------- /python/freetensor/core/passes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/passes.py -------------------------------------------------------------------------------- /python/freetensor/core/return_values_pack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/return_values_pack.py -------------------------------------------------------------------------------- /python/freetensor/core/schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/schedule.py -------------------------------------------------------------------------------- /python/freetensor/core/serialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/serialize.py -------------------------------------------------------------------------------- /python/freetensor/core/staging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/staging.py -------------------------------------------------------------------------------- /python/freetensor/core/stmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/stmt.py -------------------------------------------------------------------------------- /python/freetensor/core/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/transform.py -------------------------------------------------------------------------------- /python/freetensor/core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/core/utils.py -------------------------------------------------------------------------------- /python/freetensor/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/debug.py -------------------------------------------------------------------------------- /python/freetensor/ffi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/ffi.py -------------------------------------------------------------------------------- /python/freetensor/libop/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/__init__.py -------------------------------------------------------------------------------- /python/freetensor/libop/assign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/assign.py -------------------------------------------------------------------------------- /python/freetensor/libop/concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/concat.py -------------------------------------------------------------------------------- /python/freetensor/libop/constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/constant.py -------------------------------------------------------------------------------- /python/freetensor/libop/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/conv.py -------------------------------------------------------------------------------- /python/freetensor/libop/conv_shape_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/conv_shape_utils.py -------------------------------------------------------------------------------- /python/freetensor/libop/element_wise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/element_wise.py -------------------------------------------------------------------------------- /python/freetensor/libop/logsumexp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/logsumexp.py -------------------------------------------------------------------------------- /python/freetensor/libop/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/matmul.py -------------------------------------------------------------------------------- /python/freetensor/libop/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/pooling.py -------------------------------------------------------------------------------- /python/freetensor/libop/reduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/reduction.py -------------------------------------------------------------------------------- /python/freetensor/libop/reshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/reshape.py -------------------------------------------------------------------------------- /python/freetensor/libop/shape_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/shape_utils.py -------------------------------------------------------------------------------- /python/freetensor/libop/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/softmax.py -------------------------------------------------------------------------------- /python/freetensor/libop/transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/transpose.py -------------------------------------------------------------------------------- /python/freetensor/libop/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/python/freetensor/libop/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/requirements.txt -------------------------------------------------------------------------------- /runtime/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/runtime/README.md -------------------------------------------------------------------------------- /runtime/context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/runtime/context.h -------------------------------------------------------------------------------- /runtime/cpu_context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/runtime/cpu_context.h -------------------------------------------------------------------------------- /runtime/cpu_runtime.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/runtime/cpu_runtime.h -------------------------------------------------------------------------------- /runtime/gpu_context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/runtime/gpu_context.h -------------------------------------------------------------------------------- /runtime/gpu_runtime.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/runtime/gpu_runtime.h -------------------------------------------------------------------------------- /runtime/mdspan.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/runtime/mdspan.h -------------------------------------------------------------------------------- /runtime/micro_kernel/matmul/cutlass/gemm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/runtime/micro_kernel/matmul/cutlass/gemm.h -------------------------------------------------------------------------------- /runtime/micro_kernel/matmul/cutlass/gemm_sm80.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/runtime/micro_kernel/matmul/cutlass/gemm_sm80.h -------------------------------------------------------------------------------- /runtime/unchecked_opt.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/runtime/unchecked_opt.h -------------------------------------------------------------------------------- /src/allocator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/allocator.cc -------------------------------------------------------------------------------- /src/analyze/all_uses.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/all_uses.cc -------------------------------------------------------------------------------- /src/analyze/analyze_linear.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/analyze_linear.cc -------------------------------------------------------------------------------- /src/analyze/as_dnf.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/as_dnf.cc -------------------------------------------------------------------------------- /src/analyze/check_not_modified.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/check_not_modified.cc -------------------------------------------------------------------------------- /src/analyze/comp_access_bound.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/comp_access_bound.cc -------------------------------------------------------------------------------- /src/analyze/comp_unique_bounds.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/comp_unique_bounds.cc -------------------------------------------------------------------------------- /src/analyze/comp_unique_bounds_combination.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/comp_unique_bounds_combination.cc -------------------------------------------------------------------------------- /src/analyze/comp_unique_bounds_pb.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/comp_unique_bounds_pb.cc -------------------------------------------------------------------------------- /src/analyze/count_contig_access_loops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/count_contig_access_loops.cc -------------------------------------------------------------------------------- /src/analyze/deps.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/deps.cc -------------------------------------------------------------------------------- /src/analyze/find_indexing_loops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/find_indexing_loops.cc -------------------------------------------------------------------------------- /src/analyze/find_loop_variance.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/find_loop_variance.cc -------------------------------------------------------------------------------- /src/analyze/find_stmt.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/find_stmt.cc -------------------------------------------------------------------------------- /src/analyze/merge_no_deps_hint.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/merge_no_deps_hint.cc -------------------------------------------------------------------------------- /src/analyze/normalize_conditional_expr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/normalize_conditional_expr.cc -------------------------------------------------------------------------------- /src/analyze/structural_feature.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/analyze/structural_feature.cc -------------------------------------------------------------------------------- /src/ast.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/ast.cc -------------------------------------------------------------------------------- /src/autograd/all_no_reuse_defs.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/all_no_reuse_defs.cc -------------------------------------------------------------------------------- /src/autograd/analyze_version.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/analyze_version.cc -------------------------------------------------------------------------------- /src/autograd/dedup_tape_names.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/dedup_tape_names.cc -------------------------------------------------------------------------------- /src/autograd/derivative.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/derivative.cc -------------------------------------------------------------------------------- /src/autograd/find_tape_or_recomp_stmts.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/find_tape_or_recomp_stmts.cc -------------------------------------------------------------------------------- /src/autograd/grad.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/grad.cc -------------------------------------------------------------------------------- /src/autograd/invert_from_store.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/invert_from_store.cc -------------------------------------------------------------------------------- /src/autograd/invert_stmts.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/invert_stmts.cc -------------------------------------------------------------------------------- /src/autograd/merge_tape_input.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/merge_tape_input.cc -------------------------------------------------------------------------------- /src/autograd/output_intermediates.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/output_intermediates.cc -------------------------------------------------------------------------------- /src/autograd/propagate_defs_need_grad.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/propagate_defs_need_grad.cc -------------------------------------------------------------------------------- /src/autograd/replace_by_saved.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/replace_by_saved.cc -------------------------------------------------------------------------------- /src/autograd/tape_strategy.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/tape_strategy.cc -------------------------------------------------------------------------------- /src/autograd/user_grad.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/autograd/user_grad.cc -------------------------------------------------------------------------------- /src/buffer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/buffer.cc -------------------------------------------------------------------------------- /src/codegen/code_gen.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/codegen/code_gen.cc -------------------------------------------------------------------------------- /src/codegen/code_gen_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/codegen/code_gen_cpu.cc -------------------------------------------------------------------------------- /src/codegen/code_gen_cuda.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/codegen/code_gen_cuda.cc -------------------------------------------------------------------------------- /src/codegen/detail/code_gen.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/codegen/detail/code_gen.h -------------------------------------------------------------------------------- /src/codegen/detail/code_gen_c.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/codegen/detail/code_gen_c.h -------------------------------------------------------------------------------- /src/codegen/native_code.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/codegen/native_code.cc -------------------------------------------------------------------------------- /src/config.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/config.cc -------------------------------------------------------------------------------- /src/container_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/container_utils.cc -------------------------------------------------------------------------------- /src/cutlass_micro_kernel_property.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/cutlass_micro_kernel_property.cc -------------------------------------------------------------------------------- /src/data_type_infer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/data_type_infer.cc -------------------------------------------------------------------------------- /src/debug/check_conflict_id.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/debug/check_conflict_id.cc -------------------------------------------------------------------------------- /src/debug/logger.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/debug/logger.cc -------------------------------------------------------------------------------- /src/debug/match_ast.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/debug/match_ast.cc -------------------------------------------------------------------------------- /src/debug/profile.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/debug/profile.cc -------------------------------------------------------------------------------- /src/driver.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/driver.cc -------------------------------------------------------------------------------- /src/driver/array.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/driver/array.cc -------------------------------------------------------------------------------- /src/driver/device.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/driver/device.cc -------------------------------------------------------------------------------- /src/driver/target.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/driver/target.cc -------------------------------------------------------------------------------- /src/except.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/except.cc -------------------------------------------------------------------------------- /src/expr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/expr.cc -------------------------------------------------------------------------------- /src/for_property.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/for_property.cc -------------------------------------------------------------------------------- /src/frontend/frontend_var.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/frontend/frontend_var.cc -------------------------------------------------------------------------------- /src/frontend/inlined_invoke.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/frontend/inlined_invoke.cc -------------------------------------------------------------------------------- /src/func.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/func.cc -------------------------------------------------------------------------------- /src/get_new_name.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/get_new_name.cc -------------------------------------------------------------------------------- /src/hash.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/hash.cc -------------------------------------------------------------------------------- /src/hash_combine.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/hash_combine.cc -------------------------------------------------------------------------------- /src/id.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/id.cc -------------------------------------------------------------------------------- /src/math/bounds.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/math/bounds.cc -------------------------------------------------------------------------------- /src/math/gen_pb_expr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/math/gen_pb_expr.cc -------------------------------------------------------------------------------- /src/math/min_max.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/math/min_max.cc -------------------------------------------------------------------------------- /src/math/parse_pb_expr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/math/parse_pb_expr.cc -------------------------------------------------------------------------------- /src/math/presburger.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/math/presburger.cc -------------------------------------------------------------------------------- /src/metadata.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/metadata.cc -------------------------------------------------------------------------------- /src/mutator.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/mutator.cc -------------------------------------------------------------------------------- /src/pass/annotate_conds.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/annotate_conds.cc -------------------------------------------------------------------------------- /src/pass/const_fold.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/const_fold.cc -------------------------------------------------------------------------------- /src/pass/cpu/lower_parallel_reduction.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/cpu/lower_parallel_reduction.cc -------------------------------------------------------------------------------- /src/pass/flatten_stmt_seq.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/flatten_stmt_seq.cc -------------------------------------------------------------------------------- /src/pass/float_simplify.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/float_simplify.cc -------------------------------------------------------------------------------- /src/pass/gpu/lower_parallel_reduction.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/gpu/lower_parallel_reduction.cc -------------------------------------------------------------------------------- /src/pass/gpu/lower_vector.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/gpu/lower_vector.cc -------------------------------------------------------------------------------- /src/pass/gpu/make_sync.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/gpu/make_sync.cc -------------------------------------------------------------------------------- /src/pass/gpu/multiplex_buffers.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/gpu/multiplex_buffers.cc -------------------------------------------------------------------------------- /src/pass/gpu/normalize_thread_dims.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/gpu/normalize_thread_dims.cc -------------------------------------------------------------------------------- /src/pass/gpu/normalize_threads.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/gpu/normalize_threads.cc -------------------------------------------------------------------------------- /src/pass/gpu/normalize_var_in_kernel.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/gpu/normalize_var_in_kernel.cc -------------------------------------------------------------------------------- /src/pass/gpu/simplex_buffers.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/gpu/simplex_buffers.cc -------------------------------------------------------------------------------- /src/pass/hoist_return_vars.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/hoist_return_vars.cc -------------------------------------------------------------------------------- /src/pass/hoist_var_over_stmt_seq.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/hoist_var_over_stmt_seq.cc -------------------------------------------------------------------------------- /src/pass/make_heap_alloc.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/make_heap_alloc.cc -------------------------------------------------------------------------------- /src/pass/make_parallel_reduction.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/make_parallel_reduction.cc -------------------------------------------------------------------------------- /src/pass/make_reduction.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/make_reduction.cc -------------------------------------------------------------------------------- /src/pass/merge_and_hoist_if.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/merge_and_hoist_if.cc -------------------------------------------------------------------------------- /src/pass/move_out_first_or_last_iter.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/move_out_first_or_last_iter.cc -------------------------------------------------------------------------------- /src/pass/normalize_loops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/normalize_loops.cc -------------------------------------------------------------------------------- /src/pass/prop_one_time_use.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/prop_one_time_use.cc -------------------------------------------------------------------------------- /src/pass/refine_sign_data_type.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/refine_sign_data_type.cc -------------------------------------------------------------------------------- /src/pass/remove_cyclic_assign.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/remove_cyclic_assign.cc -------------------------------------------------------------------------------- /src/pass/remove_dead_var.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/remove_dead_var.cc -------------------------------------------------------------------------------- /src/pass/remove_writes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/remove_writes.cc -------------------------------------------------------------------------------- /src/pass/rename_var.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/rename_var.cc -------------------------------------------------------------------------------- /src/pass/replace_uses.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/replace_uses.cc -------------------------------------------------------------------------------- /src/pass/scalar_prop_const.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/scalar_prop_const.cc -------------------------------------------------------------------------------- /src/pass/shrink_for.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/shrink_for.cc -------------------------------------------------------------------------------- /src/pass/shrink_linear_indices.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/shrink_linear_indices.cc -------------------------------------------------------------------------------- /src/pass/shrink_var.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/shrink_var.cc -------------------------------------------------------------------------------- /src/pass/simplify.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/simplify.cc -------------------------------------------------------------------------------- /src/pass/sink_var.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/sink_var.cc -------------------------------------------------------------------------------- /src/pass/tensor_prop_const.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/tensor_prop_const.cc -------------------------------------------------------------------------------- /src/pass/undo_make_reduction.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/undo_make_reduction.cc -------------------------------------------------------------------------------- /src/pass/use_builtin_div.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/use_builtin_div.cc -------------------------------------------------------------------------------- /src/pass/z3_simplify.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/pass/z3_simplify.cc -------------------------------------------------------------------------------- /src/probability/rand_ctx.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/probability/rand_ctx.cc -------------------------------------------------------------------------------- /src/probability/rand_var.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/probability/rand_var.cc -------------------------------------------------------------------------------- /src/random.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/random.cc -------------------------------------------------------------------------------- /src/reduce_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/reduce_op.cc -------------------------------------------------------------------------------- /src/schedule.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule.cc -------------------------------------------------------------------------------- /src/schedule/as_matmul.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/as_matmul.cc -------------------------------------------------------------------------------- /src/schedule/auto_fission_fuse.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/auto_fission_fuse.cc -------------------------------------------------------------------------------- /src/schedule/auto_inline.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/auto_inline.cc -------------------------------------------------------------------------------- /src/schedule/auto_mem_layout.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/auto_mem_layout.cc -------------------------------------------------------------------------------- /src/schedule/auto_parallelize.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/auto_parallelize.cc -------------------------------------------------------------------------------- /src/schedule/auto_pluto.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/auto_pluto.cc -------------------------------------------------------------------------------- /src/schedule/auto_reorder.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/auto_reorder.cc -------------------------------------------------------------------------------- /src/schedule/auto_set_mem_type.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/auto_set_mem_type.cc -------------------------------------------------------------------------------- /src/schedule/auto_swap.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/auto_swap.cc -------------------------------------------------------------------------------- /src/schedule/auto_unroll.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/auto_unroll.cc -------------------------------------------------------------------------------- /src/schedule/auto_use_lib.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/auto_use_lib.cc -------------------------------------------------------------------------------- /src/schedule/blend.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/blend.cc -------------------------------------------------------------------------------- /src/schedule/cache.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/cache.cc -------------------------------------------------------------------------------- /src/schedule/check_loop_order.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/check_loop_order.cc -------------------------------------------------------------------------------- /src/schedule/check_var_cross_parallel.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/check_var_cross_parallel.cc -------------------------------------------------------------------------------- /src/schedule/fission.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/fission.cc -------------------------------------------------------------------------------- /src/schedule/fuse.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/fuse.cc -------------------------------------------------------------------------------- /src/schedule/hoist_selected_var.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/hoist_selected_var.cc -------------------------------------------------------------------------------- /src/schedule/inlining.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/inlining.cc -------------------------------------------------------------------------------- /src/schedule/lower_cutlass_micro_block.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/lower_cutlass_micro_block.cc -------------------------------------------------------------------------------- /src/schedule/merge.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/merge.cc -------------------------------------------------------------------------------- /src/schedule/move_to.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/move_to.cc -------------------------------------------------------------------------------- /src/schedule/parallelize.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/parallelize.cc -------------------------------------------------------------------------------- /src/schedule/parallelize_as.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/parallelize_as.cc -------------------------------------------------------------------------------- /src/schedule/permute.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/permute.cc -------------------------------------------------------------------------------- /src/schedule/pluto.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/pluto.cc -------------------------------------------------------------------------------- /src/schedule/reorder.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/reorder.cc -------------------------------------------------------------------------------- /src/schedule/separate_tail.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/separate_tail.cc -------------------------------------------------------------------------------- /src/schedule/set_mem_type.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/set_mem_type.cc -------------------------------------------------------------------------------- /src/schedule/split.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/split.cc -------------------------------------------------------------------------------- /src/schedule/swap.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/swap.cc -------------------------------------------------------------------------------- /src/schedule/unroll.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/unroll.cc -------------------------------------------------------------------------------- /src/schedule/var_merge.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/var_merge.cc -------------------------------------------------------------------------------- /src/schedule/var_reorder.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/var_reorder.cc -------------------------------------------------------------------------------- /src/schedule/var_split.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/var_split.cc -------------------------------------------------------------------------------- /src/schedule/var_squeeze.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/var_squeeze.cc -------------------------------------------------------------------------------- /src/schedule/var_unsqueeze.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/var_unsqueeze.cc -------------------------------------------------------------------------------- /src/schedule/vectorize.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/schedule/vectorize.cc -------------------------------------------------------------------------------- /src/selector.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/selector.cc -------------------------------------------------------------------------------- /src/serialize/load_ast.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/serialize/load_ast.cc -------------------------------------------------------------------------------- /src/serialize/load_driver.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/serialize/load_driver.cc -------------------------------------------------------------------------------- /src/serialize/mangle.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/serialize/mangle.cc -------------------------------------------------------------------------------- /src/serialize/print_ast.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/serialize/print_ast.cc -------------------------------------------------------------------------------- /src/serialize/print_driver.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/serialize/print_driver.cc -------------------------------------------------------------------------------- /src/stmt.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/stmt.cc -------------------------------------------------------------------------------- /src/sub_tree.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/sub_tree.cc -------------------------------------------------------------------------------- /src/tensor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/tensor.cc -------------------------------------------------------------------------------- /src/timeout.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/timeout.cc -------------------------------------------------------------------------------- /src/type/data_type.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/type/data_type.cc -------------------------------------------------------------------------------- /src/visitor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/src/visitor.cc -------------------------------------------------------------------------------- /test/00.hello_world/test_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/00.hello_world/test_basic.py -------------------------------------------------------------------------------- /test/00.hello_world/test_doc_examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/00.hello_world/test_doc_examples.py -------------------------------------------------------------------------------- /test/00.hello_world/test_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/00.hello_world/test_io.py -------------------------------------------------------------------------------- /test/00.hello_world/test_serialize_ast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/00.hello_world/test_serialize_ast.py -------------------------------------------------------------------------------- /test/00.hello_world/test_serialize_driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/00.hello_world/test_serialize_driver.py -------------------------------------------------------------------------------- /test/10.analyze/test_find_stmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/10.analyze/test_find_stmt.py -------------------------------------------------------------------------------- /test/10.analyze/test_structural_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/10.analyze/test_structural_feature.py -------------------------------------------------------------------------------- /test/20.pass/test_float_simplify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_float_simplify.py -------------------------------------------------------------------------------- /test/20.pass/test_hoist_var_over_stmt_seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_hoist_var_over_stmt_seq.py -------------------------------------------------------------------------------- /test/20.pass/test_lower.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_lower.py -------------------------------------------------------------------------------- /test/20.pass/test_make_heap_alloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_make_heap_alloc.py -------------------------------------------------------------------------------- /test/20.pass/test_make_reduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_make_reduction.py -------------------------------------------------------------------------------- /test/20.pass/test_merge_and_hoist_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_merge_and_hoist_if.py -------------------------------------------------------------------------------- /test/20.pass/test_move_out_first_or_later_iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_move_out_first_or_later_iter.py -------------------------------------------------------------------------------- /test/20.pass/test_prop_one_time_use.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_prop_one_time_use.py -------------------------------------------------------------------------------- /test/20.pass/test_refine_sign_data_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_refine_sign_data_type.py -------------------------------------------------------------------------------- /test/20.pass/test_remove_cyclic_assign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_remove_cyclic_assign.py -------------------------------------------------------------------------------- /test/20.pass/test_remove_dead_var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_remove_dead_var.py -------------------------------------------------------------------------------- /test/20.pass/test_remove_writes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_remove_writes.py -------------------------------------------------------------------------------- /test/20.pass/test_scalar_prop_const.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_scalar_prop_const.py -------------------------------------------------------------------------------- /test/20.pass/test_shrink_for.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_shrink_for.py -------------------------------------------------------------------------------- /test/20.pass/test_shrink_var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_shrink_var.py -------------------------------------------------------------------------------- /test/20.pass/test_simplify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_simplify.py -------------------------------------------------------------------------------- /test/20.pass/test_sink_var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_sink_var.py -------------------------------------------------------------------------------- /test/20.pass/test_tensor_prop_const.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_tensor_prop_const.py -------------------------------------------------------------------------------- /test/20.pass/test_use_builtin_div.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/20.pass/test_use_builtin_div.py -------------------------------------------------------------------------------- /test/21.autograd/test_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/21.autograd/test_grad.py -------------------------------------------------------------------------------- /test/21.autograd/test_invert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/21.autograd/test_invert.py -------------------------------------------------------------------------------- /test/21.autograd/test_jacrev.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/21.autograd/test_jacrev.py -------------------------------------------------------------------------------- /test/21.autograd/test_output_intermediates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/21.autograd/test_output_intermediates.py -------------------------------------------------------------------------------- /test/21.autograd/test_pytorch_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/21.autograd/test_pytorch_integration.py -------------------------------------------------------------------------------- /test/21.autograd/test_tape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/21.autograd/test_tape.py -------------------------------------------------------------------------------- /test/21.autograd/test_user_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/21.autograd/test_user_grad.py -------------------------------------------------------------------------------- /test/30.schedule/test_as_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_as_matmul.py -------------------------------------------------------------------------------- /test/30.schedule/test_blend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_blend.py -------------------------------------------------------------------------------- /test/30.schedule/test_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_cache.py -------------------------------------------------------------------------------- /test/30.schedule/test_cache_reduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_cache_reduction.py -------------------------------------------------------------------------------- /test/30.schedule/test_fission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_fission.py -------------------------------------------------------------------------------- /test/30.schedule/test_fuse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_fuse.py -------------------------------------------------------------------------------- /test/30.schedule/test_inline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_inline.py -------------------------------------------------------------------------------- /test/30.schedule/test_memoized_schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_memoized_schedules.py -------------------------------------------------------------------------------- /test/30.schedule/test_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_merge.py -------------------------------------------------------------------------------- /test/30.schedule/test_move_to.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_move_to.py -------------------------------------------------------------------------------- /test/30.schedule/test_parallelize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_parallelize.py -------------------------------------------------------------------------------- /test/30.schedule/test_parallelize_as.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_parallelize_as.py -------------------------------------------------------------------------------- /test/30.schedule/test_permute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_permute.py -------------------------------------------------------------------------------- /test/30.schedule/test_pluto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_pluto.py -------------------------------------------------------------------------------- /test/30.schedule/test_reorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_reorder.py -------------------------------------------------------------------------------- /test/30.schedule/test_separate_tail.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_separate_tail.py -------------------------------------------------------------------------------- /test/30.schedule/test_set_mem_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_set_mem_type.py -------------------------------------------------------------------------------- /test/30.schedule/test_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_split.py -------------------------------------------------------------------------------- /test/30.schedule/test_swap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_swap.py -------------------------------------------------------------------------------- /test/30.schedule/test_unroll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_unroll.py -------------------------------------------------------------------------------- /test/30.schedule/test_var_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_var_merge.py -------------------------------------------------------------------------------- /test/30.schedule/test_var_reorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_var_reorder.py -------------------------------------------------------------------------------- /test/30.schedule/test_var_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_var_split.py -------------------------------------------------------------------------------- /test/30.schedule/test_var_squeeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_var_squeeze.py -------------------------------------------------------------------------------- /test/30.schedule/test_var_unsqueeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_var_unsqueeze.py -------------------------------------------------------------------------------- /test/30.schedule/test_vectorize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/30.schedule/test_vectorize.py -------------------------------------------------------------------------------- /test/31.auto_schedule/test_auto_fission_fuse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/31.auto_schedule/test_auto_fission_fuse.py -------------------------------------------------------------------------------- /test/31.auto_schedule/test_auto_inline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/31.auto_schedule/test_auto_inline.py -------------------------------------------------------------------------------- /test/31.auto_schedule/test_auto_mem_layout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/31.auto_schedule/test_auto_mem_layout.py -------------------------------------------------------------------------------- /test/31.auto_schedule/test_auto_parallelize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/31.auto_schedule/test_auto_parallelize.py -------------------------------------------------------------------------------- /test/31.auto_schedule/test_auto_pluto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/31.auto_schedule/test_auto_pluto.py -------------------------------------------------------------------------------- /test/31.auto_schedule/test_auto_reorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/31.auto_schedule/test_auto_reorder.py -------------------------------------------------------------------------------- /test/31.auto_schedule/test_auto_set_mem_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/31.auto_schedule/test_auto_set_mem_type.py -------------------------------------------------------------------------------- /test/31.auto_schedule/test_auto_swap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/31.auto_schedule/test_auto_swap.py -------------------------------------------------------------------------------- /test/31.auto_schedule/test_auto_use_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/31.auto_schedule/test_auto_use_lib.py -------------------------------------------------------------------------------- /test/40.codegen/cpu/test_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/cpu/test_cpu.py -------------------------------------------------------------------------------- /test/40.codegen/cpu/test_cpu_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/cpu/test_cpu_reduce.py -------------------------------------------------------------------------------- /test/40.codegen/cpu/test_mkl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/cpu/test_mkl.py -------------------------------------------------------------------------------- /test/40.codegen/gpu/test_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/gpu/test_gpu.py -------------------------------------------------------------------------------- /test/40.codegen/gpu/test_gpu_cublas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/gpu/test_gpu_cublas.py -------------------------------------------------------------------------------- /test/40.codegen/gpu/test_gpu_cutlass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/gpu/test_gpu_cutlass.py -------------------------------------------------------------------------------- /test/40.codegen/gpu/test_gpu_normalize_buffer_shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/gpu/test_gpu_normalize_buffer_shapes.py -------------------------------------------------------------------------------- /test/40.codegen/gpu/test_gpu_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/gpu/test_gpu_reduce.py -------------------------------------------------------------------------------- /test/40.codegen/gpu/test_gpu_sync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/gpu/test_gpu_sync.py -------------------------------------------------------------------------------- /test/40.codegen/gpu/test_gpu_vector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/gpu/test_gpu_vector.py -------------------------------------------------------------------------------- /test/40.codegen/gpu/test_gpu_warpshuffle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/40.codegen/gpu/test_gpu_warpshuffle.py -------------------------------------------------------------------------------- /test/50.frontend/test_ast_inline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/50.frontend/test_ast_inline.py -------------------------------------------------------------------------------- /test/50.frontend/test_error_report.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/50.frontend/test_error_report.py -------------------------------------------------------------------------------- /test/50.frontend/test_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/50.frontend/test_jit.py -------------------------------------------------------------------------------- /test/50.frontend/test_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/50.frontend/test_ops.py -------------------------------------------------------------------------------- /test/50.frontend/test_staging_inline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/50.frontend/test_staging_inline.py -------------------------------------------------------------------------------- /test/50.frontend/test_transformer_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/50.frontend/test_transformer_basic.py -------------------------------------------------------------------------------- /test/50.frontend/test_var.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/50.frontend/test_var.py -------------------------------------------------------------------------------- /test/60.libop/test_assign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_assign.py -------------------------------------------------------------------------------- /test/60.libop/test_augment_assign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_augment_assign.py -------------------------------------------------------------------------------- /test/60.libop/test_binary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_binary.py -------------------------------------------------------------------------------- /test/60.libop/test_concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_concat.py -------------------------------------------------------------------------------- /test/60.libop/test_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_conv.py -------------------------------------------------------------------------------- /test/60.libop/test_einsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_einsum.py -------------------------------------------------------------------------------- /test/60.libop/test_expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_expand.py -------------------------------------------------------------------------------- /test/60.libop/test_flatten_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_flatten_onnx.py -------------------------------------------------------------------------------- /test/60.libop/test_flatten_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_flatten_pytorch.py -------------------------------------------------------------------------------- /test/60.libop/test_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_gemm.py -------------------------------------------------------------------------------- /test/60.libop/test_logsumexp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_logsumexp.py -------------------------------------------------------------------------------- /test/60.libop/test_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_matmul.py -------------------------------------------------------------------------------- /test/60.libop/test_ones.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_ones.py -------------------------------------------------------------------------------- /test/60.libop/test_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_pooling.py -------------------------------------------------------------------------------- /test/60.libop/test_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_reduce.py -------------------------------------------------------------------------------- /test/60.libop/test_reshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_reshape.py -------------------------------------------------------------------------------- /test/60.libop/test_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_softmax.py -------------------------------------------------------------------------------- /test/60.libop/test_squeeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_squeeze.py -------------------------------------------------------------------------------- /test/60.libop/test_stack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_stack.py -------------------------------------------------------------------------------- /test/60.libop/test_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_transpose.py -------------------------------------------------------------------------------- /test/60.libop/test_unary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_unary.py -------------------------------------------------------------------------------- /test/60.libop/test_unsqueeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_unsqueeze.py -------------------------------------------------------------------------------- /test/60.libop/test_zeros.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/60.libop/test_zeros.py -------------------------------------------------------------------------------- /test/70.program/test_common_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/70.program/test_common_transforms.py -------------------------------------------------------------------------------- /test/70.program/test_gpu_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/70.program/test_gpu_conv2d.py -------------------------------------------------------------------------------- /test/70.program/test_gpu_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/70.program/test_gpu_softmax.py -------------------------------------------------------------------------------- /test/70.program/test_program_with_micro_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/roastduck/FreeTensor/HEAD/test/70.program/test_program_with_micro_kernel.py -------------------------------------------------------------------------------- /with-cuda.toml: -------------------------------------------------------------------------------- 1 | [cmake.options] 2 | FT_WITH_CUDA = "ON" 3 | -------------------------------------------------------------------------------- /with-mkl.toml: -------------------------------------------------------------------------------- 1 | [cmake.options] 2 | FT_WITH_MKL = "ON" 3 | -------------------------------------------------------------------------------- /with-pytorch.toml: -------------------------------------------------------------------------------- 1 | [cmake.options] 2 | FT_WITH_PYTORCH = "ON" 3 | --------------------------------------------------------------------------------