mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
fb18c29486
s/Facebook, Inc./Meta Platforms, Inc/ s/Confidential and proprietary./This source code is licensed under the BSD-style license/ Per https://www.internalfb.com/intern/wiki/Open_Source/Licenses/Straight_BSD/ Also, add linter that prevents adding those in the future Fixes https://github.com/pytorch/pytorch/issues/90187 Pull Request resolved: https://github.com/pytorch/pytorch/pull/90805 Approved by: https://github.com/zpao
35 lines
1.2 KiB
C++
35 lines
1.2 KiB
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>
|
|
|
|
#include <typeinfo>
|
|
|
|
inline std::string torch_tensor_device_name(const at::Tensor& ten) {
|
|
return c10::DeviceTypeName(ten.device().type());
|
|
}
|
|
|
|
#define TENSOR_NDIM_EQUALS(ten, dims) \
|
|
TORCH_CHECK( \
|
|
(ten).ndimension() == (dims), \
|
|
"Tensor '" #ten "' must have " #dims \
|
|
" dimension(s). " \
|
|
"Found ", \
|
|
(ten).ndimension())
|
|
|
|
#define TENSOR_ON_CPU(x) \
|
|
TORCH_CHECK( \
|
|
!x.is_cuda(), \
|
|
#x " must be a CPU tensor; it is currently on device ", \
|
|
torch_tensor_device_name(x))
|
|
|
|
#define TENSOR_ON_CUDA_GPU(x) \
|
|
TORCH_CHECK( \
|
|
x.is_cuda(), \
|
|
#x " must be a CUDA tensor; it is currently on device ", \
|
|
torch_tensor_device_name(x))
|