Jan Wassenberg 8532da47f7 Major refactor of allocator/args: ... use new ThreadingContext2 instead of monostate/init in each frontend Add ThreadingArgs(replaces AppArgs) backprop: use Packed() accessor and MakePacked factory and row-based access to allow for stride compress_weights: remove, moving to py-only exporter instead Move MatPtr to mat.h and revise interface: - Generic MatOwner - rename accessors to Packed* - support stride/row accessors, fix RowPtr stride Add TypeBits(Type) Move GenerateMat to test_util-inl for sharing between matmul test/bench Move internal init to gemma.cc to avoid duplication Rename GemmaEnv model_ to gemma_ for disambiguating vs upcoming ModelStorage Remove --compressed_weights, use --weights instead. tensor_index: add ExtentsFromInfo and TensorIndexLLM/Img Allocator: use normal unique_ptr for AllocBytes so users can call directly threading: use -> because AlignedPtr no longer assumes arrays PiperOrigin-RevId: 745918637		2025-04-10 01:29:54 -07:00
..
pytree	Add Python code for converting Griffin Orbax weights. Refs #301	2024-07-29 12:53:30 -07:00
BUILD.bazel	Major refactor of allocator/args:	2025-04-10 01:29:54 -07:00
compression_clif_aux.cc	Major refactor of allocator/args:	2025-04-10 01:29:54 -07:00
compression_clif_aux.h	Add python wrappers for configs and inference.	2025-01-28 08:22:03 -08:00
compression_extension.cc	Add python wrappers for configs and inference.	2025-01-28 08:22:03 -08:00
compression_test.py	Allow conversion, loading and inference with NUQ.	2025-02-05 07:45:54 -08:00
requirements.txt	Add python wrappers for configs and inference.	2025-01-28 08:22:03 -08:00