{ | |
"use_habana_mixed_precision": true, | |
"hmp_opt_level": "O1", | |
"hmp_is_verbose": false, | |
"hmp_bf16_ops": [ | |
"addmm", | |
"addcmul", | |
"bmm", | |
"dot", | |
"mm", | |
"conv2d", | |
"layer_norm", | |
"linear", | |
"matmul", | |
"mv" | |
], | |
"hmp_fp32_ops": [ | |
"batch_norm", | |
"cross_entropy", | |
"log_softmax", | |
"nll_loss", | |
"softmax", | |
"topk" | |
] | |
} | |