We disabled conformer pruning for ETKDG with torch output because their interaction is nontrivial and going from Gpu->cpu->prune->GPU defeats the purpose. Implement device-side pruning. Will probably have negligible impact on CPU return path but could be a tiny speedup.
We disabled conformer pruning for ETKDG with torch output because their interaction is nontrivial and going from Gpu->cpu->prune->GPU defeats the purpose. Implement device-side pruning. Will probably have negligible impact on CPU return path but could be a tiny speedup.