torch-mlir/frontends/pytorch/test/extension_coexistence.py

19 lines
408 B
Python

# -*- Python -*-
# This file is licensed under a pytorch-style license
# See frontends/pytorch/LICENSE for license information.
# Some checks that we can import the various extensions and libraries and
# not have symbol collisions or other goings on.
# RUN: %PYTHON %s
import sys
print(f"PYTHONPATH={sys.path}")
import mlir
import npcomp
import _npcomp
import _torch_mlir
print("Extensions all loaded")