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

[PYTHON] Now triton.code_gen.Binary can print PTX and LLIR (#88)

Browse Source
This commit is contained in:
Philippe Tillet
2021-04-22 21:50:19 -04:00
committed by Philippe Tillet
parent 29e33e50b7
commit d9112144b4
2 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/src/triton.cc
View File

@@ -59,7 +59,10 @@ void init_triton_driver(py::module &&m) {
});
py::class_<drv::module>(m, "module");
//py::class_<drv::cu_module, drv::module>(m, "cu_module");
py::class_<drv::cu_module, drv::module>(m, "cu_module")
.def("ptx", &drv::cu_module::ptx)
.def("llir", &drv::cu_module::llir);
py::class_<drv::kernel>(m, "kernel");
}