From 1c1e9b002d75415b54a884ddcc22ed4411945134 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Sun, 26 Sep 2021 19:40:37 +0800
Subject: [PATCH] fix(rng): init layout strides

GitOrigin-RevId: 9833d866dac7d0ed440ffebe55e19dedfb827db9
---
 imperative/src/impl/ops/rng.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/imperative/src/impl/ops/rng.cpp b/imperative/src/impl/ops/rng.cpp
index cef314e38..70afeb068 100644
--- a/imperative/src/impl/ops/rng.cpp
+++ b/imperative/src/impl/ops/rng.cpp
@@ -334,6 +334,7 @@ struct _InferLayout<true> {
         for (size_t i = 0; i < target_ndim; ++i) {
             out_layout.shape[i] = ptr[i];
         }
+        out_layout.init_contiguous_stride();
         return out_layout;
     }
 };
-- 
GitLab