From 2c37e7dcde070ef795926ea8b5af53ae4503e6f4 Mon Sep 17 00:00:00 2001 From: xgqdut2016 Date: Thu, 15 May 2025 10:25:20 +0800 Subject: [PATCH] debug teco rope.py --- operatorspy/tests/rotary_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/operatorspy/tests/rotary_embedding.py b/operatorspy/tests/rotary_embedding.py index 83d8b574..1216634b 100644 --- a/operatorspy/tests/rotary_embedding.py +++ b/operatorspy/tests/rotary_embedding.py @@ -225,5 +225,5 @@ def test_teco(lib, test_cases): test_ascend(lib, test_cases) if args.teco: test_teco(lib, test_cases) - if not (args.cpu or args.cuda or args.bang or args.ascend): + if not (args.cpu or args.cuda or args.bang or args.ascend or args.teco): test_cpu(lib, test_cases)