You are viewing a plain text version of this content. The canonical link for it is here.

Posted to discuss-archive@tvm.apache.org by Akhil via Apache TVM Discuss <no...@discuss.tvm.ai> on 2020/09/05 03:26:00 UTC

[Apache TVM Discuss] [Questions] How does TVM choose the platform and device


Hey @jcf94,

 I was looking into the opencl_device_type.cc file and noticed that depending on which context was used in the python code like (tvm.cpu, tvm.gpu, tvm.opencl) it chooses a device type and sets the dtype accordingly.

When exactly is the dtype set to the CL_DEVICE_TYPE_ACCELERATOR, is there any way to specify it with the opencl context as opencl could be targeted to either cpu or gpu ?

Also I have noticed that is is the init fucntion that is responsible for sending the device type to this function, but from what i have noticed it seems like there is no argument being passed when init is being called ?





---
[Visit Topic](https://discuss.tvm.apache.org/t/how-does-tvm-choose-the-platform-and-device/7695/5) to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/4daf3fe4424a9e3e1cf9570c8ef0be0ae1534fe290bb1e1c02d91499e9121833).