[Diffusion] Flux2 tp support (#16219)

Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Xiaoyu Zhang
2026-01-03 17:02:27 +08:00
committed by GitHub
parent 65b0b5b248
commit d0fb24ee7b
3 changed files with 132 additions and 83 deletions

View File

@@ -1508,56 +1508,56 @@
"DecodingStage": 321.94
},
"denoise_step_ms": {
"0": 66.94,
"1": 511.02,
"2": 537.24,
"3": 516.92,
"4": 537.18,
"5": 520.79,
"6": 531.95,
"7": 522.69,
"8": 532.77,
"9": 522.36,
"10": 530.91,
"11": 526.67,
"12": 530.23,
"13": 526.16,
"14": 528.55,
"15": 524.96,
"16": 526.36,
"17": 527.71,
"18": 528.84,
"19": 528.73,
"20": 527.95,
"21": 525.1,
"22": 528.21,
"23": 528.06,
"24": 527.11,
"25": 530.93,
"26": 526.27,
"27": 528.66,
"28": 526.33,
"29": 530.99,
"30": 528.51,
"31": 528.53,
"32": 528.73,
"33": 527.02,
"34": 527.49,
"35": 528.25,
"36": 529.96,
"37": 528.43,
"38": 528.5,
"39": 525.72,
"40": 529.65,
"41": 525.62,
"42": 526.24,
"43": 528.77,
"44": 526.68,
"45": 528.06,
"46": 524.92,
"47": 529.5,
"48": 524.22,
"49": 528.21
"0": 129.07,
"1": 437.16,
"2": 437.7,
"3": 437.67,
"4": 437.84,
"5": 438.03,
"6": 438.09,
"7": 437.65,
"8": 437.95,
"9": 438.31,
"10": 437.99,
"11": 438.54,
"12": 438.47,
"13": 438.2,
"14": 438.56,
"15": 438.69,
"16": 438.69,
"17": 438.98,
"18": 437.96,
"19": 438.9,
"20": 438.87,
"21": 438.04,
"22": 437.88,
"23": 439.09,
"24": 438.61,
"25": 437.68,
"26": 439.2,
"27": 439.63,
"28": 438.65,
"29": 439.32,
"30": 439.01,
"31": 438.84,
"32": 438.72,
"33": 439.09,
"34": 438.3,
"35": 439.48,
"36": 438.2,
"37": 439.67,
"38": 440.65,
"39": 439.96,
"40": 439.0,
"41": 439.2,
"42": 439.37,
"43": 439.98,
"44": 438.6,
"45": 439.58,
"46": 440.23,
"47": 440.1,
"48": 440.21,
"49": 439.22
},
"expected_e2e_ms": 27624.8,
"expected_avg_denoise_ms": 518.23,