mirror of https://github.com/NVIDIA/cuda-samples.git synced 2024-10-06 20:29:18 +08:00

History

Allard Hendriksen 5925483b33 Add TMA example		2023-06-30 17:39:53 +02:00
..
globalToShmemTMACopy.cu	Add TMA example	2023-06-30 17:39:53 +02:00
README.md	Add TMA example	2023-06-30 17:39:53 +02:00
util.h	Add TMA example	2023-06-30 17:39:53 +02:00

README.md

globalToShmemTMACopy - Global Memory to Shared Memory TMA Copy

Description

This sample shows how to use the CUDA driver API and inline PTX assembly to copy a 2D tile of a tensor into shared memory. It also demonstrates arrive-wait barrier for synchronization.

Key Concepts

CUDA Runtime API, CUDA Driver API, PTX ISA, CPP11 CUDA

Supported SM Architectures

This sample requires compute capability 9.0 or higher.

SM 9.0

Supported OSes

Linux, Windows, QNX

Supported CPU Architecture

x86_64, ppc64le, armv7l, aarch64

CUDA APIs involved

CUDA Runtime API

cudaMalloc, cudaMemcpy, cudaFree, cudaDeviceSynchronize

CUDA Driver API

cudaMalloc, cudaMemcpy, cudaFree, cudaDeviceSynchronize

CUDA PTX ISA

Dependencies needed to build/run

CPP11

Prerequisites

Download and install the CUDA Toolkit 12.2 for your corresponding platform. Make sure the dependencies mentioned in Dependencies section above are installed.