Skip to content

Commit 6396c10

Browse files
committed
Fix batchnorm
1 parent c04ed59 commit 6396c10

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sw/kernels/dnn/batchnorm/src/batchnorm.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ static inline void batchnorm_layer(const batchnorm_layer_t *l) {
8585
uint32_t weights_size = l->CI;
8686
uint32_t ofmap_size = 2 * l->IW * l->TILE_CI;
8787

88-
double *ptr = (double *)(snrt_cluster()->tcdm.mem);
88+
double *ptr = (double *)snrt_l1_next();
8989
double *ifmap = ptr;
9090
ptr += ifmap_size;
9191
double *gamma = ptr;

0 commit comments

Comments
 (0)