Skip to content

Commit

Permalink
Set temporary opencl device to PrivateUse1 dispatch key
Browse files Browse the repository at this point in the history
  • Loading branch information
artyom-beilis committed Oct 8, 2021
1 parent 4cdfced commit eb74af1
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion c10/core/TensorOptions.h
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit eb74af1

Please sign in to comment.