Jan Wassenberg 45ad847a41 Replace RowVectorBatch with MatStorageT ... KVCache: add ctor required for MatStorageT, remove Create; bf_pre_ffw_rms_out -> pre_ffw_rms_out optimize_test: larger vocab_size requires more steps shared.h: Remove unused u128 type correctly set Activation matrix rows, avoid passing as arg ops: pass Mat instead of pointers/sizes; vectorize LayerNorm; support any weight type mat: add OverrideRows, used by SetBatchSize PiperOrigin-RevId: 757790736		2025-05-12 09:16:12 -07:00
..
benchmark.cc	Replace RowVectorBatch with MatStorageT	2025-05-12 09:16:12 -07:00
benchmark_helper.cc	Replace RowVectorBatch with MatStorageT	2025-05-12 09:16:12 -07:00
benchmark_helper.h	Rename-only: remove Allocator2 etc suffixes now that refactoring is complete	2025-05-06 09:12:43 -07:00
benchmarks.cc	Introduce QueryResult in GemmaEnv and add a shortcut for WrapAndTokenize.	2024-10-14 04:45:21 -07:00
cross_entropy.cc	Huge refactor of weight handling and model loading.	2025-05-06 04:44:21 -07:00
cross_entropy.h	Huge refactor of weight handling and model loading.	2025-05-06 04:44:21 -07:00
debug_prompt.cc	Move fields, io* and blob* from compression/ into io/	2025-05-06 11:17:19 -07:00
gemma_batch_bench.cc	Huge refactor of weight handling and model loading.	2025-05-06 04:44:21 -07:00
gemma_test.cc	Minor: mark command line flags as required	2025-05-12 08:30:44 -07:00
prompts.h	Benchmark gemma.cpp with different length inputs.	2024-10-10 15:59:26 -07:00
run_mmlu.cc	Move fields, io* and blob* from compression/ into io/	2025-05-06 11:17:19 -07:00