12345678910111213141516171819 |
- // Copyright (c) Microsoft Corporation.
- // SPDX-License-Identifier: Apache-2.0
- // DeepSpeed Team
- /*
- Copyright NVIDIA/apex
- This file is adapted from fused adam in NVIDIA/apex, commit a109f85
- */
- #ifndef TORCH_CHECK
- #define TORCH_CHECK AT_CHECK
- #endif
- #ifdef VERSION_GE_1_3
- #define DATA_PTR data_ptr
- #else
- #define DATA_PTR data
- #endif
|