We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent efeea50 commit 41725eeCopy full SHA for 41725ee
1 file changed
transformer_engine/plugin/core/backends/vendor/musa/musa_patches.py
@@ -45,6 +45,7 @@ def apply_patch() -> None:
45
# imports this module to run patches and that would cause a circular import.
46
try:
47
import transformer_engine
48
+
49
transformer_engine.TE_DEVICE_TYPE = "musa"
50
transformer_engine.TE_PLATFORM = torch.musa
51
except Exception as e:
0 commit comments