Skip to content

Commit

Permalink
Update dependencies.
Browse files Browse the repository at this point in the history
  • Loading branch information
liuliu committed Jul 2, 2024
1 parent e86a6e3 commit 705abad
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 8 deletions.
4 changes: 2 additions & 2 deletions WORKSPACE
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,9 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")

git_repository(
name = "ccv",
commit = "bd40f0bf82ad5c1d9d30e2a7a6dcd97baba8698a",
commit = "885188aec75d280972e8fc902be75b13fdd33c3e",
remote = "https://github.com/liuliu/ccv.git",
shallow_since = "1718762824 -0400",
shallow_since = "1719960038 -0400",
)

load("@ccv//config:ccv.bzl", "ccv_deps", "ccv_setting")
Expand Down
4 changes: 2 additions & 2 deletions deps.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,8 @@ def s4nnc_deps():
git_repository,
name = "ccv",
remote = "https://github.com/liuliu/ccv.git",
commit = "bd40f0bf82ad5c1d9d30e2a7a6dcd97baba8698a",
shallow_since = "1718762824 -0400",
commit = "885188aec75d280972e8fc902be75b13fdd33c3e",
shallow_since = "1719960038 -0400",
)

_maybe(
Expand Down
8 changes: 4 additions & 4 deletions nnc/ModelAddons.swift
Original file line number Diff line number Diff line change
Expand Up @@ -124,10 +124,10 @@ public final class Matmul: Model {
super.init(model)
}

public init(transposeA: (Int, Int) = (0, 0), transposeB: (Int, Int) = (0, 0), name: String = "") {
public init(transposeA: (Int, Int) = (0, 0), transposeB: (Int, Int) = (0, 0), flags: DynamicGraph.EnableBits = [], name: String = "") {
let a = [Int32(transposeA.0), Int32(transposeA.1)]
let b = [Int32(transposeB.0), Int32(transposeB.1)]
super.init(ccv_cnnp_matmul(a, b, name))
super.init(ccv_cnnp_matmul(a, b, Int32(flags.rawValue), name))
}

public func callAsFunction<T: DynamicGraph.TensorGroup>(
Expand Down Expand Up @@ -177,10 +177,10 @@ public final class Dense: Model {
super.init(model)
}

public init(count: Int, noBias: Bool = false, trainable: Bool? = nil, name: String = "") {
public init(count: Int, noBias: Bool = false, flags: DynamicGraph.EnableBits = [], trainable: Bool? = nil, name: String = "") {
super.init(
ccv_cnnp_dense(
Int32(count), noBias ? 1 : 0, trainable == true ? 1 : (trainable == false ? 0 : -1), name))
Int32(count), noBias ? 1 : 0, Int32(flags.rawValue), trainable == true ? 1 : (trainable == false ? 0 : -1), name))
}

public func callAsFunction<T: DynamicGraph.TensorGroup>(
Expand Down

0 comments on commit 705abad

Please sign in to comment.