Construct Vulkan device.
(self, dev_id=0)
| 229 | return self.device(DLDeviceType.kDLOpenCL, dev_id) |
| 230 | |
| 231 | def vulkan(self, dev_id=0): |
| 232 | """Construct Vulkan device.""" |
| 233 | return self.device(DLDeviceType.kDLVulkan, dev_id) |
| 234 | |
| 235 | def metal(self, dev_id=0): |
| 236 | """Construct Metal device.""" |