Skip to content

Commit 901538e

Browse files
UbuntuUbuntu
authored andcommitted
reduce seq len and remove unused imports in decoder_block
Signed-off-by: Ubuntu <ubuntu@UF-EV-B4-GWH01.maas>
1 parent c5d5a0a commit 901538e

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

.github/workflows/galaxy-deepseek-tests-long-impl.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -65,9 +65,9 @@ jobs:
6565
timeout-minutes: 360
6666
run: |
6767
uv pip install -r models/demos/deepseek_v3/reference/deepseek/requirements.txt
68-
DEEPSEEK_MAX_SEQ_LEN=32768 pytest models/demos/deepseek_v3/tests/test_mla.py --timeout 3600 --durations=0
69-
DEEPSEEK_MAX_SEQ_LEN=32768 pytest models/demos/deepseek_v3/tests/test_decoder_block.py --timeout 7200 --durations=0
70-
DEEPSEEK_MAX_SEQ_LEN=4096 pytest models/demos/deepseek_v3/tests/test_model.py --timeout 10800 --durations=0
68+
DEEPSEEK_MAX_SEQ_LEN=16384 pytest models/demos/deepseek_v3/tests/test_mla.py --timeout 3600 --durations=0
69+
DEEPSEEK_MAX_SEQ_LEN=16384 pytest models/demos/deepseek_v3/tests/test_decoder_block.py --timeout 7200 --durations=0
70+
DEEPSEEK_MAX_SEQ_LEN=2048 pytest models/demos/deepseek_v3/tests/test_model.py --timeout 10800 --durations=0
7171
- uses: tenstorrent/tt-metal/.github/actions/slack-report@main
7272
if: ${{ failure() }}
7373
with:

models/demos/deepseek_v3/tests/test_decoder_block.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,6 @@
1313
from models.demos.deepseek_v3.conftest import PREFILL_SEQ_LENS
1414
from models.demos.deepseek_v3.reference.modeling_deepseek import DeepseekV3DecoderLayer
1515
from models.demos.deepseek_v3.tests.pytest_utils import build_test_cases_and_ids
16-
from models.demos.deepseek_v3.tt.decoder_block.decoder_block_1d import DecoderBlock1D
17-
from models.demos.deepseek_v3.tt.decoder_block.decoder_block_1d_base import DecoderBlock1DBase
1816
from models.demos.deepseek_v3.tt.decoder_block.decoder_block_2d import DecoderBlock2D
1917
from models.demos.deepseek_v3.tt.decoder_block.decoder_block_2d_base import DecoderBlock2DBase
2018
from models.demos.deepseek_v3.tt.decoder_block.moe_decoder_block_2d import MoEDecoderBlock2D

0 commit comments

Comments
 (0)