Set temporary opencl device to PrivateUse1 dispatch key

artyom-beilis · Oct 8, 2021 · eb74af1 · eb74af1
1 parent 4cdfced
commit eb74af1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/c10/core/TensorOptions.h b/c10/core/TensorOptions.h
@@ -649,9 +649,10 @@ inline DispatchKey computeDispatchKey(
           }
           return DispatchKey::XPU;
         }
+        case DeviceType::OPENCL:
+            return DispatchKey::PrivateUse1;
         case DeviceType::MKLDNN:
         case DeviceType::OPENGL:
-        case DeviceType::OPENCL:
         case DeviceType::IDEEP:
           TORCH_INTERNAL_ASSERT(
               0,