Skip to content

Commit f658073

Browse files
feat: implement size and IndexStyle for MTKParameters
1 parent e563aaa commit f658073

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/systems/parameter_buffer.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -678,6 +678,10 @@ end
678678
return len
679679
end
680680

681+
Base.size(ps::MTKParameters) = (length(ps),)
682+
683+
Base.IndexStyle(::Type{T}) where {T <: MTKParameters} = IndexLinear()
684+
681685
Base.getindex(p::MTKParameters, pind::ParameterIndex) = parameter_values(p, pind)
682686

683687
Base.setindex!(p::MTKParameters, val, pind::ParameterIndex) = set_parameter!(p, val, pind)

0 commit comments

Comments
 (0)