| .. |
|
bindings
|
Improved timing for image tokens
|
2026-03-11 04:47:56 -07:00 |
|
evals
|
Add MMLU eval to github
|
2024-05-20 10:20:53 -07:00 |
|
activations.h
|
Replaced attention in ViT with flash - 8x speedup of image tokenizer on AMD
|
2026-03-09 08:46:04 -07:00 |
|
api_client.cc
|
Abort if args are unrecognized, refactor argument passing
|
2025-12-15 03:18:45 -08:00 |
|
api_server.cc
|
Abort if args are unrecognized, refactor argument passing
|
2025-12-15 03:18:45 -08:00 |
|
attention.cc
|
Fixed msan error by fixing padding of k_cache and v_cache
|
2026-03-06 08:11:17 -08:00 |
|
attention.h
|
Fixed msan error by fixing padding of k_cache and v_cache
|
2026-03-06 08:11:17 -08:00 |
|
attention_test.cc
|
Rollback of erroneous rollback.
|
2026-03-02 06:50:26 -08:00 |
|
configs.cc
|
Use a struct to manage the mapping between `AttentionImpl` enum values and their string names, simplifying `GetAttentionImplName` function. Add a test to ensure all valid `AttentionImpl` enums have a corresponding name and can be looked up.
|
2026-02-27 01:31:11 -08:00 |
|
configs.h
|
Utilities to convert between different encodings of kv cache
|
2026-03-18 06:16:32 -07:00 |
|
configs_test.cc
|
Use a struct to manage the mapping between `AttentionImpl` enum values and their string names, simplifying `GetAttentionImplName` function. Add a test to ensure all valid `AttentionImpl` enums have a corresponding name and can be looked up.
|
2026-02-27 01:31:11 -08:00 |
|
flash_attention.cc
|
Replaced attention in ViT with flash - 8x speedup of image tokenizer on AMD
|
2026-03-09 08:46:04 -07:00 |
|
flash_attention.h
|
Replaced attention in ViT with flash - 8x speedup of image tokenizer on AMD
|
2026-03-09 08:46:04 -07:00 |
|
flash_attention_test.cc
|
Updates to tests to use kv_transcodign library to reduce theris code size
|
2026-03-24 05:06:01 -07:00 |
|
flash_structs.h
|
Fixed msan error by fixing padding of k_cache and v_cache
|
2026-03-06 08:11:17 -08:00 |
|
gemma-inl.h
|
Add tensor stats and output
|
2025-12-11 22:52:46 -08:00 |
|
gemma.cc
|
Add min_verbosity to MaybePrint
|
2026-03-19 04:22:01 -07:00 |
|
gemma.h
|
Improved timing for image tokens
|
2026-03-11 04:47:56 -07:00 |
|
gemma_args.h
|
Int8 + microscaling support for kv cache formats.
|
2026-03-09 02:50:08 -07:00 |
|
gemma_args_test.cc
|
Abort if args are unrecognized, refactor argument passing
|
2025-12-15 03:18:45 -08:00 |
|
kv_cache.cc
|
Int8 + microscaling support for kv cache formats.
|
2026-03-09 02:50:08 -07:00 |
|
kv_cache.h
|
Int8 + microscaling support for kv cache formats.
|
2026-03-09 02:50:08 -07:00 |
|
kv_cache_test.cc
|
Internal changes
|
2026-01-09 06:35:36 -08:00 |
|
kv_transcoding.cc
|
Utilities to convert between different encodings of kv cache
|
2026-03-18 06:16:32 -07:00 |
|
kv_transcoding.h
|
Utilities to convert between different encodings of kv cache
|
2026-03-18 06:16:32 -07:00 |
|
kv_transcoding_test.cc
|
Utilities to convert between different encodings of kv cache
|
2026-03-18 06:16:32 -07:00 |
|
model_store.cc
|
Add ability to load custom models which are fully described by the ModelConfig blob.
|
2026-03-02 01:18:33 -08:00 |
|
model_store.h
|
Allow overriding hardcoded max_seq_len by cmdline argument seq_len.
|
2026-01-08 04:28:59 -08:00 |
|
query.h
|
Warning fixes (sign mismatch), switch default
|
2025-12-15 02:41:19 -08:00 |
|
run.cc
|
Add/use MaybePrint; also ShowConfig in non-interactive builds
|
2026-03-12 11:20:41 -07:00 |
|
tensor_info.cc
|
Add tensor stats and output
|
2025-12-11 22:52:46 -08:00 |
|
tensor_info.h
|
Add tensor stats and output
|
2025-12-11 22:52:46 -08:00 |
|
tensor_info_test.cc
|
Minor: ModelWeightsPtrs -> WeightsPtrs
|
2025-07-11 06:11:51 -07:00 |
|
tensor_stats.cc
|
Add int8 quantization stats
|
2025-12-19 12:43:03 -08:00 |
|
tensor_stats.h
|
Add int8 quantization stats
|
2025-12-19 12:43:03 -08:00 |
|
tiled_attention.cc
|
Warning fix: cast enum for HWY_ABORT %d
|
2026-03-19 10:11:17 -07:00 |
|
tiled_attention.h
|
Implementation of tiled attention with bf16 and circular buffers which reduces memory requirements by 4x on longer context on gemma models.
|
2026-02-24 03:26:49 -08:00 |
|
tiled_attention_test.cc
|
Updates to tests to use kv_transcodign library to reduce theris code size
|
2026-03-24 05:06:01 -07:00 |
|
tokenizer.cc
|
(Resubmit) Prepare profiler annotations for new API
|
2025-08-13 01:38:24 -07:00 |
|
tokenizer.h
|
6x large-batch, short-prompt prefill speedup
|
2025-06-10 09:56:20 -07:00 |
|
vit.cc
|
Replaced attention in ViT with flash - 8x speedup of image tokenizer on AMD
|
2026-03-09 08:46:04 -07:00 |
|
vit.h
|
Minor: ModelWeightsPtrs -> WeightsPtrs
|
2025-07-11 06:11:51 -07:00 |
|
weights.cc
|
Minor: ParallelismStrategy->Parallelism
|
2025-11-06 06:56:10 -08:00 |
|
weights.h
|
Add tensor stats and output
|
2025-12-11 22:52:46 -08:00 |
8a5e37eeb7