mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-11-04 19:04:29 +08:00
20 lines
336 B
C
20 lines
336 B
C
// 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
|