Nodes/Parametric Studio/PSN Split Batch For GPU
ComfyUI Node

PSN Split Batch For GPU

Split an IMAGE batch into num_gpus contiguous sub-batches (one per GPU).

By parametricparadigm·Created 4 days ago·Updated about 3 hours ago· 1
PSN Split Batch For GPU
  • image_batch
  • gpu_0
  • gpu_1
  • gpu_2
  • gpu_3
  • gpu_4
  • gpu_5
  • gpu_6
  • gpu_7
  • gpu_8
  • gpu_9
  • gpu_10
  • gpu_11
  • gpu_12
  • gpu_13
  • gpu_14
  • gpu_15
  • gpu_16
  • gpu_17
  • gpu_18
  • gpu_19
  • gpu_20
  • gpu_21
  • gpu_22
  • gpu_23
  • gpu_24
  • gpu_25
  • gpu_26
  • gpu_27
  • gpu_28
  • gpu_29
  • gpu_30
  • gpu_31
  • gpu_32
  • gpu_33
  • gpu_34
  • gpu_35
  • gpu_36
  • gpu_37
  • gpu_38
  • gpu_39
  • gpu_40
  • gpu_41
  • gpu_42
  • gpu_43
  • gpu_44
  • gpu_45
  • gpu_46
  • gpu_47
  • gpu_48
  • gpu_49
  • gpu_50
  • gpu_51
  • gpu_52
  • gpu_53
  • gpu_54
  • gpu_55
  • gpu_56
  • gpu_57
  • gpu_58
  • gpu_59
  • gpu_60
  • gpu_61
  • gpu_62
  • gpu_63
◄num_gpus2►
CategoryParametric/Video

Inputs (2)

NameTypeDefaultDescription
image_batchIMAGEImage batch.
num_gpusINT21–64Num gpus. From 1 to 64. Default 2.

Outputs (64)

NameTypeDescription
gpu_0IMAGE—
gpu_1IMAGE—
gpu_2IMAGE—
gpu_3IMAGE—
gpu_4IMAGE—
gpu_5IMAGE—
gpu_6IMAGE—
gpu_7IMAGE—
gpu_8IMAGE—
gpu_9IMAGE—
gpu_10IMAGE—
gpu_11IMAGE—
gpu_12IMAGE—
gpu_13IMAGE—
gpu_14IMAGE—
gpu_15IMAGE—
gpu_16IMAGE—
gpu_17IMAGE—
gpu_18IMAGE—
gpu_19IMAGE—
gpu_20IMAGE—
gpu_21IMAGE—
gpu_22IMAGE—
gpu_23IMAGE—
gpu_24IMAGE—
gpu_25IMAGE—
gpu_26IMAGE—
gpu_27IMAGE—
gpu_28IMAGE—
gpu_29IMAGE—
gpu_30IMAGE—
gpu_31IMAGE—
gpu_32IMAGE—
gpu_33IMAGE—
gpu_34IMAGE—
gpu_35IMAGE—
gpu_36IMAGE—
gpu_37IMAGE—
gpu_38IMAGE—
gpu_39IMAGE—
gpu_40IMAGE—
gpu_41IMAGE—
gpu_42IMAGE—
gpu_43IMAGE—
gpu_44IMAGE—
gpu_45IMAGE—
gpu_46IMAGE—
gpu_47IMAGE—
gpu_48IMAGE—
gpu_49IMAGE—
gpu_50IMAGE—
gpu_51IMAGE—
gpu_52IMAGE—
gpu_53IMAGE—
gpu_54IMAGE—
gpu_55IMAGE—
gpu_56IMAGE—
gpu_57IMAGE—
gpu_58IMAGE—
gpu_59IMAGE—
gpu_60IMAGE—
gpu_61IMAGE—
gpu_62IMAGE—
gpu_63IMAGE—