 when I try to run the code using cpu, it will works but when I try to use it with gpu, an error will occurs