Commit Graph

3249 Commits

Author SHA1 Message Date
Margaret Qian efe662fe66 [Relay][Pass] Add a relay pass to extract fake quantized ops (#10089)
* add relay pass to collect fake quantized ops

* add more tests

* more tests

* lint

* lint

* remove unused imports

* update comment

* lint

* reuse SubgraphExtractor and update test assertions

* remove print

* lint

* remove unneeded comment

Co-authored-by: Margaret Qian <mqian@octoml.ai>
2022-02-01 20:57:45 -08:00
Josh Fromm ac4815c0d0 [AutoScheduler] Allow device specification for AutoScheduler Runners. (#10123)
* Changed the python api to support device.

* Finished implementation and updated tests.

* Fix typo.
2022-02-02 08:09:27 +09:00
Tristan Konolige 780f88a425 [FIX,AUTOTVM] Add backtraces to tuning errors (#9901)
* [FIX,AUTOTVM] Add backtraces to tuning errors

Collects tracebacks in LocalBuilder and LocalRunner and adds them to the
error messages.

* formatting

* correctly unpack traceback and exception

* add assert

* fix?

* one remaining measureresult

* formatting

* fixed
2022-02-01 14:20:42 -08:00
Masahiro Masuda a1f51aa230 [CUTLASS] Conv2d dgrad (#10110)
* add conv2d transpose nhwc cudnn test

* support conv2d transpose nhwc direct offload to cudnn

* add cutlass dgrad support

* remove unused arg

* allow target none

* fix beta initiaization condition

* disable dynamic dense fp16 test since it fails on cuda 11.6
2022-02-02 05:51:57 +09:00
Matthew Barrett 339f8886f4 [USMP] Add performance characteristics to PoolInfo (#10005)
* [USMP] Add performance characteristics to PoolInfo

Scheduling algorithms that wish to optimize around
memory pools require further information about the
perfomance characteristics of those pools. This
commit adds clock frequency, bandwidth, latency and
burst length as optional fields to PoolInfo.

Change-Id: I4cf3f35324d093fb38e874f0f2e587cb84d4ba1e

* Remove unused import

Change-Id: I1e2ef885425f4361b80c2bab9261ec129e61a756
2022-02-01 18:12:46 +00:00
Tristan Konolige 187aeb5fe8 [AUTOTVM] Use opt level 3 when extracting tasks (#10065)
* [AUTOTVM] Use opt level 3 when extracting tasks

Autotvm was implicitly ignoring opt_level when extracting tasks because
pass opt_level is a thread local variable and extraction happens in a
new thread. Not having opt_level 3 causes alter op layout to not
fire, which in turn prevents tuning from finding all possible kernels.

* disable alter op layout
2022-02-01 12:26:46 +09:00
Tristan Konolige 24d2a38116 [LLVM,TIR] Print LLVM intrinsic names instead of ids (#9964)
* [LLVM,TIR] Print LLVM intrinsic names instead of ids

This makes it much easy to understand what is happening with llvm
intrinsics.

* add test, version llvm
2022-02-01 09:16:59 +09:00
Huang, Guangtai dad8f62fc1 [Bugfix][Op] Fix shape inference of adv_index (#9717)
* init

* test

* lint
2022-01-31 09:53:45 -08:00
lhutton1 02a7a4182f [microNPU] Add support for nearest neighbor and bilinear upsampling (#9841)
* [microNPU] Add support for nearest neighbor and bilinear upsampling

Adds support for 2x2 nearest neighbor and bilinear upsampling. In the
case of bilinear upsampling with align_corners set to true, the
upsampling size must be `2*input_size - 1` (as opposed to `2*input_size`).

Change-Id: I95d215eabfaac983629dcdedcda2b90efb8e0ddf

* rebase and add support for no-upsampling case.

Change-Id: I840d8ee3671a40c5c99f22119442c349dbed39cf
2022-01-31 16:05:21 +00:00
Leo-arm 3de25b83f9 [ETHOSN] Per-tensor support for int8 operations (#10018)
* Per-axis quantization to follow
2022-01-31 11:38:04 +00:00
Grant Watson 3b20c21f9e [microTVM] Include standalone_crt dependencies in MLF (#10095)
* Adds runtime to AOTExecutorFactoryModule
 * Standalone CRT files are added to MLF tarball if runtime is crt
 * external_dependencies info added to metadata.json for crt runtime
 * microNPU demo Makefile references standalone crt files from MLF tarball
2022-01-31 10:50:29 +00:00
KennyTang1988 f2d60fe968 [Caffe Frontend] Add support for Power layer (#9655)
Co-authored-by: tangkun <kun.tang@hexintek.com>
2022-01-31 15:49:50 +09:00
Tony d8d00530bb [Fix Bug]fix the bugs of keras frontend when parsing LSTM, GRU, RNN layers. (#9850)
* [Fix Bug]fix the bugs of keras frontend when parsing LSTM, GRU, RNN layers.

* Reformat files with black formatter.

Co-authored-by: AndrewZhaoLuo <andrew.zhao.luo@gmail.com>
2022-01-30 21:55:18 -08:00
Xiyou Zhou 779dc51e13 [MetaSchedule][M4a] User-API: Tune-TE/TIR/Relay (#10079)
* Add tuning scripts for tir, te & relay.

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>

Minor fix.

Nits.

Add back tests.

* slightly improve tune.py

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
2022-01-30 13:50:24 +08:00
Siyuan Feng 538347e49f [MetaSchedule] postproc: rewrite_cooperative_fetch (#10081)
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
2022-01-29 16:59:09 -05:00
Sunghyun Park ba651974c8 [MetaSchedule][M4b] Testcases for TensorRT builder/runner (#10055)
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
2022-01-29 13:31:07 -08:00
Ruihang Lai 4d0dac3e55 [MetaSchedule][M4a] Mutator: Mutate-Tile-Size (#10092)
* [MetaSchedule][M4a] Mutator: Mutate-Tile-Size

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

* Python 3.8 has no `math.prod`

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
2022-01-29 12:24:03 +08:00
Mehrdad Hessar 6a274af9cb [Hexagon] Update hexagon API build instruction and cleanup hexagon_proxy_rpc (#10068)
* Fix hexagon api build and Update Readme

* Cleanup hexagon_proxy_rpc

* Target Hack

* Remove hack

* address @cconvey comments

* remove the rest of proxy rpc
2022-01-28 16:55:22 -06:00
Hongyi Jin 85d42f894e [MetaSchedule][M4a] Mutator: Mutate Parallel (#10096) 2022-01-28 15:52:21 -05:00
Hua Jiang 80d4d05e83 [Runtime][PipelineExecutor] Pipeline Executor Sequential execution (#10082)
* [Runtime][PipelineExecutor] Pipeline Executor Sequential execution

In the first, adding the "get output" logic. Secondly, adding the the sequential executing
logic of pipeline executor. In the last, testing the pipeline executor interface and
checking the output data.

* Address review comments.

Co-authored-by: Cody Yu <comaniac0422@gmail.com>

* trigger build.

Co-authored-by: Cody Yu <comaniac0422@gmail.com>
2022-01-28 12:36:31 -08:00
Manupa Karunaratne 7b9fd1e2ab [microNPU] Removing constant args from PrimFunc (#9951)
Before this commit, microNPU creates PrimFunc as if
it accepts constants from the callee. This commit
changes the PrimFunc to remove the constants as an
argument to PrimFunc as they are not provided from
the main function.
2022-01-28 15:07:37 +00:00
Cody Yu e6af87491e [PyTorch] Fix rsub type (#10090)
* [PyTorch] Fix rsub type

* fix
2022-01-28 17:10:24 +09:00
lhutton1 fa317edf78 [CI][Fix] Remove additional qnn.op.transpose_conv2d from docs (#10083)
Fixes CI after #10077, and replaces misuse elsewhere.

Change-Id: I095fc8ea2b8d268b09538832cba1f5482a73a9d9
2022-01-27 11:45:05 -08:00
Lily Orth-Smith 248ad45c6a Change function constructors to WithFields (#9690)
* Change function constructors to WithFields

Get rid of std::moves, they were causing problems

* Fix bad rebase

* flaky

* try to trigger ci

* try again
2022-01-26 15:47:45 -08:00
yuanfz 90e454a0a7 [Relay] QLinearMatMul allows 1D weight_scale, weight_zero_point inputs (#10047)
* fix after cr

* fix after cr 2

* emptycommit

* emptycommit 2nd try
2022-01-27 07:02:58 +09:00
Siyuan Feng 1935341af7 [MetaSchedule] postproc: rewrite_parallel_vectorize_unroll (#10071)
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
2022-01-26 14:55:29 -05:00
Chun-I Tsai 095b63935e Revert "[Frontend] Add Span filling for frontends to Relay (#9723)" (#10072)
Because of the failure of LSTM conversion from Pytorch
2022-01-26 14:35:38 +00:00
Hongyi Jin ffbe491b0f [MetaSchedule][M4a] Schedule Rule: Multi-Level-Tiling (#10043)
* multi level tiling

* remove tensor core related code

* pylint

* fix

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
2022-01-26 21:10:51 +08:00
ninesheep 92cd754c45 [Fix Bug]fix the bug of tensorflow frontend when parsing Range layer (#9999)
Co-authored-by: wangjiuyang <wang.jiuyang@intellif.com>
2022-01-26 21:27:30 +09:00
Alexey Voronov ffff8dd300 Add FP requantize flow. Set float32 flow by default for llvm x86 targets with (#9637)
sse4.1 support
2022-01-26 21:22:33 +09:00
Manupa Karunaratne b9728773c1 [microNPU] Refactor base address determination to codegen (#9929)
This commit introduces BaseAddress ObjectRef to determine
base addresses in the codegen for microNPU. This is
required when multiple memory pools become available. Thus,
base addresses could not be statically determined in the
source module.
2022-01-26 12:19:28 +00:00
Wuwei Lin 5e7438feaa [TIR][Schedule] Blockize and Tensorize (#9871)
* WIP

* WIP

* WIP

* test cases

* add examples

* lint

* Amend co-authors information

Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>

* WIP

* address comments and changed tensorized comparator

* update

* nit

* fix example

* lint

* lint

* lint

* remove unused

* trigger ci

* clang-format

* fix

* rebase

Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
2022-01-26 16:00:07 +08:00
Bohan Hou 88cbf1b7a4 [MetaSchedule] Mutator Rule: Mutate Unroll (#10045)
* mutate-unroll

* mutate-unroll
2022-01-25 23:48:38 -08:00
Xiyou Zhou 94c4e0eecf [Meta Schedule] Add ApplyHisotryBest Meta Schedule Context (#10049)
* Add ApplyHisotryBest.

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>

* Retrigger CI.

* Update integration.py

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
2022-01-25 23:45:24 -08:00
Masahiro Masuda 1b9b05e61d [CUTLASS] Profile only the largest-possible alignment by default (#10036)
* introduce profile_all_alignments option

* add profile_all_alignment option to API

* wip

* fixed dynamic case

* black

* update gen_gemm too

* minor improvement

* fix

* all tests work

* add doc

* fixed for sm = 75 case

* fix typo

* remove unused import

* profile_all -> find_first_valid

* fix
2022-01-26 15:42:54 +09:00
Hua Jiang 6720d3593d [Runtime][PipelineExecutor] Add Pipeline Executor Interface (#10010)
Adding interfaces into Pipeline Executor to "run", "stop","set input",
and "get input" from the pipeline executor,

In this patch, we also implemented the "BackendRuntime" structure to
wrap the graph runtime interface in order to support  pipeline executor
interface and implement data copy method. This method is used to
transfer data between two backend runtimes.
2022-01-25 09:40:08 +09:00
Sevin F. Varoglu 6f2b35feb5 [QNN] Add qnn.rsqrt op (#9982)
* Add qnn.rsqrt op

* Add comment
2022-01-24 10:59:56 -08:00
Grant Watson 65b4b09381 [CMSIS-NN] Update microNPU demo to include offloading to CMSIS-NN (#9979)
* [CMSIS-NN] Update microNPU demo to include offloading to CMSIS-NN

Change-Id: I6a3ba9db3e3cb2bd7c10383ebd52f9a1cdad74d0

* [CMSIS-NN] Update microNPU demo to include offloading to CMSIS-NN

* Addressing comments

Change-Id: I98fcdf95bf408700968827e1abd084a916b3b21c

* [CMSIS-NN] Update microNPU demo to include offloading to CMSIS-NN

    * Addressing comments
    * Remove build folder before running demo to address #10020

Change-Id: Ifa7ad3ff431f427f8afb8b3c9f06711b3b59ad62

* Correctly filter tvmc Targets

Fixed logic to check for >2 TVM Target to be based on none-hybrid
Targets only

Co-authored-by: Chris Sidebottom <chris.sidebottom@arm.com>
2022-01-24 18:43:54 +00:00
lhutton1 74a2fa8046 [microNPU] Add support for requantize (#9910)
* [microNPU] Add support for requantize

Adds support for stand-alone requantize operation which is legalized to
an identity operation on the NPU.

Change-Id: Ie2450c5fc72f405eddf517593236074aa4716c3b

* fix concatenate tests failing due to not being bit exact

Since requantize is now offloaded, concatenate tests were failing
due a reference not being used.

Change-Id: I44b26b5daecfefb776ca19e6646f3690f5570f52

* test multiple requantize offload

Change-Id: I60a3283461a7a7083c05289e84f570698388077b

* address comments

Change-Id: I7196a0fa468eb7c6a96f2b8a68f3a2dcf5a5693c
2022-01-24 09:14:05 +09:00
Ruihang Lai de01c3e2a7 [MetaSchedule] Schedule Rule: Parallelize-Vectorize-Unroll (#10033)
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
2022-01-23 08:06:10 -08:00
Ruihang Lai cc67040e62 [MetaSchedule] Post Processor: Rewrite Unbound Block (#10027)
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
2022-01-23 02:48:11 -08:00
Siyuan Feng fc1814e675 [MetaSchedule] Mutator: Mutate compute location (#10028)
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
2022-01-23 02:47:34 -08:00
Masahiro Masuda d35b858ceb [CUDNN] Support gradient kernels (#9986)
* Dgrad nchw, nhwc, fp16 working

commit 426e5dca446a27da49270f45171b58f1bfa21fa9
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Tue Jan 18 11:48:53 2022 +0900

    black

commit 211a58b80f4d0f0b5b0230720e41f35e50cb1eaf
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Tue Jan 18 11:43:52 2022 +0900

    fp16 also works

commit c2a34d473b063873628bff00e51a44cd8e4d0e4f
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Tue Jan 18 11:36:36 2022 +0900

    nhwc test also worked

commit c0609ab147fef30c230a94d16b6c1ba35f7dd9c0
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Tue Jan 18 11:21:23 2022 +0900

    nchw test worked

commit 2bf68c72763708151e9f49f09916a210b2547be8
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Tue Jan 18 10:41:35 2022 +0900

    add test stub

commit c86b1288d5e371f12cba4e1b1866966cb9264401
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Tue Jan 18 10:32:09 2022 +0900

    add python definition stub

commit 3166952f9673376801bf4b5b39eeb6f89452f30a
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Tue Jan 18 06:57:18 2022 +0900

    bwd filter compiled

commit e311ba3d05c5f9424ecb952cb5a520ce81a0828a
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Tue Jan 18 06:27:55 2022 +0900

    dgrad compiled

commit 47f35beb5eeeb7cbf9f6ec7cf8f5c80c65e8da46
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Tue Jan 18 06:16:43 2022 +0900

    add dgrad stub

commit ebed032d15b1c3895f541c46ce5d80b6dd769034
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Mon Jan 17 17:01:56 2022 +0900

    cpplint

commit 834f54a8c13512130e7d91ca0f54268dc06c5481
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Mon Jan 17 16:55:58 2022 +0900

    remove cudnn get output

commit dcbd9c95fdb8ffef9db9c2350430b270461a31c3
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Mon Jan 17 16:28:07 2022 +0900

    more refactor

commit 146464e8496fff972bdb1687c4e9d432fe3278d5
Author: Masahiro Masuda <masahi129@gmail.com>
Date:   Mon Jan 17 15:57:35 2022 +0900

    Introduce SetConvdescriptors to refactor cudnn/conv_forward.cc

* add python function for cudnn wgrad

* adding wgrad test

* black

* wgrad nchw and nhwc worked

* remove bwd algo name stuff

* compute output shape properly

* swap arg order in wgrad

* add kernel size arg in test

* black

* cleanup

* more fix

* fix dgrad test

* support running relay conv2d_backward_weight directly with cudnn

* black

* refactor reference function to support nhwc

* removed unused function

* lint

* enable offloading conv2d_transpose to cudnn dgrad

* relax tol

* name fix, remove print
2022-01-23 06:58:31 +09:00
Siyuan Feng 64f2939290 [MetaSchedule] disallow_dynamic_loop (#9997)
* [MetaSchedule] disallow_dynamic_loop

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

* Update src/meta_schedule/postproc/disallow_dynamic_loop.cc

Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
2022-01-22 13:23:49 -08:00
Tristan Konolige e9ee73f441 [TOPI,CUDA] Don't enable cudnn conv2d kernel if is not supported (#10021)
* [TOPI,CUDA] Don't enable cudnn conv2d kernel if is not supported

Specifically, check that layout is not NCHW if datatype is int8.

* remove all conv2d_cudnn int8 support
2022-01-22 13:04:02 +09:00
Hongyi Jin 1ac01b4aa1 [MetaSchedule] Schedule Rule: Cross Thread Reduction (#9994)
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
2022-01-21 14:53:29 -08:00
Krzysztof Parzyszek 751f83b565 Auto-discover C/C++ compiler instead of hardcoding g++ (#10007)
Some platforms (e.g. FreeBSD) use clang as the default OS compiler,
and there is no g++.
2022-01-21 13:25:08 -08:00
Steven S. Lyubomirsky 25c8f4cea5 [frontend][keras] Add support for TimeDistributed (#7006)
* First pass on modifying Keras importer to handle TimeDistributed

* Use squeeze inside TimeDistributed, add tests

* linter fixes

* More linting

* Even more linting

* Fix unused argument annotations

* Forgot one pylint annotation

* Forgot to set up data layout in _convert_activation

* Decouple data_layout from etab

* Linting fix

* Forgot to set data_layout argument

* Missed an etab.data_format, also test_conv1d was not in the test file's main

* Rebase fixes

* Linting fix

* _convert_lambda needs a data layout argument too

* linting fix too

* Lint the test file too

* Redundant variables

* Simplify further

* Another simplification

Co-authored-by: Steven Lyubomirsky <slyubomirsky@octoml.ai>
2022-01-21 10:05:19 -08:00
Ruihang Lai d97274ce19 [MetaSchedule] Post Processor: Rewrite Reduction Block (#10013)
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Xiyou Zhou <xiyou@octoml.ai>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
2022-01-21 09:46:39 -08:00
Xiyou Zhou e05a62ba7c [Minor] Typo Fixes (#10000)
* Fix typos.

* Missed funtion -> function.
2022-01-21 10:26:09 +08:00