forked from triton-lang/triton-cpu
-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathzen5.patch
31 lines (28 loc) · 1.34 KB
/
zen5.patch
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
diff --git a/python/src/llvm.cc b/python/src/llvm.cc
index 70e36449..a786a96a 100644
--- a/python/src/llvm.cc
+++ b/python/src/llvm.cc
@@ -410,7 +410,7 @@ void init_triton_llvm(py::module &&m) {
auto target =
llvm::TargetRegistry::lookupTarget(mod->getTargetTriple(), error);
std::unique_ptr<llvm::TargetMachine> machine{target->createTargetMachine(
- mod->getTargetTriple(), llvm::sys::getHostCPUName(), "", {},
+ mod->getTargetTriple(), "znver4", "", {},
llvm::Reloc::PIC_)};
mod->setDataLayout(machine->createDataLayout());
});
@@ -437,7 +437,7 @@ void init_triton_llvm(py::module &&m) {
}
res =
translateLLVMIRToASM(*module, llvm::sys::getDefaultTargetTriple(),
- llvm::sys::getHostCPUName().str(), "", {},
+ "znver4", "", {},
enable_fp_fusion, false, enable_fast_math);
}
return py::str(res);
@@ -553,7 +553,7 @@ void init_triton_llvm(py::module &&m) {
m.def("get_cpu_tripple", []() { return llvm::sys::getProcessTriple(); });
- m.def("get_cpu_name", []() { return llvm::sys::getHostCPUName().str(); });
+ m.def("get_cpu_name", []() { return "znver4"; });
m.def("get_cpu_features", []() {
auto features = llvm::sys::getHostCPUFeatures();