From 11bbe299127a984f790dd84e4fe9804abd99f3b4 Mon Sep 17 00:00:00 2001
From: skishore <sriramkumar.kishorekumar@amd.com>
Date: Sun, 1 Mar 2026 18:37:49 +0000
Subject: [PATCH] remove hipblas flags

---
 tools/setup_helpers/extension.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tools/setup_helpers/extension.py b/tools/setup_helpers/extension.py
index ea5be54a8a..a8d0689cb5 100644
--- a/tools/setup_helpers/extension.py
+++ b/tools/setup_helpers/extension.py
@@ -73,8 +73,7 @@ def get_ext_modules():
         extra_compile_args["nvcc"] = ["-O2", "-DUSE_CUDA"]
     if _USE_ROCM:
         extension = CUDAExtension
-        extra_compile_args["cxx"].append("-DHIPBLAS_V2")
-        extra_compile_args["nvcc"] = ["-O3", "-DHIPBLAS_V2"]
+        extra_compile_args["nvcc"] = ["-O3"]
         # TORCH_HIP_VERSION is used by hipified C++ (e.g. utils_hip.cpp); PyTorch only defines it when building PyTorch.
         if torch.version.hip:
             parts = torch.version.hip.split(".")