Load _C ops before registering meta implementations (#8470)

pytorch · Jun 6, 2024 · a839642 · a839642
1 parent bcf6cda
commit a839642
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/torchvision/__init__.py b/torchvision/__init__.py
@@ -3,9 +3,11 @@
 from modulefinder import Module
 
 import torch
-from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils
 
-from .extension import _HAS_OPS
+# Don't re-order these, we need to load the _C extension (done when importing
+# .extensions) before entering _meta_registrations.
+from .extension import _HAS_OPS  # usort:skip
+from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils  # usort:skip
 
 try:
     from .version import __version__  # noqa: F401