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 2525# *****************************************************************************
2626
2727import os
28+ import sys
2829
2930mypath = os .path .dirname (os .path .realpath (__file__ ))
3031
4546 if hasattr (os , "add_dll_directory" ):
4647 os .add_dll_directory (mypath )
4748 os .add_dll_directory (dpctlpath )
49+
4850 os .environ ["PATH" ] = os .pathsep .join (
4951 [os .getenv ("PATH" , "" ), mypath , dpctlpath ]
5052 )
5153
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+
5265# Borrowed from DPCTL
5366from dpctl .tensor import DLDeviceType
5467
You can’t perform that action at this time.
0 commit comments