6 ms·
Matrix-vector multiplication implemented in off-the-shelf DRAM for Low-Bit LLMs
- robwwilliams 1y agoThis is just mind-bendingly weird and wonderfully creative. It can pay to work in the weeds! Bravo.
- userbinator 1y agoThis behaviour has been around since the earliest DRAMs with multiplexed row/column addresses. The Mostek MK4096 of 1973 could probably do this. Only took about half a century for someone to figure it out.
- Bolwin 1y agoThey're doing matrix operations in the Dram itself? That sounds insane and also fascinating
- summarity 1y agoGetting LLM inference running on any thing is going to be the next “it runs Doom”
- nkurz 1y agoYup, and incredibly they are able to do this on standard RAM by "intentionally violating the timing parameters": Processing-Using-DRAM (PUD) leverages the inherent analog operational characteristics of DRAM to enable highly parallel bit-serial computations directly within memory arrays. Prior research has demonstrated that commercial off-the-shelf DRAM can achieve PUD functionality without hardware modifications by intentionally violating the timing parameters. These studies have established two fundamental PUD operations: RowCopy and majority-of-X (MAJX) (Fig. 1). The RowCopy operation facilitates data movement between different rows within a subarray by issuing a PRE command followed immediately by an ACT command before bitline precharging completes, enabling data transfer through the bitlines. This operation affects all cells along a row simultaneously, making it approximately 100 times faster than processor-mediated data movement. The MAJX operation performs a majority vote among X cells sharing the same bitline that are activated simultaneously, implemented in commercial DRAM by issuing ACT, PRE, and ACT commands in rapid succession without delays. This allows concurrent activation of 2∼32 rows. MAJX enables bit-serial computations that leverage the parallelism of subarrays with 65,536 columns, serving as the fundamental computational unit for PUD.
- nayuki 1y agoThis kind of low-level protocol manipulation of DRAM has some similarities to rowhammer attacks.
- gwern 1y agoCan it be used to covertly run computations invisible to the OS or CPU?
- nsteel 1y agoThis research requires a custom memory controller that's doing "weird" things, the CPU isn't really getting involved here. It's very different compared to row hammer in my opinion. If you have a custom memory controller then I think all bets are off.
- wtallis 1y agoOnly to the same extent that any other co-processor add-in card can do stuff that's not observable by the CPU. Your CPU's RAM is managed by the CPU's memory controller hardware, and that memory controller does not give software the ability to issue individual DRAM commands like precharge. This research uses a memory controller implemented on a FPGA, talking to its own pool of RAM.
- elcritch 1y agoI hope Micron or another commercial player builds a product on this!
- tamlin 1y agoSamsung and SK-Hynix have had specs and papers for a few years already for HBM and GDDR. e.g. * https://www.servethehome.com/sk-hynix-ai-memory-at-hot-chips-2023/ https://www.servethehome.com/sk-hynix-ai-memory-at-hot-chips... * https://www.servethehome.com/samsung-processing-in-memory-technology-at-hot-chips-2023/ https://www.servethehome.com/samsung-processing-in-memory-te... Not sure anyone has started using it in production.
- walterbell 1y ago> By intentionally issuing DRAM commands that violate manufacturer-specified timing parameters.. [gaining] massive parallelism up to 65,536 bitwise operations in parallel. Take that, binary blobs for DRAM training!
- willvarfar 1y agoCan we expect to see matrix multiplication and perhaps other ops move from classic CPUs out into the DRAM, perhaps with deliberate hardware support? And does such a processing shift give advantage to Samsung etc? Where does this leave NVIDIA etc?
- imtringued 1y agoYour questions are kind of amusing since Apple will use LPDDR6-PIM on the next generation of iPhones. https://www.patentlyapple.com/2024/12/apple-plans-to-transition-to-the-next-generation-memory-standard-lpddr6-pim-for-the-iphone-to-support-ai-in-2026.html https://www.patentlyapple.com/2024/12/apple-plans-to-transit...
- deleted 1y ago[deleted]
- nsteel 1y agoI don't get it, what's the joke?
- userbinator 1y agoDid anyone else notice the absolutely insane author lists of references 1 and 3? I was expecting to find this 2016 article in there: https://news.ycombinator.com/item?id=12469270 https://news.ycombinator.com/item?id=12469270 This 2019 one does show up: https://news.ycombinator.com/item?id=22712811 https://news.ycombinator.com/item?id=22712811 Of course, this "out of spec" behaviour of DRAM, more specifically the ability to do copying, is also implicated in this infamous bug: https://news.ycombinator.com/item?id=5314959 https://news.ycombinator.com/item?id=5314959 It seems more than one person independently observed such a thing, and thought "this might be a useful behaviour".
- s-macke 1y agoThis seems to be a formatting error. For such a huge author list, you usually write only the first name and then "et al." for "others".
- tomsmeding 1y agoThe 'et al.' is used for in-article citations, if done in author-year format; references in the reference list are, to the extent that I've seen, always written out in full. I guess Google just wanted to make the life of any academic citing their work miserable. There are (unfortunately) conferences that have page limits that include the reference list; I wonder if an exception would be made here.
- esafak 1y agoThey want authors to think twice before citing someone. A curious incentive!
- throwaway519 1y agoOne day, I'm going to credit my entire department, deli and everyone in the park at 2pm as contributors too.
- swimwiththebeat 1y agoSo is this a new technique of doing computations within existing DRAM to overcome the memory wall issue of modern computing?
- cpldcpu 1y agoSome more background information: One of the original proposals for in-DRAM compute: https://users.ece.cmu.edu/~omutlu/pub/in-DRAM-bulk-AND-OR-ieee_cal15.pdf https://users.ece.cmu.edu/~omutlu/pub/in-DRAM-bulk-AND-OR-ie... First demonstration with off-the-shelf parts: https://parallel.princeton.edu/papers/micro19-gao.pdf https://parallel.princeton.edu/papers/micro19-gao.pdf DRAM Bender, the tool they are using to implement this: https://github.com/CMU-SAFARI/DRAM-Bender https://github.com/CMU-SAFARI/DRAM-Bender Memory-Centric Computing: Recent Advances in Processing-in-DRAMhttps://arxiv.org/abs/2412.19275 https://arxiv.org/abs/2412.19275
- xhkkffbf 1y agoIn-DRAM goes back a long time. There were plenty of papers in the 90s about various ideas for turning a bank of DRAM into a SIMD machine. They weren't as clever as some of these ideas or as well developed but these papers are just the latest versions of an old idea.
- therealcamino 1y agoDo any of those techniques use unmodified DRAM or are you talking about processor-in-memory approaches?
- dr_zoidberg 1y agoThe abstract of OPs link mentions "Processing-Using-DRAM (PUD)" as exactly that, using off the shelf components. I do wonder how they achieve that, I guess fiddling with the controller in ways that are not standard but get the job (processing data in memory) done. Edit: Oh and cpldcpu linked the ComputeDRAM paper that explains how to do it with off the shelf parts.
- jiveturkey 1y agoThat context is very helpful. But you don't need to poo-poo the ideas as "just another iteration". Everything we have today is built on top of decades of prior work. The paper itself mentions a lot of prior work.
- morphle 1y agoA bit unscientific that they don't cite the original Intelligent RAM (IRAM) sources from 1997: https://scholar.google.com/scholar?hl=en&as_sdt=0%2C5&q=iram+patterson&btnG= https://scholar.google.com/scholar?hl=en&as_sdt=0%2C5&q=iram...
- cpldcpu 1y agoI also strongly suspect that there are earlier sources. However, IRAM looks like compute near memory where they will add an ALU to the memory chip. compute in memory is about using the memory array itself. To be fair, CIM looked much less appealing before the advent of deep-learning with crazy vector lengths. So people rather tried to build something that allows more fine grained control of the operations.
- morphle 1y ago>I also strongly suspect that there are earlier sources. You are right, I remember 1972-ish papers where they did compute in memory. I just couldn't locate links to these papers in a few minutes.
- xiphias2 1y agoThis woule be a cool way to make a cheap inferencing device for the largest LLMs
- protocolture 1y ago>General matrix-vector multiplication (GeMV) Ok, so my math isnt great. When I was studying Quaternions during my 3d math class (That I failed the first time, like I said, not a math guy) they briefly covered the history of matrix calculation in graphics development. My understanding is that Quaternions became popular because they are almost as accurate as matrices but much less complex computationally. Has anyone tried building an LLM using Quats instead of matrices? Or are the optimisations with Quaternions more useful in realtime?
- monocasa 1y agoMy understanding was that the main benefit of quaternions in computer graphics was representing rotations in a way that doesn't result in gimble lock. And beyond that, for those rotations, a quaternion doesn't scale nearly as well as you add dimensions. Complex numbers are a complex representation of two space, quaternions are a complex representation of three space, and to go to four space you have octonions, which have eight elements.
- eru 1y agoQuaternions have four dimensions.
- thomaskoopman 1y agoYes, but quaternions of unit length are a representation of the rotation group in 3D space ( https://en.wikipedia.org/wiki/Representation_theory_of_SU(2)#Relation_to_the_representations_of_SO(3) https://en.wikipedia.org/wiki/Representation_theory_of_SU(2)... ), which is how they are used for rotations.
- suspended_state 1y agoThe original question was: can quaternions be used in place of matrices to perform LLMs tasks, and the answer is: quaternions are 4 dimensions, with the implied meaning that matrices can cover different dimentionalities, which are needed for LLMs (and neural network in general).
- chasd00 1y agoIn the hardware world are there risks of taking advantage of a bug knowing that the manufacturer may someday fix the bug? I know in the software world it's a bad idea to leverage a bug in a platform to enable a feature (or fix another bug). The bug you're counting on being present may get fixed 15 years in the future and then your system explodes and no one knows why. edit: seems like there was a recent discussion about something similar... undefined behavior in some C function iirc
- vlovich123 1y agoUndefined behavior in C/C++ has been discussed for a very very long time. I'd say the impact of it when combined with optimizing compilers first came to broader public awareness around the 2010ish time frame (maybe 2013?) which is now about 12+ years old. As for this paper, it's not about relying on a bug but rather presenting what might be possible with DRAM in the hopes of standardizing capabilities.
- alexpotato 1y agoThis pops up in low latency HFT specifically with networking cards. Certain network cards have either a bug or combination of features that work in an interesting way to the benefit of the trading firm. These bugs (and features too) sometimes get removed in favor of either getting rid of the bug or those features are seen as not needed for the larger market etc. Therefore, firms will sometimes attempt to buy up all available supply of certain models.
- nomel 1y agoThis usually falls under "interoperability testing", but usually mitigated through your firmware rather than hardware. In the worst types of cases, you need to make sure your hardware works with some popular defunct vendor from 15 years ago since some big customers have used that hardware for 15 years, without issue, and will see your hardware as the problem when they plug it in. For communication equipment, this is super important, with all sorts of "quirks" put in for vendors that didn't follow the spec. And, that includes keeping quirks in your firmware, so you don't break anyone else's. Imagine entire walls of legacy and long-gone and current competitor equipment, with robot arms to plug things in, and you have an idea of what some hardware validation labs look like. Motherboard manufacturer firmware is also filled with quirks for specific CPUs, chipsets, etc.
- lolc 1y agoFunny hack. Without having read the paper I'd assume the operations to be thermally unstable. So LLM inference results will vary based on environmental temperature :-)