Update Pytorch AMP supported kernels logic #3396
+150
−49
Merged
Loading