[QAT Lora 6/N] torch.compile for reference implementation (#3343)

[QAT Lora 6/N] torch.compile for reference implementation (#3343) #1144