search

microsoft / knossos-ksc

Compiler with automatic differentiation
Other
45 stars 10 forks source link

Use "inference_mode" for PyTorch inference? #910

Open awf opened 3 years ago

awf commented 3 years ago

I see discussion of "inference_mode" at https://pytorch.org/docs/stable/generated/torch.autograd.inference_mode.html#torch.autograd.inference_mode

I guess we should do this in pytest-bench?