[RUNTIME] Re-vamped cache so users can manually patch IR / ptx / cubin files (#845)

Also deprecates a couple of tests
This commit is contained in:
Philippe Tillet
2022-11-04 10:57:29 -07:00
committed by GitHub
parent 4218e68d74
commit b6dbe959f0
5 changed files with 93 additions and 197 deletions

View File

@@ -261,7 +261,7 @@ void init_triton_ir(py::module &&m) {
},
ret::reference)
.def("dump", [](mlir::OpState &self) { self->dump(); })
.def("str",
.def("__str__",
[](mlir::OpState &self) -> std::string {
std::string str;
llvm::raw_string_ostream os(str);
@@ -1280,8 +1280,8 @@ void init_triton_translation(py::module &m) {
using ret = py::return_value_policy;
m.def("get_shared_memory_size", [](mlir::ModuleOp module) {
return module->getAttrOfType<mlir::IntegerAttr>("triton_gpu.shared")
.getInt();
auto shared = module->getAttrOfType<mlir::IntegerAttr>("triton_gpu.shared");
return shared.getInt();
});
m.def(