move some target specfific code to the backend

python/src/llvm.cc

@@ -390,12 +390,6 @@ void init_triton_llvm(py::module &&m) {
                 "support on the AMD backend\n";
             throw std::runtime_error(ErrMsg);
           }
-          for (llvm::Function &f : mod->functions()) {
-            if (f.isIntrinsic() || f.isDeclaration())
-              continue;
-            f.addFnAttr("target-features", "+xnack");
-            f.addFnAttr(llvm::Attribute::SanitizeAddress);
-          }
           AddressSanitizerOptions Opts;
           mpm.addPass(AddressSanitizerPass(Opts));
         }

third_party/amd/backend/compiler.py

@@ -335,6 +335,9 @@ def make_llir(src, metadata, options):
         fns[0].add_fn_attr("amdgpu-waves-per-eu", f"{options.waves_per_eu}")
         denormal_mode = "preserve-sign" if options.allow_flush_denorm else "ieee"
         fns[0].add_fn_attr("denormal-fp-math-f32", denormal_mode)
+        if os.environ.get("TRITON_ENABLE_ASAN", "0") == "1":
+             fns[0].add_fn_attr("target-features", target_features)
+             fns[0].add_fn_attr("sanitize_address")
 
         # Hint the compiler that we'd like the firmware to set the kernel arguments
         # to user SGPRs so that the kernel does not need to s_load its arguments