5 ms·
The weird part of the programming model is that threadblocks don't map 1:1 to warps or SMs. A single threadblock executes on a single SM, but each SM has multip
by jms55 2y ago
The weird part of the programming model is that threadblocks don't map 1:1 to warps or SMs. A single threadblock executes on a single SM, but each SM has multiple warps, and the threadblock could be the size of a single warp, or larger than the combined thread count of all warps in the SM.
So, how large do you make your threadblocks to get optimal SM/warp scheduling? Well it "depends" based on resource usage, divergence, etc. Basically run it, profile, switch the threadblock size, profile again, etc. Repeat on every GPU/platform (if you're programming for multiple GPU platforms and not just CUDA, like games do). It's a huge pain, and very sensitive to code changes.
People new to GPU programming ask me "how big do I make the threadblock size?" and I tell them go with 64 or 128 to start, and then profile and adjust as needed.
Two articles on the AMD side of things:
https://gpuopen.com/learn/occupancy-explained https://gpuopen.com/learn/occupancy-explained
https://gpuopen.com/learn/optimizing-gpu-occupancy-resource-usage-large-thread-groups https://gpuopen.com/learn/optimizing-gpu-occupancy-resource-...
- EarlKing 2y agoSounds like the sort of thing that would lend itself to runtime optimization.
- jms55 2y agoI'm not too informed on the details, but iirc drivers _do_ try and optimize shaders in the background, and then when ready swaps in a better version. But I doubt it does stuff like change threadgroup size, the programmer might assume a certain size and their shader would be broken if changed. Also drivers doing background work means unpredictable performance and stuttering, which developers really don't like. Someone correct me if I'm wrong, maybe drivers don't do this anymore.
- EarlKing 2y agoWell, if the user isn't going to be sharing the GPU with another task then you could push things back to install-time. In other words: At install time you conduct a benchmark on the relevant shaders, rewrite as necessary, recompile, and save the results accordingly. Now the user has a version of your shaders optimized to their particular configuration. Since installation times are already somewhat lengthy anyway you can be reasonably certain that no one is going to miss an extra minute or two needed to conduct benchmarks, especially if it results in installing optimized code.
- saagarjha 2y agoThis is how autotuning often works yes
- charles_irl 2y agoComing from the neural network world, rather than the shader world, but: I'd say you're absolutely right! Right now NNs and their workloads are changing quickly enough that people tend to prefer runtime optimization (like the dynamic/JIT compilation provided by Torch's compiler), but when you're confident you understand the workload and have the know-how, you can do static compilation (e.g. with ONNX, TensorRT). I work on a serverless infrastructure product that gets used for NN inference on GPUs, so we're very interested in ways to amortize as much of that compilation and configuration work as possible. Maybe someday we'll even have something like what Redshift has in their query engine -- pre-compiled binaries cached across users.
- lostmsu 2y agoThis reminds me of the dreaded Vulkan Shaders Compilation dialog when you try to play some games after driver update.
- terribleperson 2y agoPeople complain a lot about shader compilation, but shader compilation on start-up is much nicer than when a game doesn't do that ahead of time and does it when you need those shaders.
- amelius 2y agoBut which programming languages are most amenable to automatic runtime optimization? Should we go back to FORTRAN?
- EarlKing 2y agoThe sad answer is... probably none of them. Runtime optimization has always been one of those things that sends most programmers running away screaming, and those who make languages never seem to come from the ranks of those who understand the clear utility of it.
- morphle 2y agoSqueak Smalltalk has several automatic runtime optimizations and compilers like JIT, parallel load balancing compiler [1], adaptive compiler [2] and a metacircular simulator and byte code virtual machine written in itself that allows you to do runtime optimisations on GPUs. The byte codes are of course replaced with the native GPU instructions at runtime. There are dozens of scientific papers and active research is still being done [1]. I've worked on automatic parallel runtime optimizations and adaptive compilers since 1981. We make reconfigurable hardware (chips and wafers) that also optimises at runtime. Truffle/GraalVM is very rigid and overly complicated [6]. With a meta compiler like Ometa or Ohm we can give any programming language the runtime adaptive compilation for GPUs [3][4]. I'm currently adapting my adaptive compiler to Apple Silicon M4 GPU and neural engine to unlock the trillions of operations per second these chips can do. I can adapt them to more NVIDIA GPUs with the information of the website in the title. Thank you very much charles_irl! I would love to be able to save the whole website in a single PDF. I can optimise your GPU software a lot with my adaptive compilers. It will cost less than 100K in labour to speed up your GPU code by a factor 4-8 at least, sometimes I see 30-50 times speedup. [1] https://www.youtube.com/watch?v=wDhnjEQyuDk https://www.youtube.com/watch?v=wDhnjEQyuDk [2] https://www.youtube.com/watch?v=CfYnzVxdwZE https://www.youtube.com/watch?v=CfYnzVxdwZE [3] https://tinlizzie.org/~ohshima/shadama2/ https://tinlizzie.org/~ohshima/shadama2/ [4] https://github.com/yoshikiohshima/Shadama https://github.com/yoshikiohshima/Shadama [5] http://www.tinlizzie.org/ometa/ http://www.tinlizzie.org/ometa/ [6] https://github.com/NVIDIA/grcuda https://github.com/NVIDIA/grcuda
- bassp 2y agoI was taught that you want, usually, more threads per block than each SM can execute, because SMs context switch between threads (fancy hardware multi threading!) on memory read stalls to achieve super high throughput. There are, ofc, other concerns like register pressure that could affect the calculus, but if an SM is waiting on a memory read to proceed and doesn’t have any other threads available to run, you’re probably leaving perf on the table (iirc).
- saagarjha 2y agoPretty sure CUDA will limit your thread count to hardware constraints? You can’t just request a million threads.
- bassp 2y agoYou can request up to 1024-2048 threads per block depending on the gpu; each SM can execute between 32 and 128 threads at a time! So you can have a lot more threads assigned to an SM than the SM can run at once
- saagarjha 2y agoRight, ok. So you mean a handful of warps and not like a plethora of them for no reason.
- buildbot 2y agoThread counts per block are limited to 1024 (unless I’ve missed and change and wikipedia is wrong), but total threads per kernel is 1024(2^32-1)65535*65535 ~= 2^74 threads https://en.wikipedia.org/wiki/Thread_block_(CUDA_programming) https://en.wikipedia.org/wiki/Thread_block_(CUDA_programming...
- saagarjha 2y agoYeah I’m talking about the limit per-block.
- 2y ago
- charles_irl 2y ago100% -- there's basically no substitue for benchmarking! I find the empiricism kind of comforting, coming from a research science background. IIUC, even CuBLAS basically just uses a bunch of heuristics that are mostly derived from benchmarking to decide with kernels to use.
- einpoklum 2y ago> It's a huge pain, and very sensitive to code changes. Optimization is very often like that. Making things generic, uniform and simple typically has a performance penalty - and you use your GPU because you care about that stuff.