Supporting flags on model that turns off optimizations.

Author: liuliu

WORKSPACE

@@ -5,9 +5,9 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
 git_repository(
     name = "ccv",
-    commit = "b2bcf7f26cb8967ffcad71218e41431bf40b78fe",
+    commit = "84769a56bacd851ad26d1d2843fa5075aff661e5",
     remote = "https://github.com/liuliu/ccv.git",
-    shallow_since = "1740446209 -0500",
+    shallow_since = "1741587468 -0400",
 )
 
 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 = "b2bcf7f26cb8967ffcad71218e41431bf40b78fe",
-        shallow_since = "1740446209 -0500",
+        commit = "84769a56bacd851ad26d1d2843fa5075aff661e5",
+        shallow_since = "1741587468 -0400",
     )
 
     _maybe(

nnc/Model.swift

@@ -244,6 +244,31 @@ public class Model: AnyModel {
     }
   }
 
+  public struct EnableBits: OptionSet, CaseIterable {
+    public let rawValue: Int32
+    public init(rawValue: Int32) {
+      self.rawValue = rawValue
+    }
+    /**
+     * Disable Optimizations
+     */
+    public static let disableOpt = EnableBits(
+      rawValue: Int32(CCV_NNC_GRAPH_EXEC_DISABLE_OPT))
+    public static let allCases: [EnableBits] = [.disableOpt]
+  }
+  /**
+   * The flags for the underlying execution node.
+   */
+  public var flags: EnableBits {
+    get {
+      let value = ccv_cnnp_model_flags(cModel)
+      return EnableBits(rawValue: value)
+    }
+    set {
+      let _ = ccv_cnnp_model_set_flags(cModel, newValue.rawValue)
+    }
+  }
+
   public enum ParametersType {
     case weight
     case bias

nnc/ModelAddons.swift

@@ -1148,14 +1148,20 @@ public final class Concat: Model {
 }
 
 extension Functional {
-  public static func concat(axis: Int, _ inputs: ModelIOConvertible...) -> Model.IO {
-    return Concat(axis: axis).apply(inputs)
+  public static func concat(
+    axis: Int, _ inputs: ModelIOConvertible..., flags: Model.EnableBits = []
+  ) -> Model.IO {
+    let concat = Concat(axis: axis)
+    concat.flags = flags
+    return concat.apply(inputs)
   }
 
   public static func concat<T: DynamicGraph.TensorGroup>(
-    axis: Int, _ inputs: T..., streamContext: StreamContext? = nil
+    axis: Int, _ inputs: T..., flags: Model.EnableBits = [], streamContext: StreamContext? = nil
   ) -> T {
-    let outputs = Concat(axis: axis)(
+    let concat = Concat(axis: axis)
+    concat.flags = flags
+    let outputs = concat(
       inputs: inputs[0], Array(inputs.suffix(from: 1)), streamContext: streamContext)
     return T(outputs[0])
   }
@@ -1535,7 +1541,8 @@ public final class Debug: Model {
 extension ModelIOConvertible {
   /// Move the value to another Model.IO. This is a special operation that can perform optimizations
   /// violates SSA. Use it with extreme care.
-  public func debug(name: String = "", _ callback: @escaping ([AnyTensor?], StreamContext?) -> Void) -> Model.IO
+  public func debug(name: String = "", _ callback: @escaping ([AnyTensor?], StreamContext?) -> Void)
+    -> Model.IO
   {
     return Debug(name: name, callback)(self)
   }