From d8834f18f7662df2ccf0f86578271eb2baef18db Mon Sep 17 00:00:00 2001 From: Stanislav Minakov Date: Mon, 24 Feb 2025 21:27:40 -0800 Subject: [PATCH] Cleanup --- ttnn/cpp/pybind11/device.cpp | 46 ++++++++++++++++++------------------ 1 file changed, 23 insertions(+), 23 deletions(-) diff --git a/ttnn/cpp/pybind11/device.cpp b/ttnn/cpp/pybind11/device.cpp index 40fd854e6c4..92f93b1570c 100644 --- a/ttnn/cpp/pybind11/device.cpp +++ b/ttnn/cpp/pybind11/device.cpp @@ -260,13 +260,13 @@ void device_module(py::module& m_device) { Args: sub_device_manager_id (SubDeviceManagerId): The ID of the sub-device manager to remove. )doc") - .def( - "set_sub_device_stall_group", - [](IDevice* device, const std::vector& sub_device_ids) { - device->push_work([device, sub_device_ids] { device->set_sub_device_stall_group(sub_device_ids); }); - }, - py::arg("sub_device_ids"), - R"doc( + .def( + "set_sub_device_stall_group", + [](IDevice* device, const std::vector& sub_device_ids) { + device->push_work([device, sub_device_ids] { device->set_sub_device_stall_group(sub_device_ids); }); + }, + py::arg("sub_device_ids"), + R"doc( Set the SubDevice IDs that will be stalled on by default for Fast Dispatch commands such as reading, writing, synchronizing. Stalling here refers to the Fast Dispatch cores waiting for programs to complete execution on the specified SubDevices before proceeding with the specified instruction. The default SubDevice IDs to stall on are set to all SubDevice IDs, and whenever a new SubDevice Manager is loaded. @@ -274,25 +274,25 @@ void device_module(py::module& m_device) { Args: sub_device_ids (List[SubDeviceId]): The IDs of the SubDevices to stall on. )doc") - .def( - "reset_sub_device_stall_group", - [](IDevice* device) { device->push_work([device] { device->reset_sub_device_stall_group(); }); }, - R"doc( + .def( + "reset_sub_device_stall_group", + [](IDevice* device) { device->push_work([device] { device->reset_sub_device_stall_group(); }); }, + R"doc( Resets the sub_device_ids that will be stalled on by default for Fast Dispatch commands such as reading, writing, synchronizing back to all SubDevice IDs. )doc") - .def( - "sfpu_eps", - [](IDevice* device) { return tt::tt_metal::experimental::hal::get_eps(); }, - R"doc(Returns machine epsilon value for current architecture.)doc") - .def( - "sfpu_nan", - [](IDevice* device) { return tt::tt_metal::experimental::hal::get_nan(); }, - R"doc(Returns NaN value for current architecture.)doc") - .def( - "sfpu_inf", - [](IDevice* device) { return tt::tt_metal::experimental::hal::get_inf(); }, - R"doc(Returns Infinity value for current architecture.)doc"); + .def( + "sfpu_eps", + [](IDevice* device) { return tt::tt_metal::experimental::hal::get_eps(); }, + R"doc(Returns machine epsilon value for current architecture.)doc") + .def( + "sfpu_nan", + [](IDevice* device) { return tt::tt_metal::experimental::hal::get_nan(); }, + R"doc(Returns NaN value for current architecture.)doc") + .def( + "sfpu_inf", + [](IDevice* device) { return tt::tt_metal::experimental::hal::get_inf(); }, + R"doc(Returns Infinity value for current architecture.)doc"); auto pyDevice = static_cast>>(m_device.attr("Device")); pyDevice