OpenCL support: fix bug, adding new requirements files

2019-02-19 20:48:39 +04:00
parent 72ba6b103c
commit b939fe9d85
4 changed files with 18 additions and 6 deletions
@@ -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,