#pragma once #include <ATen/Tensor.h> #include <c10/util/Half.h> #include <cuda.h> #include <cuda_runtime.h> #include <cuda_fp16.h> namespace at { template <> inline __half* Tensor::data() const { return reinterpret_cast<__half*>(data<Half>()); } } // namespace at
