fix(rng): init layout strides · 1c1e9b002d - MegEngine - OpenI

fix(rng): init layout strides

GitOrigin-RevId: 9833d866da

tags/v1.9.0

Megvii Engine Team 3 years ago

parent

9527859cc8

commit

1c1e9b002d

1 changed files with 1 additions and 0 deletions

Unified View

							
								+ 1
								
								- 0
							
imperative/src/impl/ops/rng.cppView File
								
																@@ -334,6 +334,7 @@ struct _InferLayout<true> {
															
																        for (size_t i = 0; i < target_ndim; ++i) {
																
																        for (size_t i = 0; i < target_ndim; ++i) {
															
																            out_layout.shape[i] = ptr[i];
																
																            out_layout.shape[i] = ptr[i];
															
																        }
																
																        }
															
																        out_layout.init_contiguous_stride();
															
																        return out_layout;
																
																        return out_layout;
															
																    }
																
																    }
															
																};
																
																};