Add large dim test of log_softmax (#53957)

* Add large dim test of log_softmax * fix

Add large dim test of log_softmax (#53957)
* Add large dim test of log_softmax * fix
dd410963 · Zhang Zheng · GitHub · 9f54be49 · dd410963
隐藏空白更改
内联并排

Showing with 6 addition and 0 deletion

python/paddle/fluid/tests/unittests/test_log_softmax.py python/paddle/fluid/tests/unittests/test_log_softmax.py +6 -0

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_log_softmax.py
+++ b/python/paddle/fluid/tests/unittests/test_log_softmax.py
@@ -158,6 +158,12 @@ class TestLogSoftmaxBF16Op(OpTest):
        )


+class TestLogSoftmaxLargeDimFP16OP(TestLogSoftmaxOp):
+    def set_attrs(self):
+        self.dtype = np.float16
+        self.shape = [16, 100000]
+
+
 class TestNNLogSoftmaxAPI(unittest.TestCase):
    def setUp(self):
        self.x_shape = [2, 3, 4, 5]