wehub-resource-sync
17 lines
383 B
C++
17 lines
383 B
C++
// Copyright (c) Microsoft Corporation.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// DeepSpeed Team
|
|
|
|
/*
|
|
Utilities for type casting torch tensors without data movement.
|
|
*/
|
|
|
|
#include <torch/extension.h>
|
|
#include <vector>
|
|
|
|
using namespace std;
|
|
at::Tensor cast_to_byte_tensor(at::Tensor& src_tensor);
|
|
|
|
std::vector<at::Tensor> cast_to_byte_tensor(std::vector<at::Tensor>& tensor_list);
|