News for hackers. Stuff that matters.
Re:Using mmap to make LLaMA load faster by londons_explore on Wednesday April 05, 2023 @12:47PM (#35456989)
In this case, the main benefit is from multiple invocations of the same program.

Using mmap, you avoid doing any work at all the 2nd time you load the file.

Re:Using mmap to make LLaMA load faster by bigodanktime on Wednesday April 05, 2023 @12:54PM (#35457092)
What do you mean by work. The underlying page cache will keep much of the data actual cached if it's recent. Even databases like PostGreSQL use this to their advantage (https://github.com/postgres/postgres/blob/master/src/backend...).
Re:Using mmap to make LLaMA load faster by astrange on Wednesday April 05, 2023 @01:16PM (#35457419)
Copying the file backed pages to heap memory and possibly having to swap them out.
Re:Using mmap to make LLaMA load faster by bigodanktime on Wednesday April 05, 2023 @01:25PM (#35457557)
I may have parsed your statement incorrectly, but I'm assuming you are talking about the copy of data when using either mmap or File IO (memcpy versus write) Whether you do File IO versus mmap, there's going to be copy. With files, the copy occurs within kernel space with data being copied into the pages in the buffer cache, with mmap the copy occurs in userspace with data being copied into the address space. Swapping can occur in the buffer cache or mmap, this is why so many databases implement their own buffer cache to ensure specific data isn't flushed, leaving them in an inconsistent state.

An advantage of copying in userspace is the ability to use more performant instructions to perform the memcopy, which the kernel does not typically have access to (https://www.mongodb.com/blog/post/getting-storage-engines-re...)

Re:Using mmap to make LLaMA load faster by astrange on Wednesday April 05, 2023 @01:32PM (#35457647)
> With files, the copy occurs within kernel space with data being copied into the pages in the buffer cache, with mmap the copy occurs in userspace with data being copied into the address space.

There is no copy with mmap, the page is either unwritable or CoW. There's always a copy with read(). (But read() can still be faster and more memory efficient nevertheless.)

> An advantage of copying in userspace is the ability to use more performant instructions to perform the memcopy, which the kernel does not typically have access to (https://www.mongodb.com/blog/post/getting-storage-engines-re...)

Darwin kernel does though.

I believe Linux uses the builtin old memcpy instructions on Intel, just to force CPU vendors to keep them usable.

Re:Using mmap to make LLaMA load faster by bigodanktime on Wednesday April 05, 2023 @01:52PM (#35457959)
> There is no copy with mmap

You are right, if you are directly modifying the mmaped region. I always internally model my data as staging my changes to be synchronized to the mmaped region, so thats my mistake there.

> the page is either unwritable or CoW.

This is not universally true, or maybe I'm confused on this statement. MAP_SHARED exists, but maybe you are referencing a specific kernels' implementation on how they achieve coherence between file backed shared memory regions in two processes? Im not sure.

> Darwin kernel does though.

Sure we can always point to a kernel that has has implemented some feature or another, which is why I said typically you don't see it.

Re:Using mmap to make LLaMA load faster by saagarjha on Wednesday April 05, 2023 @02:31PM (#35458516)
> Darwin kernel does though.

It does not. Compare the implementation of _bcopyout against _platform_memmove, you'll see the difference :)

Re:Using mmap to make LLaMA load faster by astrange on Wednesday April 05, 2023 @03:00PM (#35458905)
Huh, maybe I was thinking of "you can use floating point in the kernel".

That doesn't work in every kernel because they don't want to bother saving/restoring the extra registers.

Re:Using mmap to make LLaMA load faster by saagarjha on Thursday April 06, 2023 @04:29AM (#35465775)
To be entirely honest I'm not sure why the kernel doesn't use better routines here, I think on ARM at least it saves the entire NEON state on context switch…
Re:Using mmap to make LLaMA load faster by dekhn on Wednesday April 05, 2023 @12:55PM (#35457096)
Yes- I have 35 years experience with UNIX and used to use mmapping with BLAST, a sequence search tool, as well as my own codes.

I'll repeat myself: mmap is subtle. If what you mmap is larger than your host RAM, only some of the pages will be loaded at any time, and depending on access patterns, can lead to significant paging.