Skip to content

Commit 656f9d0

Browse files
committed
Add Accelerate framework blas__ldflags tests
1 parent 6132203 commit 656f9d0

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

pytensor/link/c/cmodule.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2873,9 +2873,18 @@ def check_libs(
28732873
)
28742874
except Exception as e:
28752875
_logger.debug(e)
2876+
try:
2877+
# 3. Mac Accelerate framework
2878+
_logger.debug("Checking Accelerate framework")
2879+
validated_flags = try_blas_flag(["-framework", "Accelerate", "-rpath", rpath])
2880+
if validated_flags == "":
2881+
raise Exception("Accelerate framework flag failed ")
2882+
return validated_flags
2883+
except Exception as e:
2884+
_logger.debug(e)
28762885
try:
28772886
_logger.debug("Checking Lapack + blas")
2878-
# 3. Try to use LAPACK + BLAS
2887+
# 4. Try to use LAPACK + BLAS
28792888
return check_libs(
28802889
all_libs,
28812890
required_libs=["lapack", "blas", "cblas", "m"],
@@ -2885,7 +2894,7 @@ def check_libs(
28852894
except Exception as e:
28862895
_logger.debug(e)
28872896
try:
2888-
# 4. Try to use BLAS alone
2897+
# 5. Try to use BLAS alone
28892898
_logger.debug("Checking blas alone")
28902899
return check_libs(
28912900
all_libs,
@@ -2896,7 +2905,7 @@ def check_libs(
28962905
except Exception as e:
28972906
_logger.debug(e)
28982907
try:
2899-
# 5. Try to use openblas
2908+
# 6. Try to use openblas
29002909
_logger.debug("Checking openblas")
29012910
return check_libs(
29022911
all_libs,

0 commit comments

Comments
 (0)