From f58f52297cf09bec0312b8e36827263286118c5e Mon Sep 17 00:00:00 2001 From: Sigbjoern Loeland Bore Date: Mon, 23 Jan 2023 13:12:58 +0100 Subject: [PATCH] Added option to read HIPCC_FLAGS environment variable. --- setup.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/setup.py b/setup.py index 76b2cbafd5..19c7ac2e28 100644 --- a/setup.py +++ b/setup.py @@ -27,6 +27,9 @@ rocm_root = os.environ.get("ROCM_ROOT") if rocm_root: cmake_args.append(f"-DCMAKE_HIP_COMPILER_ROCM_ROOT:STRING={rocm_root}") + hipcc_flags = os.environ.get("HIP_HIPCC_FLAGS") + if hipcc_flags: + cmake_args.append(f"-DHIP_HIPCC_FLAGS:STRING={hipcc_flags}") else: raise RuntimeError("Unsupported DP_VARIANT option: %s" % dp_variant)