├── .clang-format ├── .gitignore ├── CMakeLists.txt ├── LICENSE ├── README.md ├── benchmarks ├── C │ └── templates │ │ ├── ba.h │ │ ├── ba_driver.c │ │ ├── blas_wrapper.c │ │ ├── cache_loop_driver.c │ │ ├── dot_driver.c │ │ ├── enzyme_ba.c │ │ ├── enzyme_dot.c │ │ ├── enzyme_gmm.c │ │ ├── enzyme_hand.c │ │ ├── enzyme_hand_packed.c │ │ ├── enzyme_kernels.c │ │ ├── enzyme_lstm.c │ │ ├── enzyme_main_term.c │ │ ├── gmm.h │ │ ├── gmm_adbench_driver.c │ │ ├── gmm_blas.c │ │ ├── gmm_compressed_driver.c │ │ ├── gmm_driver.c │ │ ├── hand.h │ │ ├── hand_driver.c │ │ ├── kernels.c │ │ ├── loop_driver.c │ │ ├── lstm.h │ │ ├── lstm_driver.c │ │ ├── main_term_raised.c │ │ ├── matmul_driver.c │ │ ├── matvec_driver.c │ │ ├── memusage.c │ │ ├── memusage.h │ │ ├── microbench_dot_driver.c │ │ ├── microbench_main_term_driver.c │ │ ├── microbench_matmul_driver.c │ │ ├── microbench_vecadd_driver.c │ │ ├── microbench_vecscal_driver.c │ │ ├── microbenchmark.c │ │ ├── microbenchmark_enzyme.c │ │ ├── microkernels.c │ │ ├── mlir_c_abi.c │ │ ├── mlir_c_abi.h │ │ ├── shared_types.h │ │ ├── trimatvec_driver.c │ │ └── vecmat_driver.c ├── analyze.py ├── analyze_mem.py ├── ba │ ├── ba.h │ ├── ba_enzyme.c │ ├── build.py │ ├── enzyme_ba_reproj.mlir │ ├── enzyme_ba_w.mlir │ ├── lagrad_ba.mlir │ └── main.c ├── compile.py ├── count_malloc_size.py ├── data │ ├── ba10_n1197_m126327_p563734.txt │ ├── ba1_n49_m7776_p31843.txt │ ├── gmm_d10_K25.txt │ ├── gmm_d128_K200.txt │ ├── gmm_d128_K200_results.txt │ ├── gmm_results.txt │ ├── hand1_t26_c100.txt │ └── hand_model │ │ ├── anchors.txt │ │ ├── atl_triangle_labels.txt │ │ ├── bones.txt │ │ ├── triangles.txt │ │ └── vertices.txt ├── df_utils.py ├── gmm │ ├── build.py │ ├── full │ │ ├── enzyme_gmm_full.mlir │ │ ├── gmm_full_blas.c │ │ ├── gmm_full_enzyme.c │ │ ├── lagrad_gmm_full.mlir │ │ └── main.c │ ├── include │ │ ├── gmm.h │ │ └── gmm_types.h │ ├── packed │ │ ├── enzyme_gmm_packed.mlir │ │ ├── gmm_packed_enzyme.c │ │ ├── lagrad_gmm_packed.mlir │ │ └── main.c │ ├── run_with_datasets.py │ └── tri │ │ ├── enzyme_gmm_tri.mlir │ │ ├── gmm_tri_enzyme.c │ │ ├── gmm_tri_main.c │ │ └── lagrad_gmm_tri.mlir ├── hand │ ├── build.py │ ├── enzyme_hand.mlir │ ├── hand.h │ ├── hand_enzyme.c │ ├── hand_types.h │ ├── lagrad_hand.mlir │ ├── main.c │ └── run_with_datasets.py ├── lagrad_utils │ ├── build.ronin │ ├── lagrad_utils.c │ ├── lagrad_utils.h │ ├── memusage.c │ └── memusage.h ├── lstm │ ├── build.py │ ├── lagrad_lstm.mlir │ ├── lstm.h │ ├── lstm_enzyme.c │ └── main.c ├── main.py ├── main_term │ ├── build.ronin │ ├── enzyme_kernel.mlir │ ├── main.c │ ├── main_term.h │ ├── main_term_enzyme.c │ └── main_term_kernel.mlir ├── matmul │ ├── blas_wrapper.c │ ├── build.py │ ├── enzyme_kernel.mlir │ ├── main.c │ ├── matmul_enzyme.c │ └── matmul_kernel.mlir ├── microbenchmarks.py ├── mlir │ └── templates │ │ ├── ba.mlir │ │ ├── ba_enzyme_reproj.mlir │ │ ├── ba_enzyme_w.mlir │ │ ├── cache_loop.mlir │ │ ├── cache_loop_enzyme.mlir │ │ ├── dot.mlir │ │ ├── gmm.mlir │ │ ├── gmm_adbench_enzyme.mlir │ │ ├── gmm_adbench_lagrad.mlir │ │ ├── gmm_adbench_lagrad_buf.mlir │ │ ├── gmm_buf.mlir │ │ ├── gmm_buf_compressed.mlir │ │ ├── gmm_buf_optimized.mlir │ │ ├── gmm_compressed.mlir │ │ ├── gmm_differentiated.mlir │ │ ├── gmm_enzyme.mlir │ │ ├── gmm_tensor_bufferized_opt.mlir │ │ ├── gmm_tensor_compressed.mlir │ │ ├── gmm_tri.mlir │ │ ├── hand.mlir │ │ ├── hand_complicated_enzyme.mlir │ │ ├── hand_enzyme.mlir │ │ ├── hand_inlined.mlir │ │ ├── lstm.mlir │ │ ├── lstm_enzyme.mlir │ │ ├── lstm_enzyme_samemem.mlir │ │ ├── lstm_hand_bufferized.mlir │ │ ├── main_term.mlir │ │ ├── main_term_bufferized.mlir │ │ ├── matmul.mlir │ │ ├── matvec.mlir │ │ ├── microbenchmark.mlir │ │ ├── microbenchmarks │ │ ├── dot.mlir │ │ ├── main_term.mlir │ │ ├── matmul.mlir │ │ ├── vecadd.mlir │ │ └── vecscal.mlir │ │ ├── nested_loop_enzyme.mlir │ │ ├── nested_loop_lagrad.mlir │ │ ├── trimatvec.mlir │ │ └── vecmat.mlir ├── nn │ ├── DELETEME_postlibrary.mlir │ ├── blas_wrapper.c │ ├── build.py │ ├── enzyme_mlir_nn.mlir │ ├── enzyme_nn.c │ ├── lagrad_nn.mlir │ ├── main.c │ ├── nn.h │ └── read_data.c ├── pytorch_benchmarks.py ├── results │ ├── basparsemat.txt │ ├── dot_first_arg.json │ ├── gmm_1000_200_128.json │ ├── gmm_1000_25_10.json │ ├── matmul_first_arg.json │ ├── matvec_first_arg.json │ ├── trimicrobench.csv │ ├── trmv_1024.json │ ├── trmv_2048.json │ └── vecmat_first_arg.json ├── run_ba.py ├── run_gmm.py ├── run_hand.py ├── run_lstm.py ├── run_with_memory.py ├── speedups.py ├── speedups_data.py ├── take_means.py └── trmv │ ├── build.py │ ├── enzyme_trmv.mlir │ ├── main.c │ ├── run_with_datasets.py │ ├── trmv.h │ ├── trmv.mlir │ ├── trmv_c_enzyme.c │ └── utils.c ├── include ├── CMakeLists.txt └── LAGrad │ ├── Analysis.h │ ├── CMakeLists.txt │ ├── LAGradDialect.h │ ├── LAGradDialect.td │ ├── LAGradOps.h │ ├── LAGradOps.td │ ├── Logger.h │ ├── Passes.h │ ├── Transforms.h │ └── Utils.h ├── lagrad-opt ├── CMakeLists.txt └── lagrad-opt.cpp ├── lib ├── CMakeLists.txt ├── LAGrad │ ├── ActivityAnalysis.cpp │ ├── Bufferize.cpp │ ├── CMakeLists.txt │ ├── DeadCodeElimination.cpp │ ├── DebugNames.cpp │ ├── DynamicShapeAnalysis.cpp │ ├── ElementwiseToAffine.cpp │ ├── InsertExtractAnalysis.cpp │ ├── LAGradDialect.cpp │ ├── LAGradOps.cpp │ ├── LAGradTransforms.cpp │ ├── LinalgCanonicalize.cpp │ ├── LinalgKnownLibraryCalls.cpp │ ├── LinalgVectorJacobianProducts.cpp │ ├── LoopAllocHoisting.cpp │ ├── LoopNestAnalysis.cpp │ ├── LowerGrad.cpp │ ├── LowerToLLVM.cpp │ ├── PrimalCaching.cpp │ ├── SCFVectorJacobianProducts.cpp │ ├── SparsePropagation.cpp │ ├── StaticAllocs.cpp │ ├── TensorConstantFolding.cpp │ ├── TriangularLoops.cpp │ ├── TriangularPacking.cpp │ └── Utils.cpp └── LLFastMath │ ├── CMakeLists.txt │ └── LLFastMath.cpp ├── memory_monitor ├── Makefile └── memory_monitor.c ├── mlir ├── ronin_phases ├── __init__.py ├── build_llvm.py ├── memory_monitor.py ├── render.py └── replace_hand_opt.py ├── scripts ├── bufferize ├── lower_to_llvm ├── run_benchmark.sh ├── run_external_call.sh ├── run_matmul.sh ├── run_matvec.sh ├── run_reference.sh └── run_vecmat.sh └── test ├── CMakeLists.txt ├── Driver ├── dot.c ├── dot_reference.c ├── matmul.c ├── matvec.c ├── simple.c ├── standaloneabi.h └── vecmat.c ├── LAGrad ├── activity │ ├── ifelse.mlir │ └── unused.mlir ├── add3.mlir ├── arith │ ├── negf.mlir │ └── tanh.mlir ├── arithops.mlir ├── arrsum.mlir ├── ba │ └── ba.mlir ├── batch_matmul.mlir ├── before.mlir ├── bufferize │ ├── collapse_shape.mlir │ ├── collapse_shape_2d.mlir │ ├── collapse_shape_col.mlir │ ├── grad_collapse_shape.mlir │ ├── grad_collapse_shape_2d.mlir │ └── grad_collapse_shape_col.mlir ├── cache │ ├── nested_pow.mlir │ ├── nested_pow_tensor.mlir │ ├── pow.mlir │ └── pow_tensor.mlir ├── cross_product.mlir ├── custom_grad_init.mlir ├── diff │ ├── const_memref.mlir │ ├── dot.mlir │ ├── generic.mlir │ ├── matvec_buffer.mlir │ ├── memref.mlir │ └── two_funcs.mlir ├── dot.mlir ├── dot_second_arg.mlir ├── dummy.mlir ├── functioncall.mlir ├── generic │ ├── broadcast.mlir │ ├── broadcast_square.mlir │ ├── closure.mlir │ ├── colsum.mlir │ ├── dot.mlir │ ├── functioncall.mlir │ ├── generic_with_call.mlir │ ├── gmm_einsum.mlir │ ├── logsumexp.mlir │ ├── logsumexp_1d.mlir │ ├── logsumexp_add2.mlir │ ├── nonconst_out.mlir │ ├── sub.mlir │ ├── three_args.mlir │ └── vecmat.mlir ├── gmm │ ├── compressed_outer_product.mlir │ ├── compressed_qtimesx.mlir │ ├── compressed_vecmat.mlir │ ├── gmm_objective_full.mlir │ └── main_term.mlir ├── hand │ ├── angle_axis_to_rotation_matrix.mlir │ ├── apply_global_transform.mlir │ ├── compute_err_simple.mlir │ ├── get_posed_relatives.mlir │ ├── get_posed_relatives_body.mlir │ ├── hand_objective.mlir │ ├── hand_objective_complicated.mlir │ ├── relatives_to_absolutes.mlir │ ├── relatives_to_absolutes_scratch.mlir │ ├── skinned_vertex_subset.mlir │ └── to_pose_params.mlir ├── if_function.mlir ├── ifdifferenttype.mlir ├── ifelse.mlir ├── insert_multi.mlir ├── kernels │ ├── dot.mlir │ ├── enzyme_dot.mlir │ ├── matmul.mlir │ ├── matvec.mlir │ ├── vecmat.mlir │ └── vecmat_kernel.c ├── lstm │ ├── lstm_model.mlir │ ├── lstm_objective.mlir │ └── lstm_predict.mlir ├── matsum.mlir ├── matvec.mlir ├── nestedcall.mlir ├── nn │ ├── batched_crossentropy.mlir │ ├── crossentropy.mlir │ └── nn.mlir ├── quadratic.mlir ├── relu.mlir ├── scf │ ├── arrmax.mlir │ ├── forintermediate.mlir │ ├── fortensor.mlir │ ├── if_funcall.mlir │ ├── nested_with_slice.mlir │ ├── product.mlir │ ├── selectdifferenttype.mlir │ └── wrt_iter_arg.mlir ├── scfdot.mlir ├── sparse │ ├── adjoint_err_nest.mlir │ ├── colhot_broadcast_mul.mlir │ ├── onehot_elementwise_square.mlir │ ├── onehot_matmul.mlir │ ├── onehot_sumreduce.mlir │ ├── rowhot_broadcast_mul.mlir │ ├── rowhot_matmul.mlir │ └── rowhot_tensor_insert.mlir ├── standalone-opt.mlir ├── standalone-translate.mlir ├── tensor │ ├── insert_overwrite.mlir │ ├── insert_overwrite_scalar.mlir │ ├── insert_slice.mlir │ └── insert_slice_loop.mlir ├── tensor_extract.mlir ├── tensor_extract_slice.mlir └── twoargs.mlir ├── lit.cfg.py ├── lit.site.cfg.py.in └── pytest ├── Makefile ├── benchmark_io.py ├── build.py ├── mlir_bindings.py ├── pytorch_ref ├── BASparseMat.py ├── pytorch_ba.py ├── pytorch_gmm.py ├── pytorch_hand.py ├── pytorch_lstm.py ├── pytorch_nn.py └── utils.py ├── stdout_parser.py ├── test_activity_analysis.py ├── test_ba.py ├── test_bufferization.py ├── test_enzyme.py ├── test_gmm.py ├── test_grad.py ├── test_grad_v2.py ├── test_hand.py ├── test_lstm.py ├── test_nn.py ├── test_scf.py ├── test_structured_sparsify.py ├── test_tensor.py └── toolchain.py /.clang-format: -------------------------------------------------------------------------------- 1 | BasedOnStyle: LLVM 2 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks/C/templates/ba.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/ba.h -------------------------------------------------------------------------------- /benchmarks/C/templates/ba_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/ba_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/blas_wrapper.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/blas_wrapper.c -------------------------------------------------------------------------------- /benchmarks/C/templates/cache_loop_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/cache_loop_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/dot_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/dot_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/enzyme_ba.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/enzyme_ba.c -------------------------------------------------------------------------------- /benchmarks/C/templates/enzyme_dot.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/enzyme_dot.c -------------------------------------------------------------------------------- /benchmarks/C/templates/enzyme_gmm.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/enzyme_gmm.c -------------------------------------------------------------------------------- /benchmarks/C/templates/enzyme_hand.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/enzyme_hand.c -------------------------------------------------------------------------------- /benchmarks/C/templates/enzyme_hand_packed.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/enzyme_hand_packed.c -------------------------------------------------------------------------------- /benchmarks/C/templates/enzyme_kernels.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/enzyme_kernels.c -------------------------------------------------------------------------------- /benchmarks/C/templates/enzyme_lstm.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/enzyme_lstm.c -------------------------------------------------------------------------------- /benchmarks/C/templates/enzyme_main_term.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/enzyme_main_term.c -------------------------------------------------------------------------------- /benchmarks/C/templates/gmm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/gmm.h -------------------------------------------------------------------------------- /benchmarks/C/templates/gmm_adbench_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/gmm_adbench_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/gmm_blas.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/gmm_blas.c -------------------------------------------------------------------------------- /benchmarks/C/templates/gmm_compressed_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/gmm_compressed_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/gmm_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/gmm_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/hand.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/hand.h -------------------------------------------------------------------------------- /benchmarks/C/templates/hand_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/hand_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/kernels.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/kernels.c -------------------------------------------------------------------------------- /benchmarks/C/templates/loop_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/loop_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/lstm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/lstm.h -------------------------------------------------------------------------------- /benchmarks/C/templates/lstm_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/lstm_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/main_term_raised.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/main_term_raised.c -------------------------------------------------------------------------------- /benchmarks/C/templates/matmul_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/matmul_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/matvec_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/matvec_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/memusage.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/memusage.c -------------------------------------------------------------------------------- /benchmarks/C/templates/memusage.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/memusage.h -------------------------------------------------------------------------------- /benchmarks/C/templates/microbench_dot_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/microbench_dot_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/microbench_main_term_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/microbench_main_term_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/microbench_matmul_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/microbench_matmul_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/microbench_vecadd_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/microbench_vecadd_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/microbench_vecscal_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/microbench_vecscal_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/microbenchmark.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/microbenchmark.c -------------------------------------------------------------------------------- /benchmarks/C/templates/microbenchmark_enzyme.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/microbenchmark_enzyme.c -------------------------------------------------------------------------------- /benchmarks/C/templates/microkernels.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/microkernels.c -------------------------------------------------------------------------------- /benchmarks/C/templates/mlir_c_abi.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/mlir_c_abi.c -------------------------------------------------------------------------------- /benchmarks/C/templates/mlir_c_abi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/mlir_c_abi.h -------------------------------------------------------------------------------- /benchmarks/C/templates/shared_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/shared_types.h -------------------------------------------------------------------------------- /benchmarks/C/templates/trimatvec_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/trimatvec_driver.c -------------------------------------------------------------------------------- /benchmarks/C/templates/vecmat_driver.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/C/templates/vecmat_driver.c -------------------------------------------------------------------------------- /benchmarks/analyze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/analyze.py -------------------------------------------------------------------------------- /benchmarks/analyze_mem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/analyze_mem.py -------------------------------------------------------------------------------- /benchmarks/ba/ba.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/ba/ba.h -------------------------------------------------------------------------------- /benchmarks/ba/ba_enzyme.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/ba/ba_enzyme.c -------------------------------------------------------------------------------- /benchmarks/ba/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/ba/build.py -------------------------------------------------------------------------------- /benchmarks/ba/enzyme_ba_reproj.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/ba/enzyme_ba_reproj.mlir -------------------------------------------------------------------------------- /benchmarks/ba/enzyme_ba_w.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/ba/enzyme_ba_w.mlir -------------------------------------------------------------------------------- /benchmarks/ba/lagrad_ba.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/ba/lagrad_ba.mlir -------------------------------------------------------------------------------- /benchmarks/ba/main.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/ba/main.c -------------------------------------------------------------------------------- /benchmarks/compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/compile.py -------------------------------------------------------------------------------- /benchmarks/count_malloc_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/count_malloc_size.py -------------------------------------------------------------------------------- /benchmarks/data/ba10_n1197_m126327_p563734.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/ba10_n1197_m126327_p563734.txt -------------------------------------------------------------------------------- /benchmarks/data/ba1_n49_m7776_p31843.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/ba1_n49_m7776_p31843.txt -------------------------------------------------------------------------------- /benchmarks/data/gmm_d10_K25.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/gmm_d10_K25.txt -------------------------------------------------------------------------------- /benchmarks/data/gmm_d128_K200.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/gmm_d128_K200.txt -------------------------------------------------------------------------------- /benchmarks/data/gmm_d128_K200_results.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/gmm_d128_K200_results.txt -------------------------------------------------------------------------------- /benchmarks/data/gmm_results.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/gmm_results.txt -------------------------------------------------------------------------------- /benchmarks/data/hand1_t26_c100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/hand1_t26_c100.txt -------------------------------------------------------------------------------- /benchmarks/data/hand_model/anchors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/hand_model/anchors.txt -------------------------------------------------------------------------------- /benchmarks/data/hand_model/atl_triangle_labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/hand_model/atl_triangle_labels.txt -------------------------------------------------------------------------------- /benchmarks/data/hand_model/bones.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/hand_model/bones.txt -------------------------------------------------------------------------------- /benchmarks/data/hand_model/triangles.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/hand_model/triangles.txt -------------------------------------------------------------------------------- /benchmarks/data/hand_model/vertices.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/data/hand_model/vertices.txt -------------------------------------------------------------------------------- /benchmarks/df_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/df_utils.py -------------------------------------------------------------------------------- /benchmarks/gmm/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/build.py -------------------------------------------------------------------------------- /benchmarks/gmm/full/enzyme_gmm_full.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/full/enzyme_gmm_full.mlir -------------------------------------------------------------------------------- /benchmarks/gmm/full/gmm_full_blas.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/full/gmm_full_blas.c -------------------------------------------------------------------------------- /benchmarks/gmm/full/gmm_full_enzyme.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/full/gmm_full_enzyme.c -------------------------------------------------------------------------------- /benchmarks/gmm/full/lagrad_gmm_full.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/full/lagrad_gmm_full.mlir -------------------------------------------------------------------------------- /benchmarks/gmm/full/main.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/full/main.c -------------------------------------------------------------------------------- /benchmarks/gmm/include/gmm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/include/gmm.h -------------------------------------------------------------------------------- /benchmarks/gmm/include/gmm_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/include/gmm_types.h -------------------------------------------------------------------------------- /benchmarks/gmm/packed/enzyme_gmm_packed.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/packed/enzyme_gmm_packed.mlir -------------------------------------------------------------------------------- /benchmarks/gmm/packed/gmm_packed_enzyme.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/packed/gmm_packed_enzyme.c -------------------------------------------------------------------------------- /benchmarks/gmm/packed/lagrad_gmm_packed.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/packed/lagrad_gmm_packed.mlir -------------------------------------------------------------------------------- /benchmarks/gmm/packed/main.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/packed/main.c -------------------------------------------------------------------------------- /benchmarks/gmm/run_with_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/run_with_datasets.py -------------------------------------------------------------------------------- /benchmarks/gmm/tri/enzyme_gmm_tri.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/tri/enzyme_gmm_tri.mlir -------------------------------------------------------------------------------- /benchmarks/gmm/tri/gmm_tri_enzyme.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/tri/gmm_tri_enzyme.c -------------------------------------------------------------------------------- /benchmarks/gmm/tri/gmm_tri_main.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/tri/gmm_tri_main.c -------------------------------------------------------------------------------- /benchmarks/gmm/tri/lagrad_gmm_tri.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/gmm/tri/lagrad_gmm_tri.mlir -------------------------------------------------------------------------------- /benchmarks/hand/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/hand/build.py -------------------------------------------------------------------------------- /benchmarks/hand/enzyme_hand.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/hand/enzyme_hand.mlir -------------------------------------------------------------------------------- /benchmarks/hand/hand.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/hand/hand.h -------------------------------------------------------------------------------- /benchmarks/hand/hand_enzyme.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/hand/hand_enzyme.c -------------------------------------------------------------------------------- /benchmarks/hand/hand_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/hand/hand_types.h -------------------------------------------------------------------------------- /benchmarks/hand/lagrad_hand.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/hand/lagrad_hand.mlir -------------------------------------------------------------------------------- /benchmarks/hand/main.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/hand/main.c -------------------------------------------------------------------------------- /benchmarks/hand/run_with_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/hand/run_with_datasets.py -------------------------------------------------------------------------------- /benchmarks/lagrad_utils/build.ronin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/lagrad_utils/build.ronin -------------------------------------------------------------------------------- /benchmarks/lagrad_utils/lagrad_utils.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/lagrad_utils/lagrad_utils.c -------------------------------------------------------------------------------- /benchmarks/lagrad_utils/lagrad_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/lagrad_utils/lagrad_utils.h -------------------------------------------------------------------------------- /benchmarks/lagrad_utils/memusage.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/lagrad_utils/memusage.c -------------------------------------------------------------------------------- /benchmarks/lagrad_utils/memusage.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/lagrad_utils/memusage.h -------------------------------------------------------------------------------- /benchmarks/lstm/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/lstm/build.py -------------------------------------------------------------------------------- /benchmarks/lstm/lagrad_lstm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/lstm/lagrad_lstm.mlir -------------------------------------------------------------------------------- /benchmarks/lstm/lstm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/lstm/lstm.h -------------------------------------------------------------------------------- /benchmarks/lstm/lstm_enzyme.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/lstm/lstm_enzyme.c -------------------------------------------------------------------------------- /benchmarks/lstm/main.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/lstm/main.c -------------------------------------------------------------------------------- /benchmarks/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/main.py -------------------------------------------------------------------------------- /benchmarks/main_term/build.ronin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/main_term/build.ronin -------------------------------------------------------------------------------- /benchmarks/main_term/enzyme_kernel.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/main_term/enzyme_kernel.mlir -------------------------------------------------------------------------------- /benchmarks/main_term/main.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/main_term/main.c -------------------------------------------------------------------------------- /benchmarks/main_term/main_term.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/main_term/main_term.h -------------------------------------------------------------------------------- /benchmarks/main_term/main_term_enzyme.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/main_term/main_term_enzyme.c -------------------------------------------------------------------------------- /benchmarks/main_term/main_term_kernel.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/main_term/main_term_kernel.mlir -------------------------------------------------------------------------------- /benchmarks/matmul/blas_wrapper.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/matmul/blas_wrapper.c -------------------------------------------------------------------------------- /benchmarks/matmul/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/matmul/build.py -------------------------------------------------------------------------------- /benchmarks/matmul/enzyme_kernel.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/matmul/enzyme_kernel.mlir -------------------------------------------------------------------------------- /benchmarks/matmul/main.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/matmul/main.c -------------------------------------------------------------------------------- /benchmarks/matmul/matmul_enzyme.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/matmul/matmul_enzyme.c -------------------------------------------------------------------------------- /benchmarks/matmul/matmul_kernel.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/matmul/matmul_kernel.mlir -------------------------------------------------------------------------------- /benchmarks/microbenchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/microbenchmarks.py -------------------------------------------------------------------------------- /benchmarks/mlir/templates/ba.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/ba.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/ba_enzyme_reproj.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/ba_enzyme_reproj.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/ba_enzyme_w.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/ba_enzyme_w.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/cache_loop.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/cache_loop.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/cache_loop_enzyme.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/cache_loop_enzyme.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/dot.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/dot.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_adbench_enzyme.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_adbench_enzyme.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_adbench_lagrad.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_adbench_lagrad.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_adbench_lagrad_buf.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_adbench_lagrad_buf.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_buf.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_buf.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_buf_compressed.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_buf_compressed.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_buf_optimized.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_buf_optimized.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_compressed.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_compressed.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_differentiated.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_differentiated.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_enzyme.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_enzyme.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_tensor_bufferized_opt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_tensor_bufferized_opt.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_tensor_compressed.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_tensor_compressed.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/gmm_tri.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/gmm_tri.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/hand.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/hand.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/hand_complicated_enzyme.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/hand_complicated_enzyme.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/hand_enzyme.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/hand_enzyme.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/hand_inlined.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/hand_inlined.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/lstm.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/lstm.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/lstm_enzyme.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/lstm_enzyme.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/lstm_enzyme_samemem.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/lstm_enzyme_samemem.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/lstm_hand_bufferized.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/lstm_hand_bufferized.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/main_term.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/main_term.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/main_term_bufferized.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/main_term_bufferized.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/matmul.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/matvec.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/matvec.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/microbenchmark.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/microbenchmark.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/microbenchmarks/dot.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/microbenchmarks/dot.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/microbenchmarks/main_term.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/microbenchmarks/main_term.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/microbenchmarks/matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/microbenchmarks/matmul.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/microbenchmarks/vecadd.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/microbenchmarks/vecadd.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/microbenchmarks/vecscal.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/microbenchmarks/vecscal.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/nested_loop_enzyme.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/nested_loop_enzyme.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/nested_loop_lagrad.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/nested_loop_lagrad.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/trimatvec.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/trimatvec.mlir -------------------------------------------------------------------------------- /benchmarks/mlir/templates/vecmat.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/mlir/templates/vecmat.mlir -------------------------------------------------------------------------------- /benchmarks/nn/DELETEME_postlibrary.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/nn/DELETEME_postlibrary.mlir -------------------------------------------------------------------------------- /benchmarks/nn/blas_wrapper.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/nn/blas_wrapper.c -------------------------------------------------------------------------------- /benchmarks/nn/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/nn/build.py -------------------------------------------------------------------------------- /benchmarks/nn/enzyme_mlir_nn.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/nn/enzyme_mlir_nn.mlir -------------------------------------------------------------------------------- /benchmarks/nn/enzyme_nn.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/nn/enzyme_nn.c -------------------------------------------------------------------------------- /benchmarks/nn/lagrad_nn.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/nn/lagrad_nn.mlir -------------------------------------------------------------------------------- /benchmarks/nn/main.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/nn/main.c -------------------------------------------------------------------------------- /benchmarks/nn/nn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/nn/nn.h -------------------------------------------------------------------------------- /benchmarks/nn/read_data.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/nn/read_data.c -------------------------------------------------------------------------------- /benchmarks/pytorch_benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/pytorch_benchmarks.py -------------------------------------------------------------------------------- /benchmarks/results/basparsemat.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/results/basparsemat.txt -------------------------------------------------------------------------------- /benchmarks/results/dot_first_arg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/results/dot_first_arg.json -------------------------------------------------------------------------------- /benchmarks/results/gmm_1000_200_128.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/results/gmm_1000_200_128.json -------------------------------------------------------------------------------- /benchmarks/results/gmm_1000_25_10.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/results/gmm_1000_25_10.json -------------------------------------------------------------------------------- /benchmarks/results/matmul_first_arg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/results/matmul_first_arg.json -------------------------------------------------------------------------------- /benchmarks/results/matvec_first_arg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/results/matvec_first_arg.json -------------------------------------------------------------------------------- /benchmarks/results/trimicrobench.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/results/trimicrobench.csv -------------------------------------------------------------------------------- /benchmarks/results/trmv_1024.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/results/trmv_1024.json -------------------------------------------------------------------------------- /benchmarks/results/trmv_2048.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/results/trmv_2048.json -------------------------------------------------------------------------------- /benchmarks/results/vecmat_first_arg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/results/vecmat_first_arg.json -------------------------------------------------------------------------------- /benchmarks/run_ba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/run_ba.py -------------------------------------------------------------------------------- /benchmarks/run_gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/run_gmm.py -------------------------------------------------------------------------------- /benchmarks/run_hand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/run_hand.py -------------------------------------------------------------------------------- /benchmarks/run_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/run_lstm.py -------------------------------------------------------------------------------- /benchmarks/run_with_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/run_with_memory.py -------------------------------------------------------------------------------- /benchmarks/speedups.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/speedups.py -------------------------------------------------------------------------------- /benchmarks/speedups_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/speedups_data.py -------------------------------------------------------------------------------- /benchmarks/take_means.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/take_means.py -------------------------------------------------------------------------------- /benchmarks/trmv/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/trmv/build.py -------------------------------------------------------------------------------- /benchmarks/trmv/enzyme_trmv.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/trmv/enzyme_trmv.mlir -------------------------------------------------------------------------------- /benchmarks/trmv/main.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/trmv/main.c -------------------------------------------------------------------------------- /benchmarks/trmv/run_with_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/trmv/run_with_datasets.py -------------------------------------------------------------------------------- /benchmarks/trmv/trmv.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/trmv/trmv.h -------------------------------------------------------------------------------- /benchmarks/trmv/trmv.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/trmv/trmv.mlir -------------------------------------------------------------------------------- /benchmarks/trmv/trmv_c_enzyme.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/trmv/trmv_c_enzyme.c -------------------------------------------------------------------------------- /benchmarks/trmv/utils.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/benchmarks/trmv/utils.c -------------------------------------------------------------------------------- /include/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | add_subdirectory(LAGrad) 2 | -------------------------------------------------------------------------------- /include/LAGrad/Analysis.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/include/LAGrad/Analysis.h -------------------------------------------------------------------------------- /include/LAGrad/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/include/LAGrad/CMakeLists.txt -------------------------------------------------------------------------------- /include/LAGrad/LAGradDialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/include/LAGrad/LAGradDialect.h -------------------------------------------------------------------------------- /include/LAGrad/LAGradDialect.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/include/LAGrad/LAGradDialect.td -------------------------------------------------------------------------------- /include/LAGrad/LAGradOps.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/include/LAGrad/LAGradOps.h -------------------------------------------------------------------------------- /include/LAGrad/LAGradOps.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/include/LAGrad/LAGradOps.td -------------------------------------------------------------------------------- /include/LAGrad/Logger.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/include/LAGrad/Logger.h -------------------------------------------------------------------------------- /include/LAGrad/Passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/include/LAGrad/Passes.h -------------------------------------------------------------------------------- /include/LAGrad/Transforms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/include/LAGrad/Transforms.h -------------------------------------------------------------------------------- /include/LAGrad/Utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/include/LAGrad/Utils.h -------------------------------------------------------------------------------- /lagrad-opt/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lagrad-opt/CMakeLists.txt -------------------------------------------------------------------------------- /lagrad-opt/lagrad-opt.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lagrad-opt/lagrad-opt.cpp -------------------------------------------------------------------------------- /lib/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/CMakeLists.txt -------------------------------------------------------------------------------- /lib/LAGrad/ActivityAnalysis.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/ActivityAnalysis.cpp -------------------------------------------------------------------------------- /lib/LAGrad/Bufferize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/Bufferize.cpp -------------------------------------------------------------------------------- /lib/LAGrad/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/CMakeLists.txt -------------------------------------------------------------------------------- /lib/LAGrad/DeadCodeElimination.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/DeadCodeElimination.cpp -------------------------------------------------------------------------------- /lib/LAGrad/DebugNames.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/DebugNames.cpp -------------------------------------------------------------------------------- /lib/LAGrad/DynamicShapeAnalysis.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/DynamicShapeAnalysis.cpp -------------------------------------------------------------------------------- /lib/LAGrad/ElementwiseToAffine.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/ElementwiseToAffine.cpp -------------------------------------------------------------------------------- /lib/LAGrad/InsertExtractAnalysis.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/InsertExtractAnalysis.cpp -------------------------------------------------------------------------------- /lib/LAGrad/LAGradDialect.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/LAGradDialect.cpp -------------------------------------------------------------------------------- /lib/LAGrad/LAGradOps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/LAGradOps.cpp -------------------------------------------------------------------------------- /lib/LAGrad/LAGradTransforms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/LAGradTransforms.cpp -------------------------------------------------------------------------------- /lib/LAGrad/LinalgCanonicalize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/LinalgCanonicalize.cpp -------------------------------------------------------------------------------- /lib/LAGrad/LinalgKnownLibraryCalls.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/LinalgKnownLibraryCalls.cpp -------------------------------------------------------------------------------- /lib/LAGrad/LinalgVectorJacobianProducts.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/LinalgVectorJacobianProducts.cpp -------------------------------------------------------------------------------- /lib/LAGrad/LoopAllocHoisting.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/LoopAllocHoisting.cpp -------------------------------------------------------------------------------- /lib/LAGrad/LoopNestAnalysis.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/LoopNestAnalysis.cpp -------------------------------------------------------------------------------- /lib/LAGrad/LowerGrad.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/LowerGrad.cpp -------------------------------------------------------------------------------- /lib/LAGrad/LowerToLLVM.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/LowerToLLVM.cpp -------------------------------------------------------------------------------- /lib/LAGrad/PrimalCaching.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/PrimalCaching.cpp -------------------------------------------------------------------------------- /lib/LAGrad/SCFVectorJacobianProducts.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/SCFVectorJacobianProducts.cpp -------------------------------------------------------------------------------- /lib/LAGrad/SparsePropagation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/SparsePropagation.cpp -------------------------------------------------------------------------------- /lib/LAGrad/StaticAllocs.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/StaticAllocs.cpp -------------------------------------------------------------------------------- /lib/LAGrad/TensorConstantFolding.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/TensorConstantFolding.cpp -------------------------------------------------------------------------------- /lib/LAGrad/TriangularLoops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/TriangularLoops.cpp -------------------------------------------------------------------------------- /lib/LAGrad/TriangularPacking.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/TriangularPacking.cpp -------------------------------------------------------------------------------- /lib/LAGrad/Utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LAGrad/Utils.cpp -------------------------------------------------------------------------------- /lib/LLFastMath/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LLFastMath/CMakeLists.txt -------------------------------------------------------------------------------- /lib/LLFastMath/LLFastMath.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/lib/LLFastMath/LLFastMath.cpp -------------------------------------------------------------------------------- /memory_monitor/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/memory_monitor/Makefile -------------------------------------------------------------------------------- /memory_monitor/memory_monitor.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/memory_monitor/memory_monitor.c -------------------------------------------------------------------------------- /mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/mlir -------------------------------------------------------------------------------- /ronin_phases/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ronin_phases/build_llvm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/ronin_phases/build_llvm.py -------------------------------------------------------------------------------- /ronin_phases/memory_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/ronin_phases/memory_monitor.py -------------------------------------------------------------------------------- /ronin_phases/render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/ronin_phases/render.py -------------------------------------------------------------------------------- /ronin_phases/replace_hand_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/ronin_phases/replace_hand_opt.py -------------------------------------------------------------------------------- /scripts/bufferize: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/scripts/bufferize -------------------------------------------------------------------------------- /scripts/lower_to_llvm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/scripts/lower_to_llvm -------------------------------------------------------------------------------- /scripts/run_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/scripts/run_benchmark.sh -------------------------------------------------------------------------------- /scripts/run_external_call.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/scripts/run_external_call.sh -------------------------------------------------------------------------------- /scripts/run_matmul.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/scripts/run_matmul.sh -------------------------------------------------------------------------------- /scripts/run_matvec.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/scripts/run_matvec.sh -------------------------------------------------------------------------------- /scripts/run_reference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/scripts/run_reference.sh -------------------------------------------------------------------------------- /scripts/run_vecmat.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/scripts/run_vecmat.sh -------------------------------------------------------------------------------- /test/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/CMakeLists.txt -------------------------------------------------------------------------------- /test/Driver/dot.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/Driver/dot.c -------------------------------------------------------------------------------- /test/Driver/dot_reference.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/Driver/dot_reference.c -------------------------------------------------------------------------------- /test/Driver/matmul.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/Driver/matmul.c -------------------------------------------------------------------------------- /test/Driver/matvec.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/Driver/matvec.c -------------------------------------------------------------------------------- /test/Driver/simple.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/Driver/simple.c -------------------------------------------------------------------------------- /test/Driver/standaloneabi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/Driver/standaloneabi.h -------------------------------------------------------------------------------- /test/Driver/vecmat.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/Driver/vecmat.c -------------------------------------------------------------------------------- /test/LAGrad/activity/ifelse.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/activity/ifelse.mlir -------------------------------------------------------------------------------- /test/LAGrad/activity/unused.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/activity/unused.mlir -------------------------------------------------------------------------------- /test/LAGrad/add3.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/add3.mlir -------------------------------------------------------------------------------- /test/LAGrad/arith/negf.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/arith/negf.mlir -------------------------------------------------------------------------------- /test/LAGrad/arith/tanh.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/arith/tanh.mlir -------------------------------------------------------------------------------- /test/LAGrad/arithops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/arithops.mlir -------------------------------------------------------------------------------- /test/LAGrad/arrsum.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/arrsum.mlir -------------------------------------------------------------------------------- /test/LAGrad/ba/ba.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/ba/ba.mlir -------------------------------------------------------------------------------- /test/LAGrad/batch_matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/batch_matmul.mlir -------------------------------------------------------------------------------- /test/LAGrad/before.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/before.mlir -------------------------------------------------------------------------------- /test/LAGrad/bufferize/collapse_shape.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/bufferize/collapse_shape.mlir -------------------------------------------------------------------------------- /test/LAGrad/bufferize/collapse_shape_2d.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/bufferize/collapse_shape_2d.mlir -------------------------------------------------------------------------------- /test/LAGrad/bufferize/collapse_shape_col.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/bufferize/collapse_shape_col.mlir -------------------------------------------------------------------------------- /test/LAGrad/bufferize/grad_collapse_shape.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/bufferize/grad_collapse_shape.mlir -------------------------------------------------------------------------------- /test/LAGrad/bufferize/grad_collapse_shape_2d.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/bufferize/grad_collapse_shape_2d.mlir -------------------------------------------------------------------------------- /test/LAGrad/bufferize/grad_collapse_shape_col.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/bufferize/grad_collapse_shape_col.mlir -------------------------------------------------------------------------------- /test/LAGrad/cache/nested_pow.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/cache/nested_pow.mlir -------------------------------------------------------------------------------- /test/LAGrad/cache/nested_pow_tensor.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/cache/nested_pow_tensor.mlir -------------------------------------------------------------------------------- /test/LAGrad/cache/pow.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/cache/pow.mlir -------------------------------------------------------------------------------- /test/LAGrad/cache/pow_tensor.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/cache/pow_tensor.mlir -------------------------------------------------------------------------------- /test/LAGrad/cross_product.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/cross_product.mlir -------------------------------------------------------------------------------- /test/LAGrad/custom_grad_init.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/custom_grad_init.mlir -------------------------------------------------------------------------------- /test/LAGrad/diff/const_memref.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/diff/const_memref.mlir -------------------------------------------------------------------------------- /test/LAGrad/diff/dot.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/diff/dot.mlir -------------------------------------------------------------------------------- /test/LAGrad/diff/generic.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/diff/generic.mlir -------------------------------------------------------------------------------- /test/LAGrad/diff/matvec_buffer.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/diff/matvec_buffer.mlir -------------------------------------------------------------------------------- /test/LAGrad/diff/memref.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/diff/memref.mlir -------------------------------------------------------------------------------- /test/LAGrad/diff/two_funcs.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/diff/two_funcs.mlir -------------------------------------------------------------------------------- /test/LAGrad/dot.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/dot.mlir -------------------------------------------------------------------------------- /test/LAGrad/dot_second_arg.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/dot_second_arg.mlir -------------------------------------------------------------------------------- /test/LAGrad/dummy.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/dummy.mlir -------------------------------------------------------------------------------- /test/LAGrad/functioncall.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/functioncall.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/broadcast.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/broadcast.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/broadcast_square.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/broadcast_square.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/closure.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/closure.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/colsum.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/colsum.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/dot.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/dot.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/functioncall.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/functioncall.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/generic_with_call.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/generic_with_call.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/gmm_einsum.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/gmm_einsum.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/logsumexp.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/logsumexp.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/logsumexp_1d.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/logsumexp_1d.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/logsumexp_add2.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/logsumexp_add2.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/nonconst_out.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/nonconst_out.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/sub.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/sub.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/three_args.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/three_args.mlir -------------------------------------------------------------------------------- /test/LAGrad/generic/vecmat.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/generic/vecmat.mlir -------------------------------------------------------------------------------- /test/LAGrad/gmm/compressed_outer_product.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/gmm/compressed_outer_product.mlir -------------------------------------------------------------------------------- /test/LAGrad/gmm/compressed_qtimesx.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/gmm/compressed_qtimesx.mlir -------------------------------------------------------------------------------- /test/LAGrad/gmm/compressed_vecmat.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/gmm/compressed_vecmat.mlir -------------------------------------------------------------------------------- /test/LAGrad/gmm/gmm_objective_full.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/gmm/gmm_objective_full.mlir -------------------------------------------------------------------------------- /test/LAGrad/gmm/main_term.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/gmm/main_term.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/angle_axis_to_rotation_matrix.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/angle_axis_to_rotation_matrix.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/apply_global_transform.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/apply_global_transform.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/compute_err_simple.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/compute_err_simple.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/get_posed_relatives.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/get_posed_relatives.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/get_posed_relatives_body.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/get_posed_relatives_body.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/hand_objective.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/hand_objective.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/hand_objective_complicated.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/hand_objective_complicated.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/relatives_to_absolutes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/relatives_to_absolutes.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/relatives_to_absolutes_scratch.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/relatives_to_absolutes_scratch.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/skinned_vertex_subset.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/skinned_vertex_subset.mlir -------------------------------------------------------------------------------- /test/LAGrad/hand/to_pose_params.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/hand/to_pose_params.mlir -------------------------------------------------------------------------------- /test/LAGrad/if_function.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/if_function.mlir -------------------------------------------------------------------------------- /test/LAGrad/ifdifferenttype.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/ifdifferenttype.mlir -------------------------------------------------------------------------------- /test/LAGrad/ifelse.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/ifelse.mlir -------------------------------------------------------------------------------- /test/LAGrad/insert_multi.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/insert_multi.mlir -------------------------------------------------------------------------------- /test/LAGrad/kernels/dot.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/kernels/dot.mlir -------------------------------------------------------------------------------- /test/LAGrad/kernels/enzyme_dot.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/kernels/enzyme_dot.mlir -------------------------------------------------------------------------------- /test/LAGrad/kernels/matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/kernels/matmul.mlir -------------------------------------------------------------------------------- /test/LAGrad/kernels/matvec.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/kernels/matvec.mlir -------------------------------------------------------------------------------- /test/LAGrad/kernels/vecmat.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/kernels/vecmat.mlir -------------------------------------------------------------------------------- /test/LAGrad/kernels/vecmat_kernel.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/kernels/vecmat_kernel.c -------------------------------------------------------------------------------- /test/LAGrad/lstm/lstm_model.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/lstm/lstm_model.mlir -------------------------------------------------------------------------------- /test/LAGrad/lstm/lstm_objective.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/lstm/lstm_objective.mlir -------------------------------------------------------------------------------- /test/LAGrad/lstm/lstm_predict.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/lstm/lstm_predict.mlir -------------------------------------------------------------------------------- /test/LAGrad/matsum.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/matsum.mlir -------------------------------------------------------------------------------- /test/LAGrad/matvec.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/matvec.mlir -------------------------------------------------------------------------------- /test/LAGrad/nestedcall.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/nestedcall.mlir -------------------------------------------------------------------------------- /test/LAGrad/nn/batched_crossentropy.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/nn/batched_crossentropy.mlir -------------------------------------------------------------------------------- /test/LAGrad/nn/crossentropy.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/nn/crossentropy.mlir -------------------------------------------------------------------------------- /test/LAGrad/nn/nn.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/nn/nn.mlir -------------------------------------------------------------------------------- /test/LAGrad/quadratic.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/quadratic.mlir -------------------------------------------------------------------------------- /test/LAGrad/relu.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/relu.mlir -------------------------------------------------------------------------------- /test/LAGrad/scf/arrmax.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/scf/arrmax.mlir -------------------------------------------------------------------------------- /test/LAGrad/scf/forintermediate.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/scf/forintermediate.mlir -------------------------------------------------------------------------------- /test/LAGrad/scf/fortensor.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/scf/fortensor.mlir -------------------------------------------------------------------------------- /test/LAGrad/scf/if_funcall.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/scf/if_funcall.mlir -------------------------------------------------------------------------------- /test/LAGrad/scf/nested_with_slice.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/scf/nested_with_slice.mlir -------------------------------------------------------------------------------- /test/LAGrad/scf/product.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/scf/product.mlir -------------------------------------------------------------------------------- /test/LAGrad/scf/selectdifferenttype.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/scf/selectdifferenttype.mlir -------------------------------------------------------------------------------- /test/LAGrad/scf/wrt_iter_arg.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/scf/wrt_iter_arg.mlir -------------------------------------------------------------------------------- /test/LAGrad/scfdot.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/scfdot.mlir -------------------------------------------------------------------------------- /test/LAGrad/sparse/adjoint_err_nest.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/sparse/adjoint_err_nest.mlir -------------------------------------------------------------------------------- /test/LAGrad/sparse/colhot_broadcast_mul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/sparse/colhot_broadcast_mul.mlir -------------------------------------------------------------------------------- /test/LAGrad/sparse/onehot_elementwise_square.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/sparse/onehot_elementwise_square.mlir -------------------------------------------------------------------------------- /test/LAGrad/sparse/onehot_matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/sparse/onehot_matmul.mlir -------------------------------------------------------------------------------- /test/LAGrad/sparse/onehot_sumreduce.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/sparse/onehot_sumreduce.mlir -------------------------------------------------------------------------------- /test/LAGrad/sparse/rowhot_broadcast_mul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/sparse/rowhot_broadcast_mul.mlir -------------------------------------------------------------------------------- /test/LAGrad/sparse/rowhot_matmul.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/sparse/rowhot_matmul.mlir -------------------------------------------------------------------------------- /test/LAGrad/sparse/rowhot_tensor_insert.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/sparse/rowhot_tensor_insert.mlir -------------------------------------------------------------------------------- /test/LAGrad/standalone-opt.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/standalone-opt.mlir -------------------------------------------------------------------------------- /test/LAGrad/standalone-translate.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/standalone-translate.mlir -------------------------------------------------------------------------------- /test/LAGrad/tensor/insert_overwrite.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/tensor/insert_overwrite.mlir -------------------------------------------------------------------------------- /test/LAGrad/tensor/insert_overwrite_scalar.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/tensor/insert_overwrite_scalar.mlir -------------------------------------------------------------------------------- /test/LAGrad/tensor/insert_slice.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/tensor/insert_slice.mlir -------------------------------------------------------------------------------- /test/LAGrad/tensor/insert_slice_loop.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/tensor/insert_slice_loop.mlir -------------------------------------------------------------------------------- /test/LAGrad/tensor_extract.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/tensor_extract.mlir -------------------------------------------------------------------------------- /test/LAGrad/tensor_extract_slice.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/tensor_extract_slice.mlir -------------------------------------------------------------------------------- /test/LAGrad/twoargs.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/LAGrad/twoargs.mlir -------------------------------------------------------------------------------- /test/lit.cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/lit.cfg.py -------------------------------------------------------------------------------- /test/lit.site.cfg.py.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/lit.site.cfg.py.in -------------------------------------------------------------------------------- /test/pytest/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/Makefile -------------------------------------------------------------------------------- /test/pytest/benchmark_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/benchmark_io.py -------------------------------------------------------------------------------- /test/pytest/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/build.py -------------------------------------------------------------------------------- /test/pytest/mlir_bindings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/mlir_bindings.py -------------------------------------------------------------------------------- /test/pytest/pytorch_ref/BASparseMat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/pytorch_ref/BASparseMat.py -------------------------------------------------------------------------------- /test/pytest/pytorch_ref/pytorch_ba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/pytorch_ref/pytorch_ba.py -------------------------------------------------------------------------------- /test/pytest/pytorch_ref/pytorch_gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/pytorch_ref/pytorch_gmm.py -------------------------------------------------------------------------------- /test/pytest/pytorch_ref/pytorch_hand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/pytorch_ref/pytorch_hand.py -------------------------------------------------------------------------------- /test/pytest/pytorch_ref/pytorch_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/pytorch_ref/pytorch_lstm.py -------------------------------------------------------------------------------- /test/pytest/pytorch_ref/pytorch_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/pytorch_ref/pytorch_nn.py -------------------------------------------------------------------------------- /test/pytest/pytorch_ref/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/pytorch_ref/utils.py -------------------------------------------------------------------------------- /test/pytest/stdout_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/stdout_parser.py -------------------------------------------------------------------------------- /test/pytest/test_activity_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_activity_analysis.py -------------------------------------------------------------------------------- /test/pytest/test_ba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_ba.py -------------------------------------------------------------------------------- /test/pytest/test_bufferization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_bufferization.py -------------------------------------------------------------------------------- /test/pytest/test_enzyme.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_enzyme.py -------------------------------------------------------------------------------- /test/pytest/test_gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_gmm.py -------------------------------------------------------------------------------- /test/pytest/test_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_grad.py -------------------------------------------------------------------------------- /test/pytest/test_grad_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_grad_v2.py -------------------------------------------------------------------------------- /test/pytest/test_hand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_hand.py -------------------------------------------------------------------------------- /test/pytest/test_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_lstm.py -------------------------------------------------------------------------------- /test/pytest/test_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_nn.py -------------------------------------------------------------------------------- /test/pytest/test_scf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_scf.py -------------------------------------------------------------------------------- /test/pytest/test_structured_sparsify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_structured_sparsify.py -------------------------------------------------------------------------------- /test/pytest/test_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/test_tensor.py -------------------------------------------------------------------------------- /test/pytest/toolchain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pengmai/lagrad/HEAD/test/pytest/toolchain.py --------------------------------------------------------------------------------