InfiniTensor/test/nnet
deathwings602 11d5aa1ccc
Add TVM codegen for MemboundOp (#35)
* Add:  interface for membound TVM kernel and test

* add getAnsorCode

* add evaluation, but link failed

* add evaluation of kernel, but link failed

* Fix: link libcuda and nvrtc

* add print

* Add: const for source of copy

* compile and evaluate the kernel

* add compute

* fix gen_ansor_op.py

* fix membound_TVM

* format and fix CMakeLists.txt

* fix memory leak

Co-authored-by: Liyan Zheng <liyan-zheng@outlook.com>
Co-authored-by: huangshuhong <huangsh19@mails.tsinghua.edu.cn>
2022-09-22 18:06:45 +08:00
..
failed Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
log Add: nnet code 2022-08-08 16:02:07 +08:00
readlog.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_Interpreter.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_OpSearch.cc Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
test_Rule2VariableMerging.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_TConv2gemm.cc Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
test_as_tvm.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_compareFormulas.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_conv2conv.cc Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
test_conv2gemm.cc Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
test_dlt.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_exprHash.cc Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
test_g2bmm.cc Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
test_guidedDLT.cc Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
test_matchConv.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_matchElementWise.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_matchMatmul.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_matchReshape.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_memboundOp.cc Add TVM codegen for MemboundOp (#35) 2022-09-22 18:06:45 +08:00
test_mergeStage.cc Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
test_oobChecker.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_rangeMagnify.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_relaxation.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_serializer.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_simplify.cc Add: nnet code 2022-08-08 16:02:07 +08:00
test_subset.cc Add: nnet code 2022-08-08 16:02:07 +08:00