search

huggingface / optimum-habana

Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)
Apache License 2.0
153 stars 202 forks source link

[SW-200913] Removed workaround for NaN bug causing graph break. #1516

Open astachowiczhabana opened 13 hours ago

astachowiczhabana commented 13 hours ago

Problem: graph brake is introduced by get_autocast_hpu_dtype call

Solution: this line is no longer needed as autocast logic will be executed in GaudiTrainer, therefore can be removed without breaking anything.

astachowiczhabana commented 13 hours ago

Hi @libinta this commit is also required with next OH release