format

ITensor · Sep 20, 2023 · 5597fe7 · 5597fe7
1 parent ca75862
commit 5597fe7
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 10 deletions.
diff --git a/NDTensors/ext/NDTensorsCUDAExt/adapt.jl b/NDTensors/ext/NDTensorsCUDAExt/adapt.jl
@@ -31,5 +31,9 @@ function Adapt.adapt_storage(arraytype::Type{<:CuArray}, xs::NDTensors.Unallocat
   )
   elt = get_parameter(arraytype_specified_3, Position(1))
   N = get_parameter(arraytype_specified_3, Position(2))
-  return NDTensors.UnallocatedZeros{elt, N, CUDA.CuArray{elt, N, default_parameter(CuArray, Position(3))}}(size(xs))
-end
+  return NDTensors.UnallocatedZeros{
+    elt,N,CUDA.CuArray{elt,N,default_parameter(CuArray, Position(3))}
+  }(
+    size(xs)
+  )
+end
diff --git a/NDTensors/ext/NDTensorsMetalExt/adapt.jl b/NDTensors/ext/NDTensorsMetalExt/adapt.jl
@@ -30,6 +30,10 @@ function Adapt.adapt_storage(arraytype::Type{<:MtlArray}, xs::NDTensors.Unalloca
   )
   elt = get_parameter(arraytype_specified_3, Position(1))
   N = get_parameter(arraytype_specified_3, Position(2))
-
-  return NDTensors.UnallocatedZeros{elt, N, Metal.MtlArray{elt, N, default_parameter(MtlArray, Position(3))}}(size(xs))
-end
+
+  return NDTensors.UnallocatedZeros{
+    elt,N,Metal.MtlArray{elt,N,default_parameter(MtlArray, Position(3))}
+  }(
+    size(xs)
+  )
+end
diff --git a/NDTensors/src/zeros/set_types.jl b/NDTensors/src/zeros/set_types.jl
@@ -1,9 +1,21 @@
 import .SetParameters: set_parameter, nparameters, default_parameter
 
 # `SetParameters.jl` overloads.
-NDTensors.SetParameters.get_parameter(::Type{<:UnallocatedZeros{P1}}, ::Position{1}) where {P1} = P1
-NDTensors.SetParameters.get_parameter(::Type{<:UnallocatedZeros{<:Any,P2}}, ::Position{2}) where {P2} = P2
-NDTensors.SetParameters.get_parameter(::Type{<:UnallocatedZeros{<:Any,<:Any,P3}}, ::Position{3}) where {P3} = P3
+function NDTensors.SetParameters.get_parameter(
+  ::Type{<:UnallocatedZeros{P1}}, ::Position{1}
+) where {P1}
+  return P1
+end
+function NDTensors.SetParameters.get_parameter(
+  ::Type{<:UnallocatedZeros{<:Any,P2}}, ::Position{2}
+) where {P2}
+  return P2
+end
+function NDTensors.SetParameters.get_parameter(
+  ::Type{<:UnallocatedZeros{<:Any,<:Any,P3}}, ::Position{3}
+) where {P3}
+  return P3
+end
 function NDTensors.SetParameters.get_parameter(
   ::Type{<:UnallocatedZeros{<:Any,<:Any,<:Any,P4}}, ::Position{4}
 ) where {P4}
@@ -36,13 +48,15 @@ end
 function set_parameter(
   ::Type{<:UnallocatedZeros{P1,P2,P3,<:Any}}, ::Position{4}, P4
 ) where {P1,P2,P3}
-@show P4
+  @show P4
   return UnallocatedZeros{P1,P2,P3,P4}
 end
 
 default_parameter(::Type{<:UnallocatedZeros}, ::Position{1}) = Float64
 default_parameter(::Type{<:UnallocatedZeros}, ::Position{2}) = 1
 default_parameter(::Type{<:UnallocatedZeros}, ::Position{3}) = Tuple{Base.OneTo{Int64}}
-default_parameter(::Type{<:UnallocatedZeros}, ::Position{4}) = Vector{default_parameter(UnallocatedZeros, Position(1))}
+function default_parameter(::Type{<:UnallocatedZeros}, ::Position{4})
+  return Vector{default_parameter(UnallocatedZeros, Position(1))}
+end
 
 nparameters(::Type{<:UnallocatedZeros}) = Val(4)