From 4179b3d8bd656f715bb66b803d84ad312267185c Mon Sep 17 00:00:00 2001 From: shw Date: Sun, 5 May 2024 22:19:15 +0800 Subject: [PATCH] MB(H3K) todo --- python/oneflow/test/modules/test_fused_rotary_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/oneflow/test/modules/test_fused_rotary_embedding.py b/python/oneflow/test/modules/test_fused_rotary_embedding.py index d698623534f..efa9fb90750 100644 --- a/python/oneflow/test/modules/test_fused_rotary_embedding.py +++ b/python/oneflow/test/modules/test_fused_rotary_embedding.py @@ -1387,7 +1387,7 @@ def test_fused_rotary_embedding_op_plane(test_case): args_dict = OrderedDict() args_dict["test_fun"] = [_test_plane] # args_dict["x_layout"] = ["MB(H3K)"] - args_dict["x_layout"] = ["BMHK", "MB(HK)"] # TODO: MB(H3K) paramdims bug; + args_dict["x_layout"] = ["BMHK", "MB(HK)"] # TODO: MB(H3K) bug; args_dict["mode"] = ["plane"] args_dict["base"] = [1e1] args_dict["rotary_size"] = [4, 8]