mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
f9ae3fac8c
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/138903 Approved by: https://github.com/ezyang
16 lines
457 B
C++
16 lines
457 B
C++
// Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
//
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree.
|
|
|
|
#pragma once
|
|
|
|
#include <ATen/ATen.h>
|
|
|
|
namespace torch::distributed::c10d::quantization {
|
|
|
|
at::Tensor _float_to_bfloat16_cuda(const at::Tensor& input);
|
|
at::Tensor _bfloat16_to_float_cuda(const at::Tensor& input);
|
|
|
|
} // namespace torch::distributed::c10d::quantization
|