Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Fix FInferShape for some ops to support partial type inference #18348

Merged
merged 2 commits into from
May 22, 2020

Conversation

Kh4L
Copy link
Contributor

@Kh4L Kh4L commented May 17, 2020

Description

Some operators were checking ndim_is_known on the input shape after accessing it, resulting in a runtime error.

This was preventing us from running a partial shape inference where some not (yet) defined input shape would trigger an exception.

This PR fixes it by adding the ndim_is_known check (or changing it to the correct position) in the FInferShape function of those operators.

@mxnet-bot
Copy link

Hey @Kh4L , Thanks for submitting the PR
All tests are already queued to run once. If tests fail, you can trigger one or more tests again with the following commands:

  • To trigger all jobs: @mxnet-bot run ci [all]
  • To trigger specific jobs: @mxnet-bot run ci [job1, job2]

CI supported jobs: [miscellaneous, unix-cpu, windows-cpu, clang, centos-gpu, website, edge, windows-gpu, centos-cpu, unix-gpu, sanity]


Note:
Only following 3 categories can trigger CI :PR Author, MXNet Committer, Jenkins Admin.
All CI tests must pass before the PR can be merged.

@szha szha merged commit d9fc74e into apache:master May 22, 2020
AntiZpvoh pushed a commit to AntiZpvoh/incubator-mxnet that referenced this pull request Jul 6, 2020
…e#18348)

* Fix FInferShape for some ops to support partial type inference

Signed-off-by: Serge Panev <[email protected]>

* Add missing ndim check in in matrix_op-inl.h

Signed-off-by: Serge Panev <[email protected]>
@Kh4L Kh4L mentioned this pull request Jul 27, 2020
Kh4L added a commit to Kh4L/incubator-mxnet that referenced this pull request Aug 10, 2020
Kh4L added a commit to Kh4L/incubator-mxnet that referenced this pull request Aug 11, 2020
Kh4L added a commit to Kh4L/incubator-mxnet that referenced this pull request Aug 12, 2020
ptrendx pushed a commit that referenced this pull request Aug 14, 2020
@szha szha added this to the v1.8.0 milestone Aug 23, 2020
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants