Preston Jiang

icon indicating a website link https://lpjiang97.github.io icon indicating an email [email protected]

icon company University of Washington icon location Seattle, WA icon location PhD student in Computer Science at the UW. Interested in theoretical neuroscience.

Results 2 comments of Preston Jiang

Slower performance on GPU than CPU with Zygote

I tried to use a larger size for the tensors, but the `CuArray` just takes too long to run. So it does look like the runtime gets worse with size....

Slower performance on GPU than CPU with Zygote

Thanks for testing! I did rewrite the code with reshape and batched_mul, which is faster on the GPUs. I guess I will leave this open for now :)