fix(gpu): dispatch/gather inputs and outputs to the ks and pbs on all GPUs

This commit is contained in:
Agnes Leroy
2024-06-07 16:04:26 +02:00
committed by Agnès Leroy
parent 95ef13f6ce
commit d9eca01631
34 changed files with 854 additions and 508 deletions

View File

@@ -6,5 +6,6 @@
"p4d.24xlarge": 32.7726,
"p5.48xlarge": 98.32,
"rtx4090": 0.04,
"n3-H100x1": 4.30
"n3-H100x1": 4.30,
"n3-H100x8-NVLink": 22.6
}