diff --git a/csrc/serde/fusion_record.cpp b/csrc/serde/fusion_record.cpp index 5de2cda9873..7e23adf2b69 100644 --- a/csrc/serde/fusion_record.cpp +++ b/csrc/serde/fusion_record.cpp @@ -840,6 +840,7 @@ void RecordFunctorFactory::setupFunctionMaps() { NVFUSER_BINARY_TV_OP("bitwise_right_shift", bitwise_right_shift) NVFUSER_BINARY_TV_OP("logical_right_shift", logical_right_shift) NVFUSER_BINARY_TV_OP("gcd", gcd) + NVFUSER_BINARY_TV_OP("ceilDiv", ceilDiv) NVFUSER_BINARY_TV_ALPHA_OP("add_alpha", add_alpha) NVFUSER_BINARY_TV_ALPHA_OP("sub_alpha", sub_alpha) diff --git a/tests/python/test_python_frontend.py b/tests/python/test_python_frontend.py index 874223471eb..085180ada20 100644 --- a/tests/python/test_python_frontend.py +++ b/tests/python/test_python_frontend.py @@ -4672,5 +4672,4 @@ def fusion_func(fd: FusionDefinition) -> None: T223 = fd.ops.cat([T169, T222], dim=-1, manual_padding=0) fd.add_output(T223) - # is_clonable=False is because translation fails with missing ceilDiv - nvf_out, _ = self.exec_nvfuser(fusion_func, inputs, is_clonable=False) + nvf_out, _ = self.exec_nvfuser(fusion_func, inputs)