initial commit to fix ck gpu targets cmake arg (#42924)

This commit is contained in:
afzpatel 2024-02-29 16:48:07 -05:00 committed by GitHub
parent 5c482d0d7e
commit 2840cb54b5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -75,15 +75,14 @@ def cmake_args(self):
"CMAKE_CXX_COMPILER", "{0}/bin/clang++".format(spec["llvm-amdgpu"].prefix)
),
self.define("CMAKE_C_COMPILER", "{0}/bin/clang".format(spec["llvm-amdgpu"].prefix)),
self.define("HIP_PATH", spec["hip"].prefix),
self.define("HIP_ROOT_DIR", "{0}".format(spec["hip"].prefix)),
self.define("CMAKE_CXX_FLAGS", "-O3"),
self.define("CMAKE_BUILD_TYPE", "Release"),
]
if "auto" not in self.spec.variants["amdgpu_target"]:
args.append(self.define_from_variant("AMDGPU_TARGETS", "amdgpu_target"))
args.append(self.define_from_variant("GPU_TARGETS", "amdgpu_target"))
if self.spec.satisfies("@5.6.0:"):
args.append(self.define("INSTANCES_ONLY", "ON"))
if self.spec.satisfies("@:5.7"):
args.append(self.define("CMAKE_CXX_FLAGS", "-O3"))
return args
def build(self, spec, prefix):