| { | |
| "use_torch_autocast": true, | |
| "use_fused_adam": true, | |
| "use_fused_clip_norm": true, | |
| "autocast_bf16_ops": [ | |
| "_convolution.deprecated", | |
| "_convolution", | |
| "conv1d", | |
| "conv2d", | |
| "conv3d", | |
| "conv_tbc", | |
| "conv_transpose1d", | |
| "conv_transpose2d.input", | |
| "conv_transpose3d.input", | |
| "convolution", | |
| "prelu", | |
| "addmm", | |
| "addmv", | |
| "addr", | |
| "matmul", | |
| "einsum", | |
| "mm", | |
| "mv", | |
| "silu", | |
| "linear", | |
| "addbmm", | |
| "baddbmm", | |
| "bmm", | |
| "chain_matmul", | |
| "linalg_multi_dot", | |
| "layer_norm", | |
| "group_norm" | |
| ], | |
| "autocast_fp32_ops": [ | |
| "acos", | |
| "asin", | |
| "cosh", | |
| "erfinv", | |
| "exp", | |
| "expm1", | |
| "log", | |
| "log10", | |
| "log2", | |
| "log1p", | |
| "reciprocal", | |
| "rsqrt", | |
| "sinh", | |
| "tan", | |
| "pow.Tensor_Scalar", | |
| "pow.Tensor_Tensor", | |
| "pow.Scalar", | |
| "softplus", | |
| "frobenius_norm", | |
| "frobenius_norm.dim", | |
| "nuclear_norm", | |
| "nuclear_norm.dim", | |
| "cosine_similarity", | |
| "poisson_nll_loss", | |
| "cosine_embedding_loss", | |
| "nll_loss", | |
| "nll_loss2d", | |
| "hinge_embedding_loss", | |
| "kl_div", | |
| "l1_loss", | |
| "smooth_l1_loss", | |
| "huber_loss", | |
| "mse_loss", | |
| "margin_ranking_loss", | |
| "multilabel_margin_loss", | |
| "soft_margin_loss", | |
| "triplet_margin_loss", | |
| "multi_margin_loss", | |
| "binary_cross_entropy_with_logits", | |
| "dist", | |
| "pdist", | |
| "cdist", | |
| "renorm", | |
| "logsumexp" | |
| ] | |
| } |