Skip to content

Conversation

vfdev-5
Copy link
Collaborator

@vfdev-5 vfdev-5 commented Sep 1, 2020

Description:

  • Added cuda and torch.jit tests for _rgb2hsv and _hsv2rgb methods.

@vfdev-5 vfdev-5 changed the title Added cuda torch.jit tests for rgb-hsv tensor conversion ops Added cuda and torch.jit tests for rgb-hsv tensor conversion ops Sep 1, 2020
@vfdev-5 vfdev-5 marked this pull request as draft September 1, 2020 23:28
@vfdev-5 vfdev-5 marked this pull request as ready for review September 2, 2020 08:42
@vfdev-5 vfdev-5 requested a review from fmassa September 2, 2020 08:43
Copy link
Member

@fmassa fmassa left a comment

Choose a reason for hiding this comment

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

Looks good to me in general, but I think there is a change I pointed out that might make the test flaky, can you have a look?

Copy link
Member

@fmassa fmassa left a comment

Choose a reason for hiding this comment

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

Thanks a lot!

@fmassa fmassa merged commit ca3d193 into pytorch:master Sep 2, 2020
@vfdev-5 vfdev-5 deleted the vfdev-5/rgb-hsv-cuda-tests branch September 2, 2020 10:16
bryant1410 pushed a commit to bryant1410/vision-1 that referenced this pull request Nov 22, 2020
…orch#2639)

* Added cuda torch.jit tests for rgb-hsv tensor conversion ops

* Reverted deleted test
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.

2 participants