fix: resolve unexpected 'out of memory' issue in multi-GPU setup (#2373)

BienBoy

2025-02-01 c1e365fea09aafda387cac12fdff43d28c598979

1
2
3
4
5
6
7
8
9
10
11
12
1    12
2    1
3    2
4    3
5    4
6    5
7    6
8    7
9    8
10    9
11    10
12    11