gpu.kernel doesn't work.

exaloop / codon

A high-performance, zero-overhead, extensible Python compiler using LLVM

https://docs.exaloop.io/codon

Other

15.01k stars 517 forks source link

gpu.kernel doesn't work. #399

Closed udnaan closed 1 year ago

udnaan commented 1 year ago

Built from source (master) with DCODON_GPU=ON.

Getting following error when simply trying to import and apply the decorator gpu.kernel: gpu.py:3:12-22: error: cannot import name 'kernel' from ''

arshajii commented 1 year ago

Hi @udnaan -- it looks like your file is called gpu.py which clashes with the gpu import. Can you try renaming the file and seeing if that resolves it?

udnaan commented 1 year ago

argh rookie mistake. 🤦

all good now for the build

now getting CUDA error at codon/codon/codon/runtime/gpu.cpp:42: a PTX JIT compilation failed