From 2dc09cce1633cd6e4edff826b881229423a6dc97 Mon Sep 17 00:00:00 2001 From: darkdi Date: Mon, 24 Aug 2026 16:41:57 +0300 Subject: [PATCH] Drop two no-op cpu() calls in the k-means cuda path --- coremltools/optimize/torch/_utils/k_means.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/coremltools/optimize/torch/_utils/k_means.py b/coremltools/optimize/torch/_utils/k_means.py index 2e39810de..96213523a 100644 --- a/coremltools/optimize/torch/_utils/k_means.py +++ b/coremltools/optimize/torch/_utils/k_means.py @@ -675,10 +675,6 @@ def _cluster_weights_2d( max_iter=300, ).fit(weight_2d, sample_weight=importance_2d) - weight_2d.cpu() - if importance_2d is not None: - importance_2d.cpu() - return kmeans_results.cluster_centers_.cpu(), kmeans_results.labels_.cpu() @classmethod