[0D-Tensor] CINN supports gaussian_random (#55547)

jiahy0825 · web-flow · commit 30f059d6e1d7 · 2023-07-20T14:56:29.000+08:00
diff --git a/paddle/cinn/hlir/op/contrib/gaussian_random.cc b/paddle/cinn/hlir/op/contrib/gaussian_random.cc
@@ -81,7 +81,6 @@ std::vector<framework::shape_t> InferShapeForGaussianRandom(
     const framework::AttrMapType &attrs) {
   CHECK(attrs.count("shape"));
   auto shape = absl::get<std::vector<int>>(attrs.at("shape"));
-  CHECK(!shape.empty()) << "shape attr is empty!";
   return {shape};
 }
 
diff --git a/test/cinn/ops/test_zero_dim_tensor.py b/test/cinn/ops/test_zero_dim_tensor.py
@@ -1299,6 +1299,45 @@ def init_input(self):
         self.target_shape = ()
 
 
+@OpTestTool.skip_if(
+    not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."
+)
+class TestGaussianRandomOp(OpTest):
+    def setUp(self):
+        np.random.seed(2023)
+        self.dtype = "float32"
+        self.target_shape = ()
+
+    def build_paddle_program(self, target):
+        out = paddle.tensor.random.gaussian(
+            shape=[],
+            mean=0.0,
+            std=0.0,
+            dtype=self.dtype,
+        )
+        self.paddle_outputs = [out]
+
+    def build_cinn_program(self, target):
+        builder = NetBuilder("gaussian_random_op")
+
+        out = builder.gaussian_random(
+            [],
+            0.0,
+            0.0,
+            1234,
+            self.dtype,
+        )
+
+        prog = builder.build()
+        res = self.get_cinn_output(prog, target, [], [], [out])
+        self.cinn_outputs = res
+
+        self.assertEqual(res[0].shape, self.target_shape)
+
+    def test_check_results(self):
+        self.check_outputs_and_grads()
+
+
 @OpTestTool.skip_if(
     not is_compiled_with_cuda(), "x86 test will be skipped due to timeout."
 )

Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,6 @@ std::vector<framework::shape_t> InferShapeForGaussianRandom(`
`81`	`81`	`const framework::AttrMapType &attrs) {`
`82`	`82`	`CHECK(attrs.count("shape"));`
`83`	`83`	`auto shape = absl::get<std::vector<int>>(attrs.at("shape"));`
`84`		`- CHECK(!shape.empty()) << "shape attr is empty!";`
`85`	`84`	`return {shape};`
`86`	`85`	`}`
`87`	`86`