Fix lowercasing of kernel
This commit is contained in:
parent
1271237d89
commit
e99a905d7c
|
@ -9,7 +9,7 @@ DEVICE_BATCH_SIZE_MAP = [(14, 16), (10,8), (7,4)]
|
||||||
from inspect import currentframe, getframeinfo
|
from inspect import currentframe, getframeinfo
|
||||||
import gc
|
import gc
|
||||||
|
|
||||||
is_WSL = 'wsl' in os.popen("uname -r").read().tolower()
|
is_WSL = 'wsl' in os.popen("uname -r").read().lower()
|
||||||
|
|
||||||
def xpu_get_mem(device=0):
|
def xpu_get_mem(device=0):
|
||||||
total_memory = ipex.xpu.get_device_properties(device).total_memory
|
total_memory = ipex.xpu.get_device_properties(device).total_memory
|
||||||
|
|
Loading…
Reference in New Issue
Block a user