Commit Graph

2827 Commits

Author SHA1 Message Date
Leandro Nunes 8ced546539 Fix line break in setup.py (#9029) 2021-09-17 10:16:57 +09:00
Anirudh Sundar a74ef13d18 [ONNX] QLinearAveragePool and QLinearGlobalAveragePool contrib op (#9017)
* [ONNX] QLinearAveragePool and QLinearGlobalAveragePool contrib op

* Fix linter error for variable name and else after return

* Separate quantized avg_pool impl and add TODO for global_avg_pool

* Fix comment typo
2021-09-16 16:57:43 -06:00
masahi db78d96cd7 [CUDA] Fix dense tensorcore legalize type error when units is specified (#9030)
* Fix dense tensorcore legalize type error when units is specified

* revert black change due to different version from CI
2021-09-16 14:34:16 -07:00
AndrewZhaoLuo 02fbaf0ed9 [Onnx] Fix NLL Loss tests (#8971)
* support negatibve indices in gather

* move check to Tensor level indexing, gathernd

* add test, update transform.h

* remove unneeded gather

* missing gather nd change

* update tests

* proper tensor comparison

* blacking

* lint

* fix error

* turn on test

* missing test case

* revert changes

* add normalize_gather_indices

* undo change

* update

* more removing diffs

* more undoing

Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>
2021-09-16 17:49:26 +09:00
Matthew Brookhart 2711229ec8 [ONNX] enable the onnx tests after PR #8274 merged (#9019)
* enable the onnx tests after PR #8274 merged

* fix lint
2021-09-16 01:32:08 -07:00
AndrewZhaoLuo 4c77bae772 [Onnx] Add momentum (#9000)
* add momentum

* make tests pass for momentum

* blacking

* lint

Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>
2021-09-16 13:29:02 +09:00
Haichen Shen 89bcc79355 fix (#9021) 2021-09-16 13:28:52 +09:00
anwang2009 777ace3f2b [Relay][Pass] Add ExtractOperators pass (#8996)
* add extractor

* extract to array

* add comments

* lint

* Update tests/python/relay/test_analysis_extract_operators.py

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

* op freqs

* add comment

* Update python/tvm/relay/analysis/analysis.py

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

* oops

* mixedmode visitor

* oops

Co-authored-by: Cody Yu <comaniac0422@gmail.com>
2021-09-16 09:49:08 +09:00
anwang2009 e44f6c0f41 [ONNX] Add Einsum converter (#8985)
* einsum

* address review

* move files around

* use generic topi op

* TODO comment

* jostle ci

* jostle ci
2021-09-16 05:05:19 +09:00
Leandro Nunes 2aebd3335d Add standalone_crt/ to be part of the wheel package, when available. (#9005)
* When using a packaged TVM such as tlcpack, it is impossible to run
  `tvm.micro.get_standalone_crt_dir()`, because the subtree
  `standalone_crt/` is not available.

* This patch adds `standalone_crt/` as `data_files`, so that they
  can be picked up by _ffi.libinfo.find_lib_path() and therefore
  be found when `tvm.micro.get_standalone_crt_dir()` is invoked.
2021-09-15 10:57:11 -07:00
Hua Jiang 9bc4dc0fb9 [Runtime] Pipeline Executor Initial patch. (#8702)
* [Runtime] Pipeline Executor Initial patch.

This patch is one of serial patch for PR 7892 splitting.this is the initial part
of the pipeline executor, this patch include the cmake change and python and C++
interface for pipeline executor.

* add pipeline config

* add config connect logic.

* fix build issue.

* set output index start from 0

* address review comments

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

* address review comments.

* address review comments.

* Update python/tvm/contrib/pipeline_executor.py

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

Update python/tvm/contrib/pipeline_executor.py

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

* address review comments.

* address review comments

* add topology sort

* add binding check logic.

* fix plint error.

* Update python/tvm/contrib/pipeline_executor.py

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

* address review comments.

* fix plint issue.

* address review comments.

* trigger build.

* Update python/tvm/contrib/pipeline_executor.py

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

address review comments

* address review comments.

* polish doc and comments.

* polish doc and address review comments.

* address review comments.

* doc change.

* doc change.

* Trigger build.

* trigge build.

* address review comments.

* address review comments.

* address review comments.

* polish documents.

* Polish the document.

* address review comments.

Co-authored-by: Cody Yu <comaniac0422@gmail.com>
2021-09-15 21:00:59 +09:00
masahi 39abead267 disable cuda int8 schedule for non-cuda gpu target (#9014) 2021-09-15 20:51:17 +09:00
masahi f350ea672c [Torch] Add an option to make imported models compatible with the Relay text parser (#9015)
* [Torch] Add an option to make imported models compatible with the
Relay text parser

* py format
2021-09-15 20:51:07 +09:00
mbaret 354019dd11 [3/6] Arm(R) Ethos(TM)-U NPU TIR compiler with conv2d support (#8806)
* Arm(R) Ethos(TM)-U NPU TIR compiler with conv2d support

This commit adds the lowering passes necessary to lower
an NPU Relay module down to a TIR module that can be
compiled for the NPU. Conv2d is supported as the first
NPU operator. An intermediate TE stage between Relay and
TIR allows support for scheduling the operators.

Co-authored-by: Manupa Karunaratne <Manupa.Karunaratne@arm.com>

* Fix Conv2D TIR type sensitivity

Change-Id: I3741f9dd8bb5952590ff8c586f6b96e5c3a03795

* Arm(R) Ethos(TM)-U NPU TIR passes and TE for Conv2D

*fixing tests

Change-Id: Id4a4c80f72ce29b98fc8b3954a1413c1c7fda500

* Fix import guards for tests

Change-Id: Iaee06017bd125d3040ce42182c4ccdb80d7fc946

* Fix typing failures with ignores

Change-Id: I81513f112a42b93cfdd3bcaf8e8852dd60ffe9e9

* Remove unused import

Change-Id: I6596b62ab56e4ca8b31ef08293686f53f38454d2

* Reintroduce get_target_accel_type

Change-Id: I0aaf83fe0204c0db435692e9b92dee6e9d6997fe

Co-authored-by: Manupa Karunaratne <Manupa.Karunaratne@arm.com>
2021-09-15 12:12:23 +01:00
AndrewZhaoLuo 34570f27ef [Onnx] Add Adam (#9002)
* add adam op

* lint

* remove tests

* lint
2021-09-15 12:38:24 +09:00
Tristan Konolige 089ffabf73 [TIR] Add conversion from FloatImm to float in Python (#9009)
This method matches the IntImm method for converting from IntImm to int.
2021-09-15 12:37:40 +09:00
Mehrdad Hessar dc2f70e3c8 [microTVM][AutoTVM] Fix autotvm bug and tests (#9003)
* debuggging

* cleanup and fix tutorial, zephyr and crt test

* fix crt test

* address comments
2021-09-14 15:41:01 -07:00
Siyuan Feng e1ae821c7d Add while node support in TVMScript (#9004)
* support while

* update synr version
2021-09-14 13:55:42 -04:00
AndrewZhaoLuo 1b99adcd8f [Onnx] Add Adagrad (#9001)
* adagrad impl

* passing tests

* docstring

Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>
2021-09-13 23:26:42 -07:00
Peter Salas 92903b4565 [Autoscheduler] Reduce task weight coercion overhead (#8995)
Co-authored-by: Peter Salas <psalas@octoml.ai>
2021-09-14 10:44:59 +09:00
Cody Yu 02f885aacd [AutoScheduler] Fix task scheduler after 8478 (#8984) 2021-09-11 11:46:27 -04:00
Jorn Tuyls 8b59f9995f [CI][VitisAI] Update CI Vitis AI PyXIR version to v0.3.1 (#8814)
* Update CI Vitis AI PyXIR version to v0.3.1

* Add Vitis AI requirements to gen_requirements.py
2021-09-11 14:28:57 +09:00
Junru Shao a44cc6ef87 [TensorIR][M2a] Compute-At (#8943)
This PR is part of the TensorIR upstreaming effort (#7527), which adds the following schedule primitives:
* `compute-at`
* `reverse-compute-at`

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>
2021-09-09 10:40:45 -07:00
Mehrdad Hessar aa2b37d35b [microTVM] Add support for AutoTVM (#8715)
* Initial commit of API server impl.

* initial commit of api client

* Add TVM-side glue code to use Project API

* Change tvm.micro.Session to use Project API

* Rework how crt_config.h is used on the host.

 * use template crt_config.h for host test runtime; delete
   src/runtime/crt/host/crt_config.h so that it doesn't diverge from
   the template
 * bring template crt_config.h inline with the one actually in use
  * rename to MAX_STRLEN_DLTYPE
 * Create a dedicated TVM-side host crt_config.h in src/runtime/micro

* Modify Transport infrastructure to work with Project API

* Add host microTVM API server

* Zephyr implementation of microTVM API server

 * move all zephyr projects to apps/microtvm/zephyr/template_project

* consolidate CcompilerAnnotator

* Allow model library format with c backend, add test.

* Update unit tests

* fix incorrect doc

* Delete old Zephyr build infrastructure

* Delete old build abstractions

* Delete old Transport implementations and simplify module

* lint

* ASF header

* address gromero comments

* final fixes?

* fix is_shutdown

* fix user-facing API

* fix TempDirectory / operator

* Update micro_tflite tutorial

* lint

* fix test_crt and test_link_params

* undo global micro import, hopefully fix fixture

* lint

* fix more tests

* Add session_constructor_args to tracker request() function.

 * Allows tracker clients to open non-traditional RPC sessions

* Generate entry_func symbol in C host codegen.

 * Needed for AutoTVM.

* print MeasureErrorNo enum value in MeasureResult repr

* Add microTVM session constructor.

 * This constructor is to be called from the RPC driver to flash and
   connect to the RPC server on the microcontroller.

* add build_kwargs as a Builder constructor arg.

 * build_kwargs is derived from pre-configured args, the runner, and
   now from the script.
 * user-supplied build kwargs override the other two, and a warning is
   printed if any key is overridden.

* Add do_fork option to Builder, to support stateful builders

 * When AutoTVM builder forks, any global state modified by the
   build_func is lost between builds

* Checkin module_loader used to build and flash microTVM for autotuning.

* Import micro into top-level when enabled.

 * AutoTVM RPC server needs to load the micro session constructor.

* Add tvm.contrib.random.random_fill to microTVM.

 * Allows autotuning with random data.

* Move compilation to runner :O

* Add a tutorial for AutoTVM with microcontrollers.

* Fix si_prefix in autotuner callback

* black format and git-clang-format

* Switch tutorial back to qemu version

* improve error reporting so CI will show test error

* black format

* autotvm is working

* fix tutorial

* fix dependencies

* fix auto tune issue

* lint

* address comments

* fix lint

* test crt and zephyr added

* fix func registery size

* moved autotune test and fixed

* fix crt test

* address comments

* change relay text

* change relay in text_zephyr

* class added

* changed relay module in tutorial and cleanup

* address comments

* address TK comments

* change fork

* final comments

* retrigger due to flahy test

* fix tutorial

* retrigger

* fix changes due to merge

Co-authored-by: Andrew Reusch <areusch@octoml.ai>
2021-09-09 18:51:44 +09:00
Yuanjing Shi 7f986d75cc change the doc to reflect previous code change (#8970) 2021-09-09 17:32:28 +09:00
anwang2009 41cef56857 [ONNX] Support depth_to_space op for FQ2I (#8966)
* WIP support per-channel quantization

* more WIP

* More WIP

* fix issue with per-channel bias_add

* Fix fake quantize tests (#4)

* Fixed fake quantize issues.

* Formatting.

* Cleanup unused imports

* Fix real int8 tests.

* Add Relu

* One more little one (#5)

* Fixed fake quantize issues.

* Formatting.

* Cleanup unused imports

* Fix real int8 tests.

* Fix requantize shape bug.

* Non-working Per-channel Dense

* Fix legalization for non spatial operators. (#6)

* Fix legalization for non spatial operators.

* Fix axis checks for end2end functionality.

* fix axis normalization

fix lint

fix lint again

* Per channel fq2i (#8)

* WIP support per-channel quantization

* more WIP

* More WIP

* fix issue with per-channel bias_add

* Fix fake quantize tests (#4)

* Fixed fake quantize issues.

* Formatting.

* Cleanup unused imports

* Fix real int8 tests.

* Add Relu

* One more little one (#5)

* Fixed fake quantize issues.

* Formatting.

* Cleanup unused imports

* Fix real int8 tests.

* Fix requantize shape bug.

* Non-working Per-channel Dense

* Fix legalization for non spatial operators. (#6)

* Fix legalization for non spatial operators.

* Fix axis checks for end2end functionality.

* fix axis normalization

fix lint

fix lint again

* Fix bug in requantize dimension expansion.

* Format.

Co-authored-by: Josh Fromm <jwfromm@octoml.ai>

* respond to review comments

* start dtos

* wip depth_to_space

* dtos ident

Co-authored-by: Matthew <mbrookhart@octoml.ai>
Co-authored-by: Josh Fromm <jwfromm@octoml.ai>
2021-09-09 14:57:25 +09:00
AndrewZhaoLuo 548675fddc [EZ] [ONNX] Remove unnecessary converters for greater and lesser (#8967)
* simplify by removing unneeded conversions

* remove uneeded

* remove test

Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>
2021-09-09 14:56:02 +09:00
Andrey Malyshev 1bebd0a692 Add sse4/avx2 support for fast x86 int8 (vpmaddubsw/vpmaddwd/vpaddd) (#8897)
* Add sse4/avx2 support for vpmaddubsw/vpmaddwd/vpaddd

- Extend the list of different target for x86 topi
- Extend tests for conv2d x86 int8 for fast i8 x86 platforms

* fix code style

* Change x86-64-v2 to nahalem in test to support llvm11

* Change test target to get NCHW8c
2021-09-09 14:55:19 +09:00
Tristan Konolige c650f9ac15 [PROFILING] Profiling over RPC (#8885)
* [PROFILING] Profiling over RPC

Allow for profiling over RPC by serializing the returned report before
sending it. Also remove collectors argument when profiling over rpc
because it cannot be serialized.

* lint

* fixes

* add comments
2021-09-09 14:53:31 +09:00
Haichen Shen 47d9fd3f5f [TOPI] Fix CUDA pooling schedule (#8957) 2021-09-08 22:44:03 -07:00
Leandro Nunes 2e0a71184a Set tvm.micro.project_api as a Python Module (#8963)
* Add missing tvm.micro.project_api module file. The missing
  __init__.py makes it impossible to import this module with
  `import tvm.micro.project_api`.

* This uncover 30-ish linting errors, which are also fixed here.
2021-09-09 06:32:26 +01:00
Sen Yang f8b1df4d29 [Bugfix] Fix visit_attrs error if its function pointer is equal to nullptr (#8920)
* fix visit_attrs equals nullptr on python container object

* add a test a for python container object about function dir and getattr

* change test_ir_container.py to the pytest style

* update the style to fix ci error

* update the style of ir container to fix ci error
2021-09-08 13:50:01 -07:00
Manupa Karunaratne 01aeeb196a [2/6] Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op (#8795)
* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

This commit adds mainly the relay passes and ethosu_conv2d
operator to relay. The relay passes include the legalizations
and preprocessing of the relay graph in the paritioning.

Co-authored-by: Matthew Barrett <matthew.barrett@arm.com>

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

* skipping the test if vela is not in the container.

Change-Id: I68cc4259dc33e1473e460956978f364fbf6596d8

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

* addressing Jared's comments

Change-Id: Ief669f788c6bd1a1be1004cbce5129ed06b63c3c

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

* addressing Elen's comments

Change-Id: Iad6315bb63f12ba318deb9c5c9eff7459ff58c48

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

* cleanup passes

Change-Id: I8e1cbedd2c4d3d0cdff481d775d9eb0577e44456

* Update TE comments

Change-Id: I7e65c2714d017c8a4b64986b111a6b51d128c963

* Address ekalda's comments in TE

Change-Id: I55cfbb3787c0aacdadf46c4859dff39287e65ddc

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

*addressing chris's comments
*addressing Nicola's comments

Change-Id: Id02788ddcdbc3679e0da37b2fa614cded0a4c1f5

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

*addressing missed 'hidden' comments of Chris
*addressing one missed comment of Nicola
*adding type annotations

Change-Id: Iadf4907b311e195731dbbed571e95a266341db8f

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

*further type fixes and one missed comment

Change-Id: I6da69fd95d17dfeaf5940da4f8d8c8ea142b39d2

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

*missed comment split_o

Change-Id: I4a4b19ff2cd18e8f568a63ae827f44358ed85b8e

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

* adding mypy check

Change-Id: Iaf58dbba2a9d8e1098a10c589d91b63c7efe646d

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

*removing premature insertion of get_accel_type utility

Change-Id: I210512e00a5eb46adf23d1d72eb16432db526d25

* Arm(R) Ethos(TM)-U NPU Relay passes and Conv2D op

* rebase fixes

Change-Id: I06c9b536a7598646efce2b664fcc405aa6008203

Co-authored-by: Matthew Barrett <matthew.barrett@arm.com>
2021-09-08 10:15:17 -07:00
Yuanjing Shi 1f2fdbf6c2 [AutoTVM] Use popenpool in local_executor (#8851)
* use popenpool in local_executor

* move auto_tvm_common to tvm.testing

* refactor

* nit

* remove LocalFutureNoFork

* exception handling

* handling two exceptions

* handling error

* add initiazlier
2021-09-08 13:08:15 -04:00
Matthew Brookhart cf439ec7e7 support slicing with out of order axes (#8959) 2021-09-08 09:33:01 -06:00
Matthew Brookhart 4ffbdcd0aa [Relay][Quantization] Per-Channel FQ2I (#8883)
* WIP support per-channel quantization

* more WIP

* More WIP

* fix issue with per-channel bias_add

* Fix fake quantize tests (#4)

* Fixed fake quantize issues.

* Formatting.

* Cleanup unused imports

* Fix real int8 tests.

* Add Relu

* One more little one (#5)

* Fixed fake quantize issues.

* Formatting.

* Cleanup unused imports

* Fix real int8 tests.

* Fix requantize shape bug.

* Non-working Per-channel Dense

* Fix legalization for non spatial operators. (#6)

* Fix legalization for non spatial operators.

* Fix axis checks for end2end functionality.

* fix axis normalization

fix lint

fix lint again

* Per channel fq2i (#8)

* WIP support per-channel quantization

* more WIP

* More WIP

* fix issue with per-channel bias_add

* Fix fake quantize tests (#4)

* Fixed fake quantize issues.

* Formatting.

* Cleanup unused imports

* Fix real int8 tests.

* Add Relu

* One more little one (#5)

* Fixed fake quantize issues.

* Formatting.

* Cleanup unused imports

* Fix real int8 tests.

* Fix requantize shape bug.

* Non-working Per-channel Dense

* Fix legalization for non spatial operators. (#6)

* Fix legalization for non spatial operators.

* Fix axis checks for end2end functionality.

* fix axis normalization

fix lint

fix lint again

* Fix bug in requantize dimension expansion.

* Format.

Co-authored-by: Josh Fromm <jwfromm@octoml.ai>

* respond to review comments

respond to review comments

Co-authored-by: Josh Fromm <jwfromm@octoml.ai>
2021-09-08 17:24:18 +09:00
Krzysztof Parzyszek e0aac9402b [Hexagon] Fix VTCM allocation (#8954)
Check if a buffer is in the `vtcm_buffers` list, before it's removed
from it.
2021-09-08 16:50:58 +09:00
AndrewZhaoLuo 9a47fc0a2f [Onnx] Pow support for other types (#8933)
* update pow

* update pow

* remove duplicate

Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>
2021-09-08 15:50:43 +09:00
anwang2009 475e9e0296 [ONNX] Add support for QLinearConcat contrib op (#8907)
* add qlinearconcat op

* fix tests

* Fix

* lint

* lint

* review

* boop ci

* fix regression

* noop

* jostle ci
2021-09-08 15:25:21 +09:00
Gustavo Romero af9d100361 [microTVM] Add method to query template info without creating a project (#8950)
Add info() method to TemplateProject class so it's possible to query all
available options for a given template project without creating a new
one. This is necessary because TVMC will query the available options for
a given template project to show them to the user so the user can use
them to finally create a new project dir.

That is also useful in general to query the available options for any
project type. For example, one can query all boards available on the
Zephyr platform with:

import tvm.micro.project as project_api

template = project_api.TemplateProject.from_directory(ZEPHYR_TEMPLATE_DIR)
boards = template.info()["project_options"][8]["choices"]

where 8 element refers to the "zephyr_board" option.

Signed-off-by: Gustavo Romero <gustavo.romero@linaro.org>
2021-09-08 09:50:32 +09:00
Valery Chernov 8027a7a692 [ONNX] [Test] fix GRU modification and reduce tolerance for RNN tests (#8923)
* fix high tolerance for RNN tests

* random seed was added to GRU test reproduction

Co-authored-by: Valery Chernov <valery.chernov@deelvin.com>
2021-09-08 09:48:33 +09:00
Wuwei Lin 332afaecdd [AutoScheduler] Fix custom build func in PopenWorker (#8939)
* [AutoScheduler] Fix custom build func in PopenWorker

* Add assertion
2021-09-07 20:04:58 -04:00
Ashutosh Parkhi 1dc1707f8c [CMSIS-NN] code generator for softmax (#8833) 2021-09-07 16:53:18 -07:00
Lunderberg 6dea994e43 [UnitTest][Vulkan] Runnable relay unit tests on Vulkan (#8947)
* [UnitTest] Added ids argument to tvm.testing.parameters

This matches the usage in `tvm.testing.parameter`, and allows for
parameter sets to be referred to by a single name.

* [Pytest] Fixed ordering issue of tvm.testing.parametrize_targets and known_failing_targets

If an explicit list of targets is given, then the
`known_failing_targets` decorator would fail to apply.  This commit
resolves the issue, and cleans up all target-specific marks to apply
in `tvm.testing.plugin._add_target_specific_marks`.

* [UnitTest][Vulkan] Runnable relay unit tests on Vulkan

This commit allows the relay test suite to be run targeting Vulkan with
`TVM_TEST_TARGETS="vulkan -from_device=0" pytest tests/python/relay`.  All
tests that require a specific environment are skipped if that environment
isn't present.  All tests that are known to fail when running on Vulkan
are marked as expected failure, and will be tracked in
https://github.com/apache/tvm/issues/8903.

- Failures during code generation
  - Type mismatches, boolean vs int8
    - tests/python/relay/test_any.py::test_any_reduce
    - tests/python/relay/test_op_level3.py::test_sparse_reshape
    - tests/python/relay/test_op_level4.py::test_reduce_functions
    - tests/python/relay/test_vm.py::test_cond
    - tests/python/relay/test_vm.py::test_simple_if

  - Incorrect strategy selection, picks NCHWc implemenation for NHWC layout
    - tests/python/relay/test_op_level2.py::test_conv2d_run

  - Unresolved CallNode operation
    - tests/python/relay/test_op_level1.py::test_unary_op[erf/tan/atan]
    - tests/python/relay/test_op_level3.py::test_scatter_add
    - tests/python/relay/test_op_level3.py::test_segment_sum

  - Generates 64-bit calls to GLSL that have only 16-/32-bit support
    - tests/python/relay/test_op_grad_level1.py::test_log_softmax_grad
    - tests/python/relay/test_op_grad_level1.py::test_softmax_grad
    - tests/python/relay/test_op_grad_level1.py::test_unary_op
    - tests/python/relay/test_op_grad_level10.py::test_cross_entropy_grad

  - Codegen raises error for variable size
    - tests/python/relay/test_any.py::test_any_batch_matmul
    - tests/python/relay/test_any.py::test_any_conv2d_NCHWc
    - tests/python/relay/test_any.py::test_any_dense

- Failures when running
  - Numeric differences (observed on GTX 1650 with NVIDIA driver)
    - tests/python/relay/test_op_level3.py::test_take
    - tests/python/relay/test_op_level5.py::TestCropAndResize
    - tests/python/relay/test_op_level5.py::TestResize1D
    - tests/python/relay/test_op_level5.py::TestResize2D
2021-09-08 07:44:35 +09:00
masahi dd25fad6cb [CUDA] Improve local_response_norm schedule (#8946)
* Improve cuda lrn schedule

* fuse reduction and the next elemwise kernel

* remove cpp schedule

* fix

* fixed unintended revert

Co-authored-by: masa <masa@pop-os.localdomain>
2021-09-07 15:37:50 -04:00
masahi 0fb840eb5d [Layout] Unify dense op input layout (#8921) 2021-09-07 10:12:58 -07:00
Krzysztof Parzyszek 9b034d729f [Hexagon] Add support for linked-in model parameters (#8865)
* [Hexagon] Add support for linked-in model parameters

* Remove entry_func, since it's not used anywhere

* Simplify linked-param codegen preparation a bit

* Detect multiple linked-params functions

* Add testcase to check for linked-param codegen

* Empty commit to restart build
2021-09-06 18:54:15 -05:00
masahi ab0f0553b2 [TIR] Fixed LowerThreadallreduce not remapping Store buffer var (#8931)
* Fixed LowerThreadallreduce not remapping Store buffer var

* reenable warp reduction schedule for softmax with fused ops

Co-authored-by: masa <masa@pop-os.localdomain>
2021-09-06 14:18:39 -07:00
Siyuan Feng 22dbc3a932 [TVMScript] Enhance printer (#8934) 2021-09-06 14:18:24 -07:00
masahi 054e2bbef9 [CUDA] Improve adaptive and global pool schedule (#8936) 2021-09-06 09:08:41 -07:00