xref: /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/CUDATensorMethods.cuh (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #pragma once
2 
3 #include <ATen/Tensor.h>
4 #include <c10/util/Half.h>
5 
6 #include <cuda.h>
7 #include <cuda_runtime.h>
8 #include <cuda_fp16.h>
9 
10 namespace at {
11 template <>
data() const12 inline __half* Tensor::data() const {
13   return reinterpret_cast<__half*>(data<Half>());
14 }
15 } // namespace at
16