OpenCL support: fix bug, adding new requirements files

This commit is contained in:
iperov committed 2019-02-19 20:48:39 +04:00
1 parent 72ba6b103c
commit b939fe9d85
4 files changed
+18 -6

No files matched your search

+2
View File
@@ -270,6 +270,8 @@ try:
ctx = plaidml.Context()
for d in plaidml.devices(ctx, return_all=True)[0]:
details = json.loads(d.details)
if details['type'] == 'CPU': #skipping opencl-CPU
continue
if 'nvidia' in details['vendor'].lower():
has_nvidia_device = True
plaidML_devices += [ {'id':d.id,