Skip to content

Commit bcc45c3

Browse files
committed
Fix format
1 parent f8010e9 commit bcc45c3

4 files changed

Lines changed: 9 additions & 10 deletions

File tree

src/graph_tools.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -261,10 +261,11 @@ function _compute_gradient_sparsity!(indices::Coloring.IndexedSet, f)
261261
push!(indices, node.index)
262262
elseif node.type == NODE_VARIABLE_BLOCK
263263
len = _length(f.sizes, k)
264-
for i in 0:(len - 1)
264+
for i in 0:(len-1)
265265
push!(indices, node.index + i)
266266
end
267-
elseif node.type == NODE_MOI_VARIABLE || node.type == NODE_MOI_VARIABLE_BLOCK
267+
elseif node.type == NODE_MOI_VARIABLE ||
268+
node.type == NODE_MOI_VARIABLE_BLOCK
268269
error(
269270
"Internal error: Invalid to compute sparsity if $(node.type) " *
270271
"nodes are present.",

src/reverse_mode.jl

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ function _forward_eval(
133133
len = length(tape_range)
134134
copyto!(
135135
view(f.forward_storage, tape_range),
136-
view(x, node.index:(node.index + len - 1)),
136+
view(x, node.index:(node.index+len-1)),
137137
)
138138
elseif node.type == NODE_VALUE_BLOCK
139139
# Pre-loaded into `forward_storage` at construction.
@@ -973,9 +973,8 @@ function _extract_reverse_pass_inner(
973973
# accumulate into the matching slice of `output`.
974974
tape_range = _storage_range(f.sizes, k)
975975
len = length(tape_range)
976-
x_range = node.index:(node.index + len - 1)
977-
cpu_buf =
978-
convert(Vector{T}, view(f.reverse_storage, tape_range))
976+
x_range = node.index:(node.index+len-1)
977+
cpu_buf = convert(Vector{T}, view(f.reverse_storage, tape_range))
979978
view(output, x_range) .+= scale .* cpu_buf
980979
elseif node.type == NODE_VARIABLE
981980
# Per-leaf scalar — rare, so the per-leaf `cudaMemcpy` is fine.

src/sizes.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -465,8 +465,8 @@ struct _SubexpressionStorage{S<:AbstractVector{Float64}}
465465
if node.type == NODE_VALUE_BLOCK
466466
j = sizes.storage_offset[k] + 1
467467
len = _length(sizes, k)
468-
cpu_buffer[j:(j + len - 1)] .=
469-
view(const_values, node.index:(node.index + len - 1))
468+
cpu_buffer[j:(j+len-1)] .=
469+
view(const_values, node.index:(node.index+len-1))
470470
end
471471
end
472472
forward_storage = convert(S, cpu_buffer)

src/types.jl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,7 @@ struct Expression{T}
2727
nodes::Vector{Node}
2828
values::Vector{T}
2929
block_shapes::Dict{Int,Vector{Int}}
30-
Expression{T}() where {T} =
31-
new{T}(Node[], T[], Dict{Int,Vector{Int}}())
30+
Expression{T}() where {T} = new{T}(Node[], T[], Dict{Int,Vector{Int}}())
3231
end
3332

3433
function Base.:(==)(x::Expression, y::Expression)

0 commit comments

Comments
 (0)