Skip to content

Fixed for cuda implementation#16

Open
steviestevepy wants to merge 1 commit into
uber-research:masterfrom
steviestevepy:fix-cuda-implementation
Open

Fixed for cuda implementation#16
steviestevepy wants to merge 1 commit into
uber-research:masterfrom
steviestevepy:fix-cuda-implementation

Conversation

@steviestevepy

Copy link
Copy Markdown

After running the adaptive example I was able to spot some errors. These were mainly related to the the tensors having different device type (cpu and cuda). Tensors expected to be on same device.

The fix has allowed the validation of tensors before processing. Resulting in no exceptions during the run.

I hope it helps.

@CLAassistant

CLAassistant commented Feb 9, 2021

Copy link
Copy Markdown

CLA assistant check
All committers have signed the CLA.

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