xla

393cb707 - Add AMP test (#3057)

Add AMP test (#3057) * autocast test * finalize autocast test * linter * Use CUDA's autocast_list with custom unsupported op list Failure is unrelated, merge this change now(since GPU test passed)