Skip to content

Make nms fallback by default.#6933

Merged
ysiraichi merged 4 commits intomasterfrom
ysiraichi/make-nms-fallback
Apr 20, 2024
Merged

Make nms fallback by default.#6933
ysiraichi merged 4 commits intomasterfrom
ysiraichi/make-nms-fallback

Conversation

@ysiraichi
Copy link
Copy Markdown
Collaborator

Fix: #6899

PR #6814 broke timm_efficientdet benchmark because of the new nms lowering. Since nms is a dynamic operation, it should, as other dynamic operations (e.g. nonzero), be flagged as experimental. i.e. only enable the lowering if XLA_EXPERIMENTAL=nms.

cc @miladm @JackCaoG

@JackCaoG
Copy link
Copy Markdown
Collaborator

test_nms_input_errors failed, this seems like a real error

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[torchbench] timm_efficientdet inference fails to run.

2 participants