Open arogozhnikov opened 6 years ago
@arogozhnikov TC indeed does not support CPU atm, this is work in progress.
Hi @arogozhnikov , thanks for reporting. We don't support CPU computations in TC right now. However, adding a simple check for is_cuda()
should definitely be possible. I'll re-title this issue to reflect that more properly. Thank you :)
When submitting a bug report, please include the following information (where relevant):
Commit hash of our repo and submodules (if compiling from source): NA
TensorComprehensions fail when given CPU tensors (and drops the kernel). Here is an example:
Result:
It also drops python kernel if you first optimize function and then try to run it with CPU tensors.
PS. Is there a plan to support CPU computations? If not, a simple check like is_cuda should be enough