3 ms·
This seems similar to semi-structured (aka 2:4) sparsity, may be worth explicitly comparing. As far as I can tell by skimming, your technique: - is optimized f
by hatthew 3y ago
This seems similar to semi-structured (aka 2:4) sparsity, may be worth explicitly comparing. As far as I can tell by skimming, your technique:
- is optimized for apple silicon
- ~2x speed at 75% sparsity
- dynamic, depends on input, applied at runtime
- can choose amount of sparsity
And 2:4 semi-structured sparsity:
- is optimized for GPUs with sparse tensor cores (nvidia ampere and beyond)
- ~2x speed at 50% sparsity
- static, applied to the model at rest
- probably worse results than your technique at 50% sparsity
The interesting comparison I'd want to see is semi-structured sparsity results (50% sparsity, 2x speedup) vs your results at 75% sparsity (2x speedup).
- kolinko 3y agoThanks for checking it out! I also can’t wait to have more tests done. Also, I chose Apple Silicon because it was easier for me to develop on. It is possible that the algorithm would also have good performance on other architectures.