chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,13 @@
|
||||
// Copyright (c) Microsoft Corporation.
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
// DeepSpeed Team
|
||||
|
||||
#include "cpu_lion.h"
|
||||
|
||||
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
|
||||
{
|
||||
m.def("lion_update", &ds_lion_step, "DeepSpeed CPU Lion update (C++)");
|
||||
m.def("create_lion", &create_lion_optimizer, "DeepSpeed CPU Lion (C++)");
|
||||
m.def("destroy_lion", &destroy_lion_optimizer, "DeepSpeed CPU Lion destroy (C++)");
|
||||
}
|
||||
Reference in New Issue
Block a user