diff --git a/optimum/habana/accelerate/utils/dataclasses.py b/optimum/habana/accelerate/utils/dataclasses.py index 07e256372f..c0484e2243 100644 --- a/optimum/habana/accelerate/utils/dataclasses.py +++ b/optimum/habana/accelerate/utils/dataclasses.py @@ -73,7 +73,8 @@ class GaudiDynamoBackend(str, BaseEnum): - **IPEX** -- Uses IPEX for inference on CPU. Inference only. [Read more](https://github.com/intel/intel-extension-for-pytorch). - **TVM** -- Uses Apach TVM for inference optimizations. [Read more](https://tvm.apache.org/) - - **AOT_HPU_TRAINING_BACKEND** -- Uses Habana Gaudi. + - **AOT_HPU_TRAINING_BACKEND** -- Uses Habana Gaudi - depracated - will be removed. + - **HPU_BACKEND** -- Uses Habana Gaudi. """ @@ -92,6 +93,7 @@ class GaudiDynamoBackend(str, BaseEnum): IPEX = "IPEX" TVM = "TVM" AOT_HPU_TRAINING_BACKEND = "AOT_HPU_TRAINING_BACKEND" + HPU_BACKEND = "HPU_BACKEND" @dataclass