Commit0aaaae6

committed

refine _DEVICES

1 parentf0c1787 commit0aaaae6Copy full SHA for 0aaaae6

File tree

-5

lines changed

-5

lines changed

Lines changed: 1 addition & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -39,12 +39,8 @@`
`39`	`39`	`ifcommon_utils.SEEDisNone:`
`40`	`40`	`common_utils.SEED=1234`
`41`	`41`
`42`		`-_DEVICES= (`
`43`		`- ["cpu"]`
`44`		`-+ (["cuda"]iftorch.cuda.is_available()else [])`
`45`		`-+ (["xpu"]iftorch.xpu.is_available()else [])`
`46`		`-)`
`47`	`42`	`_DEVICE=get_current_accelerator_device()`
	`43`	`+_DEVICES= ["cpu"]+ ([_DEVICE]iftorch.accelerator.is_available()else [])`
`48`	`44`
`49`	`45`
`50`	`46`	`def_reset():`

Comments

(0)