multipy/runtime: mock _decomp

Author: d4l3k

multipy/runtime/interpreter/interpreter_impl.cpp

@@ -125,8 +125,16 @@ import importlib.abc
 import linecache
 from zipfile import ZipFile
 
+class DummyMultiPyModule:
+    def __getattr__(self, key):
+        return self
+
+    def __call__(self, *args, **kwargs):
+        return self
+
 # Disable Python library registration since it's not compatible with multipy.
-sys.modules["torch._meta_registrations"] = object
+sys.modules["torch._meta_registrations"] = DummyMultiPyModule()
+sys.modules["torch._decomp"] = DummyMultiPyModule()
 
 class RegisterModuleImporter(importlib.abc.InspectLoader):
     def __init__(self, find_module_source):
@@ -160,6 +168,8 @@ class RegisterModuleImporter(importlib.abc.InspectLoader):
 # print("modules:", sys.modules)
 
 import torch # has to be done serially otherwise things will segfault
+torch._decomp = DummyMultiPyModule()
+
 import multipy.utils
 try:
   import torch.version # for some reason torch doesn't import this and cuda fails?