Skip to content

Commit

Permalink
wip
Browse files Browse the repository at this point in the history
  • Loading branch information
charleskawczynski committed Apr 15, 2024
1 parent c8c74c4 commit ab16491
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 6 deletions.
2 changes: 0 additions & 2 deletions .buildkite/pipeline.yml
Original file line number Diff line number Diff line change
Expand Up @@ -693,7 +693,6 @@ steps:
- label: "Unit: matrix field broadcasting (CPU)"
key: unit_matrix_field_broadcasting_cpu_scalar_14
command: "julia --color=yes --check-bounds=yes --project=test test/MatrixFields/matrix_fields_broadcasting/test_scalar_14.jl"
soft_fail: true

- label: "Unit: matrix field broadcasting (CPU)"
key: unit_matrix_field_broadcasting_cpu_scalar_15
Expand Down Expand Up @@ -819,7 +818,6 @@ steps:
- label: "Unit: matrix field broadcasting (GPU)"
key: unit_matrix_field_broadcasting_gpu_scalar_14
command: "julia --color=yes --check-bounds=yes --project=test test/MatrixFields/matrix_fields_broadcasting/test_scalar_14.jl"
soft_fail: true
agents:
slurm_gpus: 1
slurm_mem: 10GB
Expand Down
11 changes: 7 additions & 4 deletions src/MatrixFields/matrix_multiplication.jl
Original file line number Diff line number Diff line change
Expand Up @@ -310,8 +310,8 @@ boundary_modified_ud(::BottomRightMatrixCorner, ud, column_space, i) =
_matrix_rows(ntuple(i -> i + ld1 - 1, ud1 - ld1 + 1), args)
Base.@propagate_inbounds function _matrix_rows(
d::Union{Int, PlusHalf{Int}},
args,
)
args::T,
) where {T}
(
bc,
boundary_modified_ld1,
Expand All @@ -328,9 +328,9 @@ Base.@propagate_inbounds function _matrix_rows(
zero(eltype(matrix2)) # This row is outside the matrix.
end
end
Base.@propagate_inbounds _matrix_rows(tup::Tuple, args) =
Base.@propagate_inbounds _matrix_rows(tup::Tuple, args::T) where {T} =
(_matrix_rows(first(tup), args), _matrix_rows(Base.tail(tup), args)...)
Base.@propagate_inbounds _matrix_rows(tup::Tuple{<:Any}, args) =
Base.@propagate_inbounds _matrix_rows(tup::Tuple{<:Any}, args::T) where {T} =
(_matrix_rows(first(tup), args),)
@inline _matrix_rows(tup::Tuple{}, args) = ()

Expand Down Expand Up @@ -475,4 +475,7 @@ if hasfield(Method, :recursion_relation)
for m in methods(multiply_matrix_at_index)
m.recursion_relation = dont_limit
end
for m in methods(matrix_rows)
m.recursion_relation = dont_limit
end
end

0 comments on commit ab16491

Please sign in to comment.