test_fused_transformer_with_amp_decorator.py 2.3 KB