hipTensor documentation

hipTensor is a high-performance HIP C++ library for accelerating tensor primitives. It leverages specialized GPU matrix cores on the latest AMD discrete GPUs. The hipTensor API is designed to be portable with other similar libraries, letting other library users easily migrate to the AMD platform. The hipTensor library is a work in progress (WIP). For more information, see What is hipTensor?

The hipTensor library is included in the rocm-libraries public repository, and it is located at ROCm/rocm-libraries.

Note

The hipTensor repository for ROCm 7.1.1 and earlier is located at ROCm/hipTensor.

Install

• Installation guide

Conceptual

• Programming guide

• Plan cache

How to

• Transition to hipTensor 2.0

• Contribute to hipTensor

Examples

• Samples

API reference

• API reference guide

To contribute to the documentation, see Contributing to ROCm.

You can find licensing information on the Licensing page.