Skip to content

Commit a26e37b

Browse files
author
lexasub
committed
tool: update cache of streaming dataset(fix tests)
1 parent ed53dd5 commit a26e37b

File tree

4 files changed

+280
-294
lines changed

4 files changed

+280
-294
lines changed

tools/dataset-converter/streaming/llama-dataset-streaming.cpp

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -177,12 +177,9 @@ bool llama_dataset_set_adaptive_cache_sizing(struct llama_dataset* dataset, bool
177177
return false;
178178
}
179179

180-
// Enable/disable adaptive sizing
181-
cache->set_adaptive_sizing(enabled);
182-
183180
// Update the optimization manager if it exists
184181
StreamingOptimizationManager* manager = get_optimization_manager(dataset);
185-
if (manager) {
182+
if (manager != nullptr) {
186183
manager->set_adaptive_cache_enabled(enabled);
187184
}
188185

0 commit comments

Comments
 (0)