Add dummy torch.hpu

HabanaAI · Sep 3, 2024 · b5b6f8c · b5b6f8c
1 parent c9babcc
commit b5b6f8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vllm/cpu_migration.py b/vllm/cpu_migration.py
@@ -13,8 +13,8 @@
 import habana_frameworks.torch as htorch
 import torch
 
-# torch.hpu = sys.modules['torch.hpu'] = types.ModuleType('torch.hpu')
-# torch.hpu.synchronize = lambda: print('calling synchronize')
+torch.hpu = sys.modules['torch.hpu'] = types.ModuleType('torch.hpu')
+torch.hpu.synchronize = lambda: print('calling synchronize')
 
 class CpuMigration:
     def __init__(self):