1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
| #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
|
|
