Skip to content

Conversation

pmeier
Copy link
Collaborator

@pmeier pmeier commented Sep 22, 2022

Addresses 1. from #6622. In addition to the added {Kernel, Dispatcher}Info's this also introduces the ability to define test skips for specific tests on an info.

That is useful if a test does not work at all for a kernel or dispatcher, e.g. test_batched_vs_single is hard to do for {five, ten}_crop_image_tensor since the output is not a tensor, but rather a sequence of tensors. Plus, in some cases a specific parameter configuration does not work for a test, e.g. integer size's in {five, ten}_crop_image_tensor are not scriptable, so this can be individually skipped for test_scripted_vs_eager.

The same architecture can also be used for xfailing or fine tuning the tolerances, but so far there is no need for this so I didn't implement it yet.

@pmeier pmeier requested a review from vfdev-5 September 22, 2022 09:28
@pmeier pmeier changed the title Kernel dispatcher infos add more KernelInfo's and DispatcherInfo's Sep 22, 2022
Copy link
Collaborator

@vfdev-5 vfdev-5 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the PR @pmeier !

@pmeier pmeier merged commit 3c9ae0a into pytorch:main Sep 22, 2022
@pmeier pmeier deleted the kernel-dispatcher-infos branch September 22, 2022 12:15
facebook-github-bot pushed a commit that referenced this pull request Sep 23, 2022
Summary:
* add KernelInfo for adjust_brightness

* add KernelInfo for adjust_contrast

* add KernelInfo for adjust_hue

* add KernelInfo for adjust_saturation

* add KernelInfo for clamp_bounding_box

* add KernelInfo for {five, ten}_crop_image_tensor as well as skip functionality

* add KernelInfo for normalize

* add KernelInfo for adjust_gamma

* cleanup

* add DispatcherInfo's for previously add KernelInfo's

* add dispatcher info for elastic

Reviewed By: NicolasHug

Differential Revision: D39765294

fbshipit-source-id: 42bdb40ead8807f60fc75fb9259dd34e2d4f0724
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants