File tree Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -23,8 +23,14 @@ Tensor Mapping APIs
2323Motivation
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
2935API
3036--------------------------------------------------------------------------------
@@ -61,7 +67,7 @@ Changelog
6167Support
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
6672Contributors
6773--------------------------------------------------------------------------------
You can’t perform that action at this time.
0 commit comments