Only 4x speed seems rather low for GPU acceleration, does numpy already use AVX2... | Hacker News

Hacker News new | past | comments | ask | show | jobs | submit

login

moffkalast 31 days ago | parent | context | favorite | on: Nvidia adds native Python support to CUDA

Only 4x speed seems rather low for GPU acceleration, does numpy already use AVX2 or anything SIMD?

For comparison, doing something similar with torch on CPU and torch on GPU will get you like 100x speed difference.

diggan 31 days ago [–]

It's a microbenchmark (if even that), take it with a grain of salt. You'd probably see a bigger difference with bigger/more/more complicated tasks,

Join us for AI Startup School this June 16-17 in San Francisco!
Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact