constructor fix and test

ITensor · May 20, 2024 · f20a818 · f20a818
1 parent dc4d3db
commit f20a818
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/NDTensors/src/lib/UnallocatedArrays/src/unallocatedfill.jl b/NDTensors/src/lib/UnallocatedArrays/src/unallocatedfill.jl
@@ -7,7 +7,7 @@ struct UnallocatedFill{ElT,N,Axes,Alloc} <: AbstractFill{ElT,N,Axes}
 end
 
 function UnallocatedFill{ElT,N,Axes}(f::Fill, alloc::Type) where {ElT,N,Axes}
-  return new{ElT,N,Axes,Type{alloc}}(f, alloc)
+  return UnallocatedFill{ElT,N,Axes,Type{alloc}}(f, alloc)
 end
 
 function UnallocatedFill{ElT,N}(f::Fill, alloc) where {ElT,N}

diff --git a/NDTensors/src/lib/UnallocatedArrays/src/unallocatedzeros.jl b/NDTensors/src/lib/UnallocatedArrays/src/unallocatedzeros.jl
@@ -17,7 +17,7 @@ struct UnallocatedZeros{ElT,N,Axes,Alloc} <: AbstractZeros{ElT,N,Axes}
 end
 
 function UnallocatedZeros{ElT,N,Axes}(z::Zeros, alloc::Type) where {ElT,N,Axes}
-  return new{ElT,N,Axes,Type{alloc}}(z, alloc)
+  return UnallocatedZeros{ElT,N,Axes,Type{alloc}}(z, alloc)
 end
 
 function UnallocatedZeros{ElT,N}(z::Zeros, alloc) where {ElT,N}

diff --git a/NDTensors/src/lib/UnallocatedArrays/test/runtests.jl b/NDTensors/src/lib/UnallocatedArrays/test/runtests.jl
@@ -14,6 +14,7 @@ using .NDTensorsTestUtils: devices_list
   @testset "Basic funcitonality" begin
     z = Zeros{elt}((2, 3))
     Z = UnallocatedZeros(z, dev(Matrix{elt}))
+    Z = UnallocatedZeros{elt}(z, dev(Matrix{elt}))
 
     @test Z isa AbstractFill
     @test size(Z) == (2, 3)
@@ -44,6 +45,8 @@ using .NDTensorsTestUtils: devices_list
     # UnallocatedFill
     f = Fill{elt}(3, (2, 3, 4))
     F = UnallocatedFill(f, Array{elt,ndims(f)})
+    F = UnallocatedFill{elt}(f, Array{elt,ndims(f)})
+
     @test F isa AbstractFill
     @test size(F) == (2, 3, 4)
     @test length(F) == 24