From 2f1bab5ce569d6380cbd5896ca385b004c954401 Mon Sep 17 00:00:00 2001 From: "Christopher H. Jordan" Date: Wed, 17 Jan 2024 21:13:01 +0800 Subject: [PATCH] Rename the FEE GPU-calling function. This should've been done when the analytic code was introduced. --- src/fee/gpu/double.rs | 2 +- src/fee/gpu/fee.h | 12 ++++++------ src/fee/gpu/mod.rs | 4 ++-- src/fee/gpu/single.rs | 2 +- src/update_rust_bindings.sh | 2 +- 5 files changed, 11 insertions(+), 11 deletions(-) diff --git a/src/fee/gpu/double.rs b/src/fee/gpu/double.rs index 8dae059..eb867ee 100644 --- a/src/fee/gpu/double.rs +++ b/src/fee/gpu/double.rs @@ -207,7 +207,7 @@ fn bindgen_test_layout_FEECoeffs() { ); } extern "C" { - pub fn gpu_calc_jones( + pub fn gpu_fee_calc_jones( d_azs: *const f64, d_zas: *const f64, num_directions: ::std::os::raw::c_int, diff --git a/src/fee/gpu/fee.h b/src/fee/gpu/fee.h index bf90272..1c20e0d 100644 --- a/src/fee/gpu/fee.h +++ b/src/fee/gpu/fee.h @@ -28,9 +28,9 @@ typedef struct FEECoeffs { const unsigned char n_max; } FEECoeffs; -const char *gpu_calc_jones(const FLOAT *d_azs, const FLOAT *d_zas, int num_directions, const FEECoeffs *d_coeffs, - int num_coeffs, const void *d_norm_jones, const FLOAT *d_latitude_rad, const int iau_reorder, - void *d_results); +const char *gpu_fee_calc_jones(const FLOAT *d_azs, const FLOAT *d_zas, int num_directions, const FEECoeffs *d_coeffs, + int num_coeffs, const void *d_norm_jones, const FLOAT *d_latitude_rad, + const int iau_reorder, void *d_results); #ifdef __cplusplus } // extern "C" @@ -409,9 +409,9 @@ __global__ void fee_kernel(const FEECoeffs coeffs, const FLOAT *azs, const FLOAT } } -extern "C" const char *gpu_calc_jones(const FLOAT *d_azs, const FLOAT *d_zas, int num_directions, - const FEECoeffs *d_coeffs, int num_coeffs, const void *d_norm_jones, - const FLOAT *d_latitude_rad, const int iau_order, void *d_results) { +extern "C" const char *gpu_fee_calc_jones(const FLOAT *d_azs, const FLOAT *d_zas, int num_directions, + const FEECoeffs *d_coeffs, int num_coeffs, const void *d_norm_jones, + const FLOAT *d_latitude_rad, const int iau_order, void *d_results) { dim3 gridDim, blockDim; blockDim.x = warpSize; gridDim.x = (int)ceil((double)num_directions / (double)blockDim.x); diff --git a/src/fee/gpu/mod.rs b/src/fee/gpu/mod.rs index e9e4ba8..e8a6740 100644 --- a/src/fee/gpu/mod.rs +++ b/src/fee/gpu/mod.rs @@ -478,7 +478,7 @@ impl FEEBeamGpu { // The return value is a pointer to a CUDA/HIP error string. If it's null // then everything is fine. - let error_message_ptr = gpu_calc_jones( + let error_message_ptr = gpu_fee_calc_jones( d_az_rad, d_za_rad, num_directions, @@ -498,7 +498,7 @@ impl FEEBeamGpu { let error_message = CStr::from_ptr(error_message_ptr) .to_str() .unwrap_or(""); - let our_error_str = format!("fee.h:gpu_calc_jones failed with: {error_message}"); + let our_error_str = format!("fee.h:gpu_fee_calc_jones failed with: {error_message}"); Err(FEEBeamError::Gpu(GpuError::Kernel { msg: our_error_str.into(), file: file!(), diff --git a/src/fee/gpu/single.rs b/src/fee/gpu/single.rs index 8718c0b..5303be3 100644 --- a/src/fee/gpu/single.rs +++ b/src/fee/gpu/single.rs @@ -207,7 +207,7 @@ fn bindgen_test_layout_FEECoeffs() { ); } extern "C" { - pub fn gpu_calc_jones( + pub fn gpu_fee_calc_jones( d_azs: *const f32, d_zas: *const f32, num_directions: ::std::os::raw::c_int, diff --git a/src/update_rust_bindings.sh b/src/update_rust_bindings.sh index fa8b4cb..9409889 100755 --- a/src/update_rust_bindings.sh +++ b/src/update_rust_bindings.sh @@ -14,7 +14,7 @@ for PRECISION in SINGLE DOUBLE; do echo "Generating bindings for ${LOWER_CASE}-precision GPU code" bindgen "${SCRIPTPATH}"/fee/gpu/fee.h \ - --allowlist-function "gpu_calc_jones.*" \ + --allowlist-function "gpu_fee_calc_jones.*" \ --allowlist-type "FEECoeffs" \ -- -D BINDGEN -D "${PRECISION}" \ -I "${SCRIPTPATH}"/gpu_common/ \