Apply formatter

CliMA · Apr 22, 2024 · 495e8de · 495e8de
1 parent c428f93
commit 495e8de
Show file tree

Hide file tree

Showing 9 changed files with 21 additions and 18 deletions.
diff --git a/ext/cuda/data_layouts.jl b/ext/cuda/data_layouts.jl
@@ -1,6 +1,6 @@
 
-import ClimaCore.DataLayouts: IJKFVH,IJFH,VIJFH,VIFH,IFH,IJF,IF,VF,DataF
-import ClimaCore.DataLayouts: IJFHStyle,VIJFHStyle,VFStyle,DataFStyle
+import ClimaCore.DataLayouts: IJKFVH, IJFH, VIJFH, VIFH, IFH, IJF, IF, VF, DataF
+import ClimaCore.DataLayouts: IJFHStyle, VIJFHStyle, VFStyle, DataFStyle
 import ClimaCore.DataLayouts: promote_parent_array_type
 import ClimaCore.DataLayouts: parent_array_type
 import Adapt

diff --git a/ext/cuda/fields.jl b/ext/cuda/fields.jl
@@ -310,4 +310,5 @@ function Adapt.adapt_structure(
     )
 end
 
-cuda_synchronize(device::ClimaComms.CUDADevice; kwargs...) = CUDA.synchronize(; kwargs...)
+cuda_synchronize(device::ClimaComms.CUDADevice; kwargs...) =
+    CUDA.synchronize(; kwargs...)
diff --git a/ext/cuda/matrix_fields_multiple_field_solve.jl b/ext/cuda/matrix_fields_multiple_field_solve.jl
@@ -5,7 +5,8 @@ import ClimaCore.MatrixFields: multiple_field_solve!
 import ClimaCore.MatrixFields: is_CuArray_type
 import ClimaCore.MatrixFields: allow_scalar_func
 
-allow_scalar_func(::ClimaComms.CUDADevice, f, args) = CUDA.@allowscalar f(args...)
+allow_scalar_func(::ClimaComms.CUDADevice, f, args) =
+    CUDA.@allowscalar f(args...)
 
 is_CuArray_type(::Type{T}) where {T <: CUDA.CuArray} = true
 

diff --git a/ext/cuda/operators_sem_shmem.jl b/ext/cuda/operators_sem_shmem.jl
@@ -257,4 +257,4 @@ Base.@propagate_inbounds function operator_fill_shmem!(
         Wv₂[i, j, vt] = W ⊠ Geometry.covariant2(arg, local_geometry)
         Wv₃[i, j, vt] = W ⊠ Geometry.covariant3(arg, local_geometry)
     end
-end
+end
diff --git a/ext/cuda/operators_spectral_element.jl b/ext/cuda/operators_spectral_element.jl
@@ -23,7 +23,8 @@ Base.@propagate_inbounds function get_node(
     Operators.operator_evaluate(sbc.op, sbc.work, sbc.axes, ij, slabidx)
 end
 
-cuda_sync_threads(::SpectralBroadcasted{CUDASpectralStyle}) = CUDA.sync_threads()
+cuda_sync_threads(::SpectralBroadcasted{CUDASpectralStyle}) =
+    CUDA.sync_threads()
 
 function Base.copyto!(
     out::Field,
@@ -192,4 +193,3 @@ end
 Base.@propagate_inbounds function resolve_shmem!(obj, ij, slabidx)
     nothing
 end
-
diff --git a/ext/cuda/remapping_distributed.jl b/ext/cuda/remapping_distributed.jl
@@ -2,8 +2,8 @@ import ClimaCore: Topologies, Spaces, Fields
 import ClimaComms
 import CUDA
 using CUDA: @cuda
-import ClimaCore.Remapping: _set_interpolated_values_device!,
-    _set_interpolated_values!
+import ClimaCore.Remapping:
+    _set_interpolated_values_device!, _set_interpolated_values!
 function _set_interpolated_values_device!(
     out::AbstractArray,
     field::Fields.Field,
@@ -36,7 +36,7 @@ function _set_interpolated_values!(
     vert_interpolation_weights::AbstractArray,
     vert_bounding_indices::AbstractArray,
     field_values,
-    ::ClimaComms.CUDADevice
+    ::ClimaComms.CUDADevice,
 )
 
     nblocks, _ = size(interpolation_matrix[1])

diff --git a/src/MatrixFields/MatrixFields.jl b/src/MatrixFields/MatrixFields.jl
@@ -116,10 +116,10 @@ function Base.show(io::IO, field::ColumnwiseBandMatrixField)
         end
         column_field = Fields.column(field, 1, 1, 1)
         io = IOContext(io, :compact => true, :limit => true)
-        allow_scalar_func(Base.print_array, (
-            io,
-            column_field2array_view(column_field),
-        ))
+        allow_scalar_func(
+            Base.print_array,
+            (io, column_field2array_view(column_field)),
+        )
     else
         # When a BandedMatrix with non-number entries is printed, it currently
         # either prints in an illegible format (e.g., if it has AxisTensor or

diff --git a/src/MatrixFields/field_matrix_iterative_solver.jl b/src/MatrixFields/field_matrix_iterative_solver.jl
@@ -430,7 +430,8 @@ check_field_matrix_solver(alg::StationaryIterativeSolve, cache, A, b) =
 is_CuArray_type(::Type{T}) where {T} = false
 function run_field_matrix_solver!(alg::StationaryIterativeSolve, cache, x, A, b)
     P = lazy_or_concrete_preconditioner(alg.P_alg, cache.P_cache, A)
-    using_cuda = is_CuArray_type(ClimaComms.array_type(concrete_field_vector(b)))
+    using_cuda =
+        is_CuArray_type(ClimaComms.array_type(concrete_field_vector(b)))
     if get_debug(alg) && !using_cuda
         @debug begin
             e₀ = concrete_field_vector(b) # Initialize e to any nonzero vector.

diff --git a/src/Remapping/distributed_remapping.jl b/src/Remapping/distributed_remapping.jl
@@ -389,7 +389,7 @@ function _set_interpolated_values!(remapper::Remapper, fields)
         remapper.vert_interpolation_weights,
         remapper.vert_bounding_indices,
         tuple(map(f -> Fields.field_values(f), fields)...),
-        ClimaComms.device(first(fields))
+        ClimaComms.device(first(fields)),
     )
 end
 
@@ -496,7 +496,7 @@ function _set_interpolated_values!(
     vert_interpolation_weights::AbstractArray,
     vert_bounding_indices::AbstractArray,
     fields_values,
-    ::ClimaComms.AbstractDevice
+    ::ClimaComms.AbstractDevice,
 )
     set_interpolated_values_cpu_kernel!(
         out,
@@ -518,7 +518,7 @@ function _set_interpolated_values!(
     ::Nothing,
     ::Nothing,
     fields_values,
-    ::ClimaComms.AbstractDevice
+    ::ClimaComms.AbstractDevice,
 )
 
     space = axes(first(fields))
-Original file line number
+Diff line change
@@ Expand Up / @@ -257,4 +257,4 @@ Base.@propagate_inbounds function operator_fill_shmem!( @@
             Wv₂[i, j, vt] = W ⊠ Geometry.covariant2(arg, local_geometry)
             Wv₃[i, j, vt] = W ⊠ Geometry.covariant3(arg, local_geometry)
         end
-    end
+    end