Skip to content

Commit ac76ca2

Browse files
authored
Change input and output strides to use Int64 (#2915)
`cutensornetCreateNetworkDescriptor` expects Int64 as strides.
1 parent a6a8b7e commit ac76ca2

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

lib/cutensornet/src/types.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,12 +110,12 @@ mutable struct CuTensorNetwork{T}
110110
desc::CuTensorNetworkDescriptor
111111
input_modes::Vector{Vector{Int32}}
112112
input_extents::Vector{Vector{Int32}}
113-
input_strides::Vector{<:Union{Ptr{Nothing}, Vector{Int32}}}
113+
input_strides::Vector{<:Union{Ptr{Nothing}, Vector{Int64}}}
114114
input_qualifiers::Vector{cutensornetTensorQualifiers_t}
115115
input_arrs::Vector{CuArray{T}}
116116
output_modes::Vector{Int32}
117117
output_extents::Vector{Int32}
118-
output_strides::Union{Ptr{Nothing}, Vector{Int32}}
118+
output_strides::Union{Ptr{Nothing}, Vector{Int64}}
119119
output_arr::CuArray{T}
120120
end
121121
function CuTensorNetwork(T::DataType, input_modes, input_extents, input_strides, input_qualifiers, output_modes, output_extents, output_strides)

0 commit comments

Comments
 (0)