blob: d7351f4c52d473a77c4a9c33ce365d07a2182428 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
import os
import sys
import re
__all__ = ['uses_accelerate_framework', 'get_sgemv_fix']
def uses_accelerate_framework(info):
""" Returns True if Accelerate framework is used for BLAS/LAPACK """
if sys.platform != "darwin":
return False
r_accelerate = re.compile("Accelerate")
extra_link_args = info.get('extra_link_args', '')
for arg in extra_link_args:
if r_accelerate.search(arg):
return True
return False
def get_sgemv_fix():
""" Returns source file needed to correct SGEMV """
path = os.path.abspath(os.path.dirname(__file__))
return [os.path.join(path, 'src', 'apple_sgemv_fix.c')]
|
