format

Author: ASKabalan

lib/include/s2fft_callbacks.h

@@ -1,14 +1,13 @@
 /**
  * @file s2fft_callbacks.h
  * @brief CUDA CUFFT callbacks for HEALPix spherical harmonic transforms
- * 
+ *
  * @note CUFFT CALLBACKS DEPRECATED: This implementation no longer uses cuFFT callbacks.
  * The previous callback-based approach has been replaced with direct kernel launches
- * for better performance and maintainability. The files s2fft_callbacks.h and 
+ * for better performance and maintainability. The files s2fft_callbacks.h and
  * s2fft_callbacks.cc are no longer used and can be considered orphaned.
  */
 
-
 #ifndef _S2FFT_CALLBACKS_CUH_
 #define _S2FFT_CALLBACKS_CUH_
 

lib/include/s2fft_kernels.h

@@ -12,10 +12,10 @@ typedef long long int int64;
 /**
  * @file s2fft_kernels.h
  * @brief CUDA kernels for HEALPix spherical harmonic transforms
- * 
+ *
  * @note CUFT CALLBACKS DEPRECATED: This implementation no longer uses cuFFT callbacks.
  * The previous callback-based approach has been replaced with direct kernel launches
- * for better performance and maintainability. The files s2fft_callbacks.h and 
+ * for better performance and maintainability. The files s2fft_callbacks.h and
  * s2fft_callbacks.cc are no longer used and can be considered orphaned.
  */
 

lib/src/extensions.cc

@@ -70,8 +70,7 @@ constexpr bool is_double_v = is_double<T>::value;
  */
 template <ffi::DataType T>
 ffi::Error healpix_forward(cudaStream_t stream, ffi::Buffer<T> input, ffi::Result<ffi::Buffer<T>> output,
-                           ffi::Result<ffi::Buffer<T>> workspace,
-                           s2fftDescriptor descriptor) {
+                           ffi::Result<ffi::Buffer<T>> workspace, s2fftDescriptor descriptor) {
     // Step 1: Determine the complex type based on the XLA data type.
     using fft_complex_type = fft_complex_t<T>;
     const auto& dim_in = input.dimensions();
@@ -150,8 +149,7 @@ ffi::Error healpix_forward(cudaStream_t stream, ffi::Buffer<T> input, ffi::Resul
  */
 template <ffi::DataType T>
 ffi::Error healpix_backward(cudaStream_t stream, ffi::Buffer<T> input, ffi::Result<ffi::Buffer<T>> output,
-                            ffi::Result<ffi::Buffer<T>> workspace,
-                            s2fftDescriptor descriptor) {
+                            ffi::Result<ffi::Buffer<T>> workspace, s2fftDescriptor descriptor) {
     // Step 1: Determine the complex type based on the XLA data type.
     using fft_complex_type = fft_complex_t<T>;
     const auto& dim_in = input.dimensions();

lib/src/s2fft_kernels.cu

@@ -292,7 +292,6 @@ __global__ void spectral_extension(complex* data, complex* output, int nside, in
     }
 }
 
-
 /**
  * @brief CUDA kernel for FFT shifting and normalization of HEALPix data.
  *
@@ -341,15 +340,14 @@ __global__ void shift_normalize_kernel(complex* data, int nside, bool apply_shif
         long long int shifted_o = (o + nphi / 2) % nphi;
         shifted_o = shifted_o < 0 ? nphi + shifted_o : shifted_o;
         long long int dest_p = r_start + shifted_o;
-        //printf(" -> CUDA: Applying shift: p=%lld, dest_p=%lld, shifted_o=%lld\n", p, dest_p, shifted_o);
+        // printf(" -> CUDA: Applying shift: p=%lld, dest_p=%lld, shifted_o=%lld\n", p, dest_p, shifted_o);
         data[dest_p] = element;
     } else {
         // Step 4b: Write back to original position
         data[p] = element;
     }
 }
 
-
 // ============================================================================
 // C++ LAUNCH FUNCTIONS
 // ============================================================================