wehub-resource-sync
15 lines
302 B
C
15 lines
302 B
C
// Copyright (c) Microsoft Corporation.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// DeepSpeed Team
|
|
|
|
#pragma once
|
|
|
|
#include "ds_kernel_utils.h"
|
|
|
|
/*
|
|
Wrapper around cudaHostAlloc with some specific flags. Returns a pointer to the
|
|
memory region of `size` bytes.
|
|
*/
|
|
void* get_cuda_fast_buffer(int64_t size);
|