onnxruntime
Assert that the data is on the same device as ORTModule
#6942

Merged

Assert that the data is on the same device as ORTModule #6942

baijumeswani merged 3 commits into thiagofc/ortmodule-api from bmeswani/data_device_check

Assert that the data and module devices are the same

a07baef6

baijumeswani requested a review from

BowenBao 5 years ago

baijumeswani requested a review from

liqunfu 5 years ago

baijumeswani requested a review from

spandantiwari 5 years ago

baijumeswani requested a review from

thiagocrepaldi 5 years ago

thiagocrepaldi approved these changes on 2021-03-08

Check if input is a tensor before checking device

a5706e08

Change _check_device to _check_same_device

56aadcb1

thiagocrepaldi approved these changes on 2021-03-09

SherlockNoMad added component:ortmodule

baijumeswani merged f1ade14e into thiagofc/ortmodule-api 5 years ago

baijumeswani deleted the bmeswani/data_device_check branch 5 years ago

Reviewers

thiagocrepaldi

BowenBao

liqunfu

spandantiwari

Assignees

No one assigned

Labels

None yet

Milestone

No milestone