No, torch.compile
does not require amp
.
Just to understand the issue a bit better: are you seeing differences in the outputs for repeated forward passes after calling torch.compile(model)
or are you comparing the plain eager mode model vs. the compiled one?