Skip to content

Commit d75a28d

Browse files
committed
fix batch 32
1 parent 040794f commit d75a28d

3 files changed

Lines changed: 3 additions & 16 deletions

File tree

app/Graph/acc_check.cpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
1-
2-
#ifndef WIN32_LEAN_AND_MEAN
1+
#ifndef WIN32_LEAN_AND_MEAN
32
#define WIN32_LEAN_AND_MEAN
43
#endif
5-
#include <windows.h>
64
#include <psapi.h>
5+
#include <windows.h>
76
#pragma comment(lib, "psapi.lib")
87
#include <crtdbg.h>
98
#include <algorithm>

app/Graph/build.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1142,5 +1142,6 @@ int print_time_stats(Graph& graph) {
11421142
return sum;
11431143
#else
11441144
(void)graph;
1145+
return 0;
11451146
#endif
11461147
}

include/graph/graph.hpp

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -318,11 +318,9 @@ class Graph {
318318

319319
for (size_t i = 0; i < traversal.size(); ++i) {
320320
int current_layer = traversal[i];
321-
322321
#ifdef ENABLE_STATISTIC_TIME
323322
auto start = std::chrono::high_resolution_clock::now();
324323
#endif
325-
326324
if (i != 0) {
327325
inten_.clear();
328326

@@ -345,19 +343,15 @@ class Graph {
345343
}
346344

347345
it->second.count_used_ten--;
348-
349-
// Åñëè òåíçîð áîëüøå íèêîìó íå íóæåí - óäàëÿåì
350346
if (it->second.count_used_ten < 1) {
351347
branch_map_.erase(it);
352348
}
353349
}
354350
}
355351
}
356-
357352
if (outten_.empty()) {
358353
outten_.resize(1);
359354
}
360-
361355
layers_[current_layer]->run(inten_, outten_, options);
362356

363357
#ifdef ENABLE_STATISTIC_TENSORS
@@ -405,9 +399,7 @@ class Graph {
405399
}
406400
new_branch.distribution = dis;
407401
}
408-
409402
branch_map_[current_layer] = std::move(new_branch);
410-
411403
if (outtenres_ && current_layer == end_ &&
412404
!branch_map_[current_layer].give_for_all.empty() &&
413405
countinout[current_layer].second == 0) {
@@ -441,11 +433,6 @@ class Graph {
441433
}
442434
#endif
443435
}
444-
445-
// Ôèíàëüíàÿ î÷èñòêà
446-
branch_map_.clear();
447-
inten_.clear();
448-
outten_.clear();
449436
}
450437

451438
void setOutput(const std::shared_ptr<Layer>& layer, Tensor& vec) {

0 commit comments

Comments
 (0)