gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update includes

Browse Source
This commit is contained in:
Yan Da
2022-03-24 13:33:54 +08:00
parent 78c3480c85
commit cf7fc8d642
2 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/driver/llvm.cc
View File

@@ -32,6 +32,7 @@
#include "triton/tools/sys/getenv.hpp" #include "triton/tools/sys/getenv.hpp"
#include "triton/tools/sys/mkdir.hpp" #include "triton/tools/sys/mkdir.hpp"
#include "triton/tools/sys/exec.hpp" #include "triton/tools/sys/exec.hpp"
#include "llvm/MC/TargetRegistry.h"
#include "llvm/IR/IRBuilder.h" #include "llvm/IR/IRBuilder.h"
#include "llvm/IR/Verifier.h" #include "llvm/IR/Verifier.h"
#include "llvm/IR/IRPrintingPasses.h" #include "llvm/IR/IRPrintingPasses.h"
@@ -40,7 +41,6 @@
#include "llvm/Support/CommandLine.h" #include "llvm/Support/CommandLine.h"
#include "llvm/Support/SourceMgr.h" #include "llvm/Support/SourceMgr.h"
#include "llvm/Support/raw_ostream.h" #include "llvm/Support/raw_ostream.h"
#include "llvm/Support/TargetRegistry.h"
#include "llvm/Support/TargetSelect.h" #include "llvm/Support/TargetSelect.h"
#include "llvm/Target/TargetMachine.h" #include "llvm/Target/TargetMachine.h"
#include "llvm/Target/TargetOptions.h" #include "llvm/Target/TargetOptions.h"

25
python/src/triton.cc
View File

@@ -653,7 +653,6 @@ void init_triton_ir(py::module &&m) {
py::class_<MlirOperation>(m, "operation") py::class_<MlirOperation>(m, "operation")
.def("add_entry_block", [](MlirOperation &self) -> MlirBlock { .def("add_entry_block", [](MlirOperation &self) -> MlirBlock {
// if (auto FunctionOp = unwrap(self)->dyn_cast<mlir::FuncOp>()) {
if (auto info = unwrap(self)->getRegisteredInfo()) { if (auto info = unwrap(self)->getRegisteredInfo()) {
if (mlir::TypeID::get<mlir::FuncOp>() == info->getTypeID()) { if (mlir::TypeID::get<mlir::FuncOp>() == info->getTypeID()) {
auto FunctionOp = mlir::FuncOp::getFromOpaquePointer(unwrap(self)); auto FunctionOp = mlir::FuncOp::getFromOpaquePointer(unwrap(self));
@@ -1044,9 +1043,27 @@ void init_triton_ir(py::module &&m) {
unwrap(lhs), unwrap(rhs) unwrap(lhs), unwrap(rhs)
))); )));
}, ret::reference) }, ret::reference)
// .def("create_fcmpOGT", &ir::builder::create_fcmpOGT, ret::reference) .def("create_fcmpOGT", [](mlir::OpBuilder &self, MlirValue &lhs, MlirValue &rhs) -> MlirValue {
// .def("create_fcmpOLE", &ir::builder::create_fcmpOLE, ret::reference) auto loc = self.getUnknownLoc();
// .def("create_fcmpOGE", &ir::builder::create_fcmpOGE, ret::reference) return wrap(mlir::Value(self.create<mlir::arith::CmpFOp>(
loc, mlir::arith::CmpFPredicate::OGT,
unwrap(lhs), unwrap(rhs)
)));
}, ret::reference)
.def("create_fcmpOLE", [](mlir::OpBuilder &self, MlirValue &lhs, MlirValue &rhs) -> MlirValue {
auto loc = self.getUnknownLoc();
return wrap(mlir::Value(self.create<mlir::arith::CmpFOp>(
loc, mlir::arith::CmpFPredicate::OLE,
unwrap(lhs), unwrap(rhs)
)));
}, ret::reference)
.def("create_fcmpOGE", [](mlir::OpBuilder &self, MlirValue &lhs, MlirValue &rhs) -> MlirValue {
auto loc = self.getUnknownLoc();
return wrap(mlir::Value(self.create<mlir::arith::CmpFOp>(
loc, mlir::arith::CmpFPredicate::OGE,
unwrap(lhs), unwrap(rhs)
)));
}, ret::reference)
// .def("create_fcmpOEQ", &ir::builder::create_fcmpOEQ, ret::reference) // .def("create_fcmpOEQ", &ir::builder::create_fcmpOEQ, ret::reference)
// .def("create_fcmpONE", &ir::builder::create_fcmpONE, ret::reference) // .def("create_fcmpONE", &ir::builder::create_fcmpONE, ret::reference)
// .def("create_fcmpULT", &ir::builder::create_fcmpULT, ret::reference) // .def("create_fcmpULT", &ir::builder::create_fcmpULT, ret::reference)