We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2db6879 commit 3f95f13Copy full SHA for 3f95f13
1 file changed
tests/test_utils/config/platforms/cuda.yaml
@@ -21,7 +21,7 @@ a100:
21
# Add or remove test cases per model
22
train:
23
aquila: ["tp2_pp2", "tp4_pp2"]
24
- deepseek: ["tp2_pp2_ep2", "tp2_pp2_ep2_engram"]
+ deepseek: ["tp2_pp2_ep2", "tp2_pp2_ep2_engram", "tp1_pp2_ep2_v4"]
25
mixtral: ["tp2_pp1_ep2", "tp4_pp1_ep2"]
26
hetero_train:
27
aquila: ["tp2pp1_tp4pp1_tp2pp1", "tp2dp1pp1_tp2dp2pp1_tp1dp2pp1", "dp2dp4_shared_embedding"]
0 commit comments