Closed akihironitta closed 1 month ago
start_idx and end_idx used at feat.values[:, start_idx:end_idx] are on device, which leads to cuda synchronizations.
start_idx
end_idx
feat.values[:, start_idx:end_idx]
start_idx
andend_idx
used atfeat.values[:, start_idx:end_idx]
are on device, which leads to cuda synchronizations.