diff --git a/python/python_direct/multidevice.cpp b/python/python_direct/multidevice.cpp index 0855f4c281f..149f15b71e5 100644 --- a/python/python_direct/multidevice.cpp +++ b/python/python_direct/multidevice.cpp @@ -73,7 +73,7 @@ void bindDeviceMesh(py::module& nvfuser) { py::class_ device_mesh(nvfuser, "DeviceMesh", py::module_local()); device_mesh.def( py::init([](at::Tensor devices) { - return new DeviceMesh(std::move(devices)); + return std::make_unique(std::move(devices)); }), py::arg("devices"), R"( @@ -81,7 +81,7 @@ Create a new DeviceMesh from torch.Tensor. )"); device_mesh.def( py::init([](const std::vector& devices) { - return new DeviceMesh(at::tensor(devices)); + return std::make_unique(at::tensor(devices)); }), py::arg("devices"), R"( diff --git a/python/python_direct/runtime.cpp b/python/python_direct/runtime.cpp index 562049859b1..5b581dc7a3d 100644 --- a/python/python_direct/runtime.cpp +++ b/python/python_direct/runtime.cpp @@ -221,7 +221,7 @@ void bindFusionExecutorCache(py::module& nvfuser) { int64_t fusion_id, bool auto_schedule) { // Make a copy of the fusion for FusionExecutorCache to own. - return new FusionExecutorCache( + return std::make_unique( std::make_unique(*fusion), fusion_id, auto_schedule); }), py::arg("fusion"), diff --git a/python/version.txt b/python/version.txt index 69ca21255df..e7ccda1a357 100644 --- a/python/version.txt +++ b/python/version.txt @@ -1 +1 @@ -0.2.34 +0.2.35