File tree Expand file tree Collapse file tree 1 file changed +13
-0
lines changed Expand file tree Collapse file tree 1 file changed +13
-0
lines changed Original file line number Diff line number Diff line change 25
25
# *****************************************************************************
26
26
27
27
import os
28
+ import sys
28
29
29
30
mypath = os .path .dirname (os .path .realpath (__file__ ))
30
31
45
46
if hasattr (os , "add_dll_directory" ):
46
47
os .add_dll_directory (mypath )
47
48
os .add_dll_directory (dpctlpath )
49
+
48
50
os .environ ["PATH" ] = os .pathsep .join (
49
51
[os .getenv ("PATH" , "" ), mypath , dpctlpath ]
50
52
)
51
53
54
+ # For virtual environments on Windows, add folder with DPC++ libraries
55
+ # to the DLL search path
56
+ if sys .base_exec_prefix != sys .exec_prefix and os .path .isfile (
57
+ os .path .join (sys .exec_prefix , "pyvenv.cfg" )
58
+ ):
59
+ dll_path = os .path .join (sys .exec_prefix , "Library" , "bin" )
60
+ if os .path .isdir (dll_path ):
61
+ os .environ ["PATH" ] = os .pathsep .join (
62
+ [os .getenv ("PATH" , "" ), dll_path ]
63
+ )
64
+
52
65
# Borrowed from DPCTL
53
66
from dpctl .tensor import DLDeviceType
54
67
You can’t perform that action at this time.
0 commit comments