CUTLASS: host_tensor.h File Reference

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

• Main Page

• Modules

• Namespaces

• Classes

• Files

• File List

• File Members

• tools

• util

• include

• cutlass

• util

Classes | Namespaces

host_tensor.h File Reference

HostTensor contributes management for both host and device memory. More...

#include <vector>
#include "cutlass/cutlass.h"
#include "cutlass/matrix_traits.h"
#include "cutlass/tensor_ref.h"
#include "cutlass/tensor_view.h"
#include "device_memory.h"

Include dependency graph for host_tensor.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

|

Classes

| | class | cutlass::HostTensor< Element_, Layout_ > | | | Host tensor. More...
| | |

|

Namespaces

| | | cutlass | | |

Detailed Description

HostTensor allocates host and device memory upon construction. Basic element-wise operations on host memory synchronize device memory automatically. Explicit copy operations provide abstractions for CUDA memcpy operations.

Call {host, device}_{data, ref, view}() for accessing host or device memory.

See cutlass/tensor_ref.h and cutlass/tensor_view.h for more details.

---

Generated by 1.8.11