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

[FRONTEND] Refresh cache when the source code of outlined functions are changed (#590)

Browse Source
This commit is contained in:
Keren Zhou
2022-07-20 17:34:07 -07:00
committed by GitHub
parent 9b2bc88d11
commit af85f5fa46
3 changed files with 31 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/src/triton.cc
View File

@@ -236,8 +236,14 @@ void parse_args(py::list& args, py::list do_not_specialize, const std::string& f
continue;
}
// argument is `constexpr`
if(py::hasattr(arg, "value")){
if (py::hasattr(arg, "value")) {
py::object value = arg.attr("value");
// check if value is a callable object using PyCallable_Check
if (PyCallable_Check(value.ptr())) {
throw std::runtime_error(
"constant argument cannot be a callable object: " +
std::string(py::str(arg)));
}
py::object name = arg_names[i];
constants[name] = value;
py::object repr = py::repr(value);