Skip to content

Commit 2f5ff27

Browse files
committed
Add clarifications in extension documentation
1 parent b1a3286 commit 2f5ff27

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

scripts/core/EXP-TENSOR-MAP.rst

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,14 @@ Tensor Mapping APIs
2323
Motivation
2424
--------------------------------------------------------------------------------
2525

26-
Used to target the CUDA entry points cuTensorMapEncodeIm2col and
27-
cuTensorMapEncodeTiled.
26+
Used to target the CUDA entry points ``cuTensorMapEncodeIm2col`` and
27+
``cuTensorMapEncodeTiled``.
28+
29+
For some tensor core operations on ``sm_90+`` Nvidia devices, a tensor
30+
descriptor must be built on the host and passed to the kernel. The interfaces
31+
mentioned above, and mapped to UR in this extension, provide the APIs necessary
32+
to create these tensor descriptor objects, that can then be passed to the
33+
kernels.
2834

2935
API
3036
--------------------------------------------------------------------------------
@@ -61,7 +67,7 @@ Changelog
6167
Support
6268
--------------------------------------------------------------------------------
6369

64-
This is only supported in the CUDA adapter.
70+
This extension is only supported on the ``UR_PLATFORM_BACKEND_CUDA`` backend.
6571

6672
Contributors
6773
--------------------------------------------------------------------------------

0 commit comments

Comments
 (0)