Skip to content

fixed bug: features and indexes on diffent devices when using cuda.#29

Open
zergzzlun wants to merge 1 commit into
aRI0U:gatherfrom
zergzzlun:gather
Open

fixed bug: features and indexes on diffent devices when using cuda.#29
zergzzlun wants to merge 1 commit into
aRI0U:gatherfrom
zergzzlun:gather

Conversation

@zergzzlun
Copy link
Copy Markdown

Hi there.

Using CUDA version of local spatial encoding in the 'gather' branch (#17) will get the following error:

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking arugment for argument index in method wrapper_gather)

This can be fixed by moving indexes (expanded_idx) to the same device of features.

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.

1 participant