diff --git a/python/tvm/relay/backend/contrib/uma/api/partitioner.py b/python/tvm/relay/backend/contrib/uma/api/partitioner.py index 54dc08cfcf7a..48cac81d13d8 100644 --- a/python/tvm/relay/backend/contrib/uma/api/partitioner.py +++ b/python/tvm/relay/backend/contrib/uma/api/partitioner.py @@ -75,7 +75,8 @@ def register(self) -> None: def partition( self, mod: tvm.IRModule, params: Optional[Dict[str, tvm.runtime.NDArray]] = None ) -> tvm.IRModule: - """Partition the relay graph in parts supported and unsupported by the target hardware accelerator. + """Partition the relay graph in parts supported and unsupported by the + target hardware accelerator. Parameters ----------