PR #427 Support specific device id for ORTModel

Support specific device id for ORTModel #427

fxmarty merged 9 commits into huggingface:main from fxmarty:support-specific-device-id

support gpu id

ee380a16

doc

a11142e1

simplify code, add test

a3136857

skip test

9387142a

fix decorator

c42f8798

fxmarty requested a review from

JingyaHuang 3 years ago

fxmarty requested a review from

echarlaix 3 years ago

JingyaHuang commented on 2022-10-20

simplify code

f9fa4680

fix

f2466004

fix gpu count in tests

e593198a

JingyaHuang approved these changes on 2022-10-20

fxmarty force pushed from 1a7b35bf to 2cdf16a3 3 years ago

fxmarty force pushed from 2cdf16a3 to 9ed2138e 3 years ago

fxmarty force pushed from 9ed2138e to 58a66267 3 years ago

fxmarty force pushed from 58a66267 to 5c555298 3 years ago

fxmarty force pushed from 5c555298 to 84b90075 3 years ago

fxmarty force pushed from 84b90075 to a21fef80 3 years ago

fxmarty force pushed from a21fef80 to 7341bec9 3 years ago

fxmarty force pushed from 7341bec9 to 435d35a1 3 years ago

fxmarty force pushed from 435d35a1 to e593198a 3 years ago

disable parity test temporarily, tracked shuffle issue in datasets

ff35a00a

fxmarty force pushed from b04f2471 to ff35a00a 3 years ago

fxmarty merged e78572f7 into main 3 years ago

Reviewers

JingyaHuang

echarlaix

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

optimum Support specific device id for ORTModel #427 Merged

Support specific device id for ORTModel #427

optimum
Support specific device id for ORTModel
#427

Merged