InfiniTensor/include/core
zhengly123 a1974aabcd
NNET supports TVM backend and kernels (#78)
* Add: mutator InfoGAN minimum test

* Add: cache and padding (bugs!!)

* Add: expression reader as a cmake target

* Fix: [Intermediate] NMutator::expressionToGraph

To be fix: matmul with implicit broadcast

* Add: matmul broadcast

* Fix: GraphObj ctor should use cloneTensor

* Fix: cuBLAS failure when codegen is enabled

* Add: Exception for checkCuError

* Fix: graph OpList ctor

* Add: expr simplication for TVM

* Add: TVM headers and CMake include paths

* Add: CMake config

* Add: PackedFunc (broken)

* Fix: remove cuCtxCreate which makes TVM fails

* Fix: membound_tvm

* Fix: test_memboundOp

* Add: PRelu Expr and AsTVMVisitor

* Add: Random generator

* Add: support TVM packed function

* Fix: specify runtime

* Add: CMake support of TVM

* Add: detailed output of Matmul

* Add: comments for Matmul

* Chore: format and comments

* Chore: GraphObj::selfCheck without assert control

* Fix: CMAKE_CXX_FLAGS in CMakeLists

* fix merge bug

* update api for mkl batchnorm test

* fix lotus env

* fig header bug

---------

Co-authored-by: Liyan Zheng <liyan-zheng@outlook.com>
Co-authored-by: huangshuhong <huangsh19@mails.tsinghua.edu.cn>
Co-authored-by: whjthu <haojie0429@gmail.com>
2023-04-18 00:26:36 +08:00
..
blob.h Support bang c kernel wanghailu 0927 (#43) 2022-09-30 11:01:52 +08:00
common.h Add ConvTransposedNHWC (#67) 2023-03-01 14:15:02 +08:00
constants.h Add activation operators and kernels 2022-09-16 13:58:57 +08:00
data_type.h style: remove non-ascii comments from cpp 2023-02-16 14:57:51 +08:00
dummy_mutator.h Add search engine (#64) 2023-02-12 18:27:52 +08:00
graph.h NNET supports TVM backend and kernels (#78) 2023-04-18 00:26:36 +08:00
graph_handler.h Cpu backend2 (#77) 2023-04-17 12:15:23 +08:00
graph_match.h ADD: sub graph replacement. (#56) 2023-04-17 13:09:07 +08:00
hash.h Tensor hash and inferShape (#4) 2022-08-15 15:08:56 +08:00
kernel.h Add search engine (#64) 2023-02-12 18:27:52 +08:00
mutator.h ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00
object.h Add search engine (#64) 2023-02-12 18:27:52 +08:00
operator.h ADD: sub graph replacement. (#56) 2023-04-17 13:09:07 +08:00
perf_engine.h Json perfrecord (#32) 2022-09-22 15:34:34 +08:00
ref.h Add: connection among tensors and operators (#45) 2022-10-18 22:02:51 +08:00
runtime.h ADD: sub graph replacement. (#56) 2023-04-17 13:09:07 +08:00
search_engine.h Add search engine (#64) 2023-02-12 18:27:52 +08:00
tensor.h NNET supports TVM backend and kernels (#78) 2023-04-18 00:26:36 +08:00
tensor_base.h ADD: sub graph replacement. (#56) 2023-04-17 13:09:07 +08:00