File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -24,7 +24,7 @@ if(INSTALL_GOOGLETEST AND BUILD_TESTS)
24
24
include (${CMAKE_ROOT} /Modules/ExternalProject.cmake )
25
25
ExternalProject_Add (googletest
26
26
GIT_REPOSITORY https://github.com/google/googletest
27
- GIT_TAG master
27
+ GIT_TAG main
28
28
SOURCE_DIR "${CMAKE_CURRENT_BINARY_DIR} /googletest-src"
29
29
BINARY_DIR "${CMAKE_CURRENT_BINARY_DIR} /googletest-build"
30
30
TEST_COMMAND "" )
Original file line number Diff line number Diff line change @@ -98,6 +98,7 @@ namespace bnn
98
98
ndims_cpu (_ndims),
99
99
shape_cpu (_init_shape_cpu(_shape, _ndims))
100
100
{
101
+ BNNMemory->push (this );
101
102
}
102
103
103
104
template <class data_type >
@@ -184,7 +185,6 @@ namespace bnn
184
185
reshape
185
186
(vector<unsigned >& shape)
186
187
{
187
- unsigned size = _calc_size (this ->shape_cpu , this ->ndims_cpu );
188
188
unsigned * new_shape = new unsigned [shape.size ()];
189
189
copy (shape.begin (), shape.end (), new_shape);
190
190
this ->reshape (new_shape, shape.size ());
You can’t perform that action at this time.
0 commit comments