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

[torchscript] use device type when concatenating tensors #4554

Merged
merged 1 commit into from
May 19, 2022

Conversation

maryamdaneshi
Copy link
Contributor

Patch description

there were errors on some predictor gpu machines for device mismatch so explicitly using the device type when concatenating the tensors

Testing steps
Fixing the error
before

281281826_325898053026523_7071148804019578843_n

after:
Screen Shot 2022-05-18 at 4 44 11 PM

Other information

Copy link
Contributor

@stephenroller stephenroller left a comment

Choose a reason for hiding this comment

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

What a wild ride

@maryamdaneshi maryamdaneshi merged commit a81608d into main May 19, 2022
@maryamdaneshi maryamdaneshi deleted the maryam_fix_torch branch May 19, 2022 18:07
kushalarora pushed a commit that referenced this pull request Jun 15, 2022
Co-authored-by: Maryam Daneshi <maryamdaneshi@fb.com>
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants