From 166b0fa45a76f8b369b6bafa093cab4303b80cbd Mon Sep 17 00:00:00 2001 From: Aaron Liu Date: Fri, 13 Dec 2024 10:07:58 +0800 Subject: [PATCH] rocr/dtif: add dtif environment variable Using HSA_ENABLE_DTIF to control dtif/native thunk code path Signed-off-by: Aaron Liu Reviewed-by: David Yat Sin --- runtime/hsa-runtime/core/util/flag.h | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/runtime/hsa-runtime/core/util/flag.h b/runtime/hsa-runtime/core/util/flag.h index 3275b677e8..1495352068 100644 --- a/runtime/hsa-runtime/core/util/flag.h +++ b/runtime/hsa-runtime/core/util/flag.h @@ -284,6 +284,12 @@ class Flag { var = os::GetEnvVar("HSA_IMAGE_ENABLE_3D_SWIZZLE_DEBUG"); enable_3d_swizzle_ = (var == "1") ? true : false; + + // This allows convient usage in scripting for enabling dtif. + // IE the user should set HSA_DTIF_ENABLED = 1 to enable DTIF. + // HSA_DTIF_ENABLED = 0 will disable DTIF backend. + var = os::GetEnvVar("HSA_ENABLE_DTIF"); + enable_dtif_ = (var == "1") ? true : false; } void parse_masks(uint32_t maxGpu, uint32_t maxCU) { @@ -404,6 +410,7 @@ class Flag { bool enable_3d_swizzle() const { return enable_3d_swizzle_; } + bool enable_dtif() const { return enable_dtif_; } private: bool check_flat_scratch_; bool enable_vm_fault_message_; @@ -437,6 +444,7 @@ class Flag { uint32_t signal_abort_timeout_; int async_events_thread_priority_; bool enable_3d_swizzle_ = false; + bool enable_dtif_; SDMA_OVERRIDE enable_sdma_; SDMA_OVERRIDE enable_peer_sdma_;