Bugfix.

LeelaChessZero · Feb 3, 2025 · 695e55d · 695e55d
1 parent 58c81a0
commit 695e55d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/neural/wrapper.cc b/src/neural/wrapper.cc
@@ -82,7 +82,7 @@ class NetworkAsBackendComputation : public BackendComputation {
         computation_(backend_->network_->NewComputation()),
         entries_(backend_->attrs_.maximum_batch_size) {}
 
-  size_t UsedBatchSize() const override { return computation_->GetBatchSize(); }
+  size_t UsedBatchSize() const override { return entries_.size(); }
 
   AddInputResult AddInput(const EvalPosition& pos,
                           EvalResultPtr result) override {