Set temporary opencl device to PrivateUse1 dispatch key

artyom-beilis · artyom-beilis · commit eb74af18af6e · 2021-10-08T20:58:42.000+03:00
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,

Original file line number	Diff line number	Diff line change
`@@ -649,9 +649,10 @@ inline DispatchKey computeDispatchKey(`
`649`	`649`	`}`
`650`	`650`	`return DispatchKey::XPU;`
`651`	`651`	`}`
	`652`	`+ case DeviceType::OPENCL:`
	`653`	`+ return DispatchKey::PrivateUse1;`
`652`	`654`	`case DeviceType::MKLDNN:`
`653`	`655`	`case DeviceType::OPENGL:`
`654`		`- case DeviceType::OPENCL:`
`655`	`656`	`case DeviceType::IDEEP:`
`656`	`657`	`TORCH_INTERNAL_ASSERT(`
`657`	`658`	`0,`