pytorch/caffe2/python/fakelowp/init_shared_libs.py
Yinghai Lu eb00bac2b5 Make FakeLowP tests work (#36525)
Summary:
Make the e2e FakeLowP python tests work with Glow lowering in OSS environment. Added a README.md as a guideline.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36525

Reviewed By: hyuen

Differential Revision: D21004706

Pulled By: yinghai

fbshipit-source-id: d182152e4a1a3368640bd7872cb9ea4d4bff4b02
2020-04-13 20:16:33 -07:00

11 lines
274 B
Python

from __future__ import absolute_import, division, print_function, unicode_literals
import ctypes
import os
if 'OSS_ONNXIFI_LIB' in os.environ:
lib = os.environ['OSS_ONNXIFI_LIB']
print("Loading ONNXIFI lib: ".format(lib))
ctypes.CDLL(lib, ctypes.RTLD_GLOBAL)