gaudi_config.json

{
  "use_habana_mixed_precision": true,
  "hmp_is_verbose": false,
  "use_fused_adam": true,
  "use_fused_clip_norm": true,
  "hmp_bf16_ops": [
    "addmm",
    "batch_norm",
    "bmm",
    "conv1d",
    "conv2d",
    "conv3d",
    "conv_transpose1d",
    "conv_transpose2d",
    "conv_transpose3d",
    "dot",
    "dropout",
    "dropout1d",
    "dropout2d",
    "dropout3d",
    "group_norm",
    "instance_norm",
    "layer_norm",
    "leaky_relu",
    "linear",
    "matmul",
    "mean",
    "mm",
    "mv",
    "relu",
    "t"
  ],
  "hmp_fp32_ops": [
    "binary_cross_entropy",
    "binary_cross_entropy_with_logits",
    "cross_entropy",
    "div",
    "divide",
    "embedding",
    "embedding_bag",
    "log",
    "log2",
    "log_softmax",
    "nll_loss",
    "smooth_l1_loss",
    "softmax",
    "topk",
    "truediv"
  ]
}