butchland/fastai_xla_extensions

xla_lr_find locks xm.xla_device in main process

Closed this issue · 0 comments

butchland commented

After running learner.xla_lr_find(), running xmp.spawn fails because the main process has locked on to an xla device.