Skip to content

Instantly share code, notes, and snippets.

@inducer

inducer/numpy-openblas-get-config.py

Created February 24, 2021 17:07
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save inducer/2f8490b82128b0591fecf791edef66a6 to your computer and use it in GitHub Desktop.
Save inducer/2f8490b82128b0591fecf791edef66a6 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
numpy-openblas-get-config.py
# based on https://github.com/numpy/numpy/issues/18471#issuecomment-785019387
import numpy
import os
import ctypes
def get_ilp64():
if os.environ.get("NPY_USE_BLAS_ILP64", "0") == "0":
return None
if IS_32BIT:
raise RuntimeError("NPY_USE_BLAS_ILP64 set on 32-bit arch")
return "64_"
ilp64 = get_ilp64()
dll = ctypes.CDLL(numpy.core._multiarray_umath.__file__)
if ilp64 == "64_":
get_config = dll.openblas_get_config64_
else:
get_config = dll.openblas_get_config
get_config.restype = ctypes.c_char_p
res = get_config()
print('OpenBLAS get_config returned', str(res))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment