[ONNX] support additional nllloss tests #9045
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
AndrewZhaoLuo
changed the title
AndrewZhaoLuo
requested review from
areusch,
comaniac,
Huyuwei,
jroesch,
junrushao,
jwfromm,
kazum,
mbrookhart,
merrymercy,
siju-samuel,
srkreddy1238,
tqchen and
yzhliu
as code owners
added 2 commits
September 22, 2021 14:52
|
Going to wait after #9095 to merge |
* main: (37 commits) [ONNX] [Relay] Dynamic squeeze (apache#9095) [Meta Schedule][M3b] Database (apache#9061) [Bugfix] Add nullptr checking for `AttrStmt` with `coproc_uop_scope` attr key (apache#9123) [Codegen] Swap out analyzer when outlining (apache#9117) [CI] bash.sh, build.sh: add option to set the container name and hostname (apache#9110) Ensure google-mock is installed and setup (apache#9107) Arm(R) Ethos(TM)-U NPU TIR to CS for Conv2D (apache#8811) Frontend: add onnx GlobalLpPool op (apache#8845) [LLVM] Refactor MakeCallPacked, NFC (apache#9118) prevent casting handle to other types (apache#9114) fix annotation of tir generic (apache#9119) [Relay] Register layout conversion function to more reduce ops (apache#9048) Fix the missing `dtype` attribute of `tir.Shuffle` in Python level (apache#9131) add `multiply` and remove `subtract` for dnnl json runtime (apache#9120) relu of dnnl json runtime only support 4-dims input (apache#9122) [Meta Schedule][M3a] SpaceGenerator (apache#9079) [TensorIR][Bugfix] Disallow fusing loops with dependency (apache#9112) adding Jorn to reviewers list (apache#9105) [BYOC] Fix incorrect conv2d padding handling of `dnnl with c source runtime` (apache#9097) [Frontend][TFLite] fix apache#9078 (apache#9099) ...
|
Ok this is ready now |
mbrookhart
approved these changes
Sep 28, 2021
* main: Fix flaky NMS test by making sure scores are unique (apache#9140) [Relay] Merge analysis/context_analysis.cc and transforms/device_annotation.cc (apache#9038) [LLVM] Make changes needed for opaque pointers (apache#9138) Arm(R) Ethos(TM)-U NPU codegen integration (apache#8849) [CI] Split Integration tests out of first phase of pipeline (apache#9128) [Meta Schedule][M3b] Runner (apache#9111) Fix Google Mock differences between Ubuntu 18.04 and 16.04 (apache#9141) [TIR] add loop partition hint pragma (apache#9121) fix things (apache#9146) [Meta Schedule][M3a] SearchStrategy (apache#9132) [Frontend][PyTorch] support for quantized conv_transpose2d op (apache#9133) [UnitTest] Parametrized test_conv2d_int8_intrinsics (apache#9143) [OpenCL] Remove redundant visit statement in CodeGen. (apache#9144) [BYOC] support arbitrary input dims for add/mul/relu of dnnl c_src codegen (apache#9127) [Relay][ConvertLayout] Support for qnn.conv2d_transpose (apache#9139) add nn.global_avgpool to fq2i (apache#9137) [UnitTests] Enable minimum testing on Vulkan target in CI (apache#9093) [Torch] Support returning quantized weights and bias for BYOC use cases (apache#9135) [Relay] Prepare for new plan_devices.cc (part II) (apache#9130) [microTVM][Zephyr] Add MIMXRT1050 board support (apache#9068)
ylc
pushed a commit
to ylc/tvm
that referenced
this pull request
Jan 7, 2022
* initial dyn unsqueeze example * simplify, properly unpack scalar * basic tests * squish bugs -- assign proper types * working topi * fix things * temp work * fix casting to int64 * shape encoding method for axis * working shape encoding metric * add comment * move to non-rank encoded axis * failing regime * fix * it works! * add test * add comment on shape func * remove unused topi * undo some file changes * more cleanup * newline * clean up * clean up * enable multiple axis tests * move tests to dynamic op * Update docs * add converter * initial dyn unsqueeze example * simplify, properly unpack scalar * basic tests * squish bugs -- assign proper types * working topi * fix things * temp work * fix casting to int64 * shape encoding method for axis * working shape encoding metric * add comment * move to non-rank encoded axis * failing regime * fix * it works! * add test * add comment on shape func * remove unused topi * undo some file changes * more cleanup * newline * clean up * clean up * enable multiple axis tests * move tests to dynamic op * Update docs * add converter * working tests * add test, remove unneeded file * fix things * more lint * more lint * pick things * disable opencl tests * unsqueeze tests * clean up * dyn stuff * add num_newaxis * add support * black * doc string * remove bad merge * fix default axis behavior * rebase * fix squeeze * jostle ci Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>
ylc
pushed a commit
to ylc/tvm
that referenced
this pull request
Jan 13, 2022
* initial dyn unsqueeze example * simplify, properly unpack scalar * basic tests * squish bugs -- assign proper types * working topi * fix things * temp work * fix casting to int64 * shape encoding method for axis * working shape encoding metric * add comment * move to non-rank encoded axis * failing regime * fix * it works! * add test * add comment on shape func * remove unused topi * undo some file changes * more cleanup * newline * clean up * clean up * enable multiple axis tests * move tests to dynamic op * Update docs * add converter * initial dyn unsqueeze example * simplify, properly unpack scalar * basic tests * squish bugs -- assign proper types * working topi * fix things * temp work * fix casting to int64 * shape encoding method for axis * working shape encoding metric * add comment * move to non-rank encoded axis * failing regime * fix * it works! * add test * add comment on shape func * remove unused topi * undo some file changes * more cleanup * newline * clean up * clean up * enable multiple axis tests * move tests to dynamic op * Update docs * add converter * working tests * add test, remove unneeded file * fix things * more lint * more lint * pick things * disable opencl tests * unsqueeze tests * clean up * dyn stuff * add num_newaxis * add support * black * doc string * remove bad merge * fix default axis behavior * rebase * fix squeeze * jostle ci Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Turns out a bunch of these tests did not really require dynamic unsqueeze support; their input nodes were constants. There is an issue with one test we should look into later but regardless with these changes many tests work.
Depends on #9039