Adding the permute method to remap the GPU.

Author: liuliu

WORKSPACE

@@ -5,9 +5,9 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
 git_repository(
     name = "ccv",
-    commit = "4ecbf13a9551ff50949f91af01dcbb3b451e1082",
+    commit = "9a5680888d4c25d22cfcbe3ce1edf6dccf5c2a0c",
     remote = "https://github.com/liuliu/ccv.git",
-    shallow_since = "1731696551 -0500",
+    shallow_since = "1732044336 -0500",
 )
 
 load("@ccv//config:ccv.bzl", "ccv_deps", "ccv_setting")

deps.bzl

@@ -17,8 +17,8 @@ def s4nnc_deps():
         git_repository,
         name = "ccv",
         remote = "https://github.com/liuliu/ccv.git",
-        commit = "4ecbf13a9551ff50949f91af01dcbb3b451e1082",
-        shallow_since = "1731696551 -0500",
+        commit = "9a5680888d4c25d22cfcbe3ce1edf6dccf5c2a0c",
+        shallow_since = "1732044336 -0500",
     )
 
     _maybe(

nnc/Tensor.swift

@@ -39,6 +39,18 @@ public enum DeviceKind {
     public static var count: Int {
       Int(ccv_nnc_device_count(Int32(CCV_STREAM_CONTEXT_GPU)))
     }
+
+    /**
+     * Remap GPU devices. It supports at most 64 devices remapping. Call this method
+     * again would clear up the old remap.
+     */
+    public static func permute(_ ordinals: Int...) {
+      Array(ordinals.map { Int32($0) }).withUnsafeBytes {
+        ccv_nnc_set_device_permutation(
+          Int32(CCV_STREAM_CONTEXT_GPU), $0.baseAddress?.assumingMemoryBound(to: Int32.self),
+          Int32(ordinals.count))
+      }
+    }
   }
 
   @usableFromInline