gemma.cpp/compression
Jan Wassenberg f74d496879 Threading/infra improvements.
* Add Parallelize*Range helpers and partitioning helpers
* Refactor Pinning class, store original affinity (required to construct another NestedPools after pinning happened)

Compress:
* prevent Compress printing stats in tests
* zero-pad tensors

Matmul:
* add matmul_unit_test (TODO) and bench_matmul
* matmul_test: change norm to row vectors (that is what is added) and include bf16 rounding error
* Prepare for L2/L3 retrieval
PiperOrigin-RevId: 700603811
2024-11-27 01:12:00 -08:00
..
python Improved consistency of compressor API, and added a universal method with a target type arg. 2024-11-21 05:27:40 -08:00
BUILD.bazel Added tensor_index as a single source of truth on tensor shapes/sources and transformations 2024-11-19 00:25:39 -08:00
analyze.h Major compression update, arbitrary-len unpack + new Dot 2024-09-10 02:22:19 -07:00
blob_compare.cc Added a blob_compare tool that compares two sbs files that may have the blobs in a different order 2024-11-14 03:26:32 -08:00
blob_store.cc Expose BlobReader::Keys() 2024-11-07 10:28:39 -08:00
blob_store.h Expose BlobReader::Keys() 2024-11-07 10:28:39 -08:00
blob_store_test.cc Expose BlobReader::Keys() 2024-11-07 10:28:39 -08:00
compress-inl.h Threading/infra improvements. 2024-11-27 01:12:00 -08:00
compress.cc Added MatPtr/MatPtrT/MatStorageT/MatStorage as a dynamically-sized replacement for CompressedArray. 2024-10-10 08:22:30 -07:00
compress.h Threading/infra improvements. 2024-11-27 01:12:00 -08:00
compress_test.cc Major compression update, arbitrary-len unpack + new Dot 2024-09-10 02:22:19 -07:00
compress_weights.cc Factor out addition of ViTConfig to a ModelConfig. 2024-10-28 05:29:33 -07:00
convert_weights.py Cleanup: move util/compress and convert_weights to compression/ 2024-07-05 04:16:52 -07:00
distortion.h Refactor/cleanup, remove even_odd 2024-09-04 09:25:13 -07:00
distortion_test.cc Major compression update, arbitrary-len unpack + new Dot 2024-09-10 02:22:19 -07:00
fields.cc Serialization for class members for use with ModelConfig 2024-10-25 03:12:34 -07:00
fields.h Serialization for class members for use with ModelConfig 2024-10-25 03:12:34 -07:00
fields_test.cc Serialization for class members for use with ModelConfig 2024-10-25 03:12:34 -07:00
io.cc Further improve IO, enable multiple backends without -D. 2024-04-19 00:40:29 -07:00
io.h Major duplicated code reduction in test/benchmarks 2024-06-14 00:16:25 -07:00
io_win.cc Further improve IO, enable multiple backends without -D. 2024-04-19 00:40:29 -07:00
nuq-inl.h Minor cleanup, Windows+Bazel build fixes 2024-10-10 09:05:06 -07:00
nuq_test.cc Major compression update, arbitrary-len unpack + new Dot 2024-09-10 02:22:19 -07:00
sfp-inl.h Major compression update, arbitrary-len unpack + new Dot 2024-09-10 02:22:19 -07:00
sfp_test.cc Major compression update, arbitrary-len unpack + new Dot 2024-09-10 02:22:19 -07:00
shared.h Simpler MatMul interface, vocab types, Tristate for use_spinning 2024-11-04 07:48:29 -08:00
test_util-inl.h Add double-precision dot variant 2024-09-26 12:09:10 -07:00