Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
python frontend fix
Browse files Browse the repository at this point in the history
naoyam committed Dec 15, 2024
1 parent 96ac0fa commit 7e9413a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion csrc/python_frontend/python_bindings.cpp
Original file line number Diff line number Diff line change
@@ -823,7 +823,8 @@ void initNvFuserPythonBindings(PyObject* module) {
.value("inner_outer_persistent", SchedulerType::InnerOuterPersistent)
.value("outer_persistent", SchedulerType::OuterPersistent)
.value("transpose", SchedulerType::Transpose)
.value("expr_eval", SchedulerType::ExprEval);
.value("expr_eval", SchedulerType::ExprEval)
.value("resize", SchedulerType::Resize);

nvfuser.def("compute_contiguity", computeContiguity);
nvfuser.def("compute_tensor_descriptor", computeTensorDescriptor);

0 comments on commit 7e9413a

Please sign in to comment.