Merge pull request #2315 from MrSidims/add-exact-round-opt

[SYCL] Add -foffload-fp32-prec-[div/sqrt] FE option handling in JIT
oneapi-src · Nov 20, 2024 · 83c02a3 · 83c02a3
2 parents 9937d02 + 4d782f1
commit 83c02a3
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 0 deletions.
diff --git a/source/adapters/cuda/platform.cpp b/source/adapters/cuda/platform.cpp
@@ -168,5 +168,10 @@ UR_APIEXPORT ur_result_t UR_APICALL urPlatformGetBackendOption(
     *ppPlatformOption = "";
     return UR_RESULT_SUCCESS;
   }
+  if (pFrontendOption == "-foffload-fp32-prec-div"sv ||
+      pFrontendOption == "-foffload-fp32-prec-sqrt"sv) {
+    *ppPlatformOption = "";
+    return UR_RESULT_SUCCESS;
+  }
   return UR_RESULT_ERROR_INVALID_VALUE;
 }
diff --git a/source/adapters/hip/platform.cpp b/source/adapters/hip/platform.cpp
@@ -154,5 +154,10 @@ urPlatformGetBackendOption(ur_platform_handle_t, const char *pFrontendOption,
     *ppPlatformOption = "";
     return UR_RESULT_SUCCESS;
   }
+  if (pFrontendOption == "-foffload-fp32-prec-div"sv ||
+      pFrontendOption == "-foffload-fp32-prec-sqrt"sv) {
+    *ppPlatformOption = "";
+    return UR_RESULT_SUCCESS;
+  }
   return UR_RESULT_ERROR_INVALID_VALUE;
 }
diff --git a/source/adapters/level_zero/platform.cpp b/source/adapters/level_zero/platform.cpp
@@ -196,6 +196,11 @@ ur_result_t urPlatformGetBackendOption(
     *PlatformOption = "-igc_opts 'PartitionUnit=1,SubroutineThreshold=50000'";
     return UR_RESULT_SUCCESS;
   }
+  if (FrontendOption == "-foffload-fp32-prec-div"sv ||
+      FrontendOption == "-foffload-fp32-prec-sqrt"sv) {
+    *PlatformOption = "-ze-fp32-correctly-rounded-divide-sqrt";
+    return UR_RESULT_SUCCESS;
+  }
   return UR_RESULT_ERROR_INVALID_VALUE;
 }
 

diff --git a/source/adapters/opencl/platform.cpp b/source/adapters/opencl/platform.cpp
@@ -144,5 +144,10 @@ urPlatformGetBackendOption(ur_platform_handle_t, const char *pFrontendOption,
     *ppPlatformOption = "-igc_opts 'PartitionUnit=1,SubroutineThreshold=50000'";
     return UR_RESULT_SUCCESS;
   }
+  if (pFrontendOption == "-foffload-fp32-prec-div"sv ||
+      pFrontendOption == "-foffload-fp32-prec-sqrt"sv) {
+    *ppPlatformOption = "-cl-fp32-correctly-rounded-divide-sqrt";
+    return UR_RESULT_SUCCESS;
+  }
   return UR_RESULT_ERROR_INVALID_VALUE;
 }