Doesnot work with torch.compile

facebookresearch / schedule_free

Schedule-Free Optimization in PyTorch

Apache License 2.0

1.83k stars 63 forks source link

Doesnot work with torch.compile #30

Open yuanzhi-zhu opened 3 months ago

yuanzhi-zhu commented 3 months ago

after adding model = torch.compile(model) in the example/mnist code, I got RuntimeError: Triton Error [CUDA]: device kernel image is invalid

adefazio commented 3 months ago

Thanks for pointing this out, we will investigate

adefazio commented 2 months ago

I added that line and it ran without crashes on my machine, could you provide more information? Full logging output, and what version of PyTorch are you using?