зеркало из https://github.com/microsoft/DeepSpeed.git
20 строки
336 B
C
20 строки
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
|