From b80fd389b3e4d11a02188571e5fee0f5a83a0258 Mon Sep 17 00:00:00 2001 From: Bhargav Date: Fri, 9 Feb 2024 13:24:10 +0200 Subject: [PATCH] Changing backend name --- optimum/habana/accelerate/utils/dataclasses.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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