Skip to content

Commit a839642

Browse files
authored
Load _C ops before registering meta implementations (#8470)
1 parent bcf6cda commit a839642

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

torchvision/__init__.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,11 @@
33
from modulefinder import Module
44

55
import torch
6-
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils
76

8-
from .extension import _HAS_OPS
7+
# Don't re-order these, we need to load the _C extension (done when importing
8+
# .extensions) before entering _meta_registrations.
9+
from .extension import _HAS_OPS # usort:skip
10+
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
911

1012
try:
1113
from .version import __version__ # noqa: F401

0 commit comments

Comments
 (0)