Change the missing files

Signed-off-by: Nghia Truong <nghiat@nvidia.com>

Author: ttnghia

src/main/cpp/src/ArithmeticJni.cpp

@@ -66,12 +66,13 @@ JNIEXPORT jlong JNICALL Java_com_nvidia_spark_rapids_jni_Arithmetic_round(
   JNIEnv* env, jclass, jlong input_ptr, jint decimal_places, jint rounding_method)
 {
   JNI_NULL_CHECK(env, input_ptr, "input is null", 0);
-  try {
+  JNI_TRY
+  {
     cudf::jni::auto_set_device(env);
     cudf::column_view* input     = reinterpret_cast<cudf::column_view*>(input_ptr);
     cudf::rounding_method method = static_cast<cudf::rounding_method>(rounding_method);
     return cudf::jni::release_as_jlong(spark_rapids_jni::round(*input, decimal_places, method));
   }
-  CATCH_STD(env, 0);
+  JNI_CATCH(env, 0);
 }
 }

src/main/cpp/src/DeviceAttrJni.cpp

@@ -20,7 +20,8 @@ extern "C" {
 JNIEXPORT jint JNICALL Java_com_nvidia_spark_rapids_jni_DeviceAttr_isDeviceIntegrated(JNIEnv* env,
                                                                                       jclass)
 {
-  try {
+  JNI_TRY
+  {
     int device{};
     CUDF_CUDA_TRY(cudaGetDevice(&device));
 
@@ -29,7 +30,7 @@ JNIEXPORT jint JNICALL Java_com_nvidia_spark_rapids_jni_DeviceAttr_isDeviceInteg
 
     return integrated;
   }
-  CATCH_STD(env, 0);
+  JNI_CATCH(env, 0);
 }
 
 }  // extern "C"