File: data-type-support.rst

package info (click to toggle)
rocprim 6.4.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 11,428 kB
  • sloc: cpp: 153,383; python: 1,397; sh: 404; xml: 217; makefile: 119
file content (49 lines) | stat: -rw-r--r-- 826 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
.. meta::
   :description: rocPRIM API library data type support
   :keywords: rocPRIM, ROCm, API library, API reference, data type, support

.. _data-type-support:

******************************************
Data type support
******************************************

The following table shows the supported input and output datatypes.

  .. list-table:: Supported Input/Output Types
    :header-rows: 1
    :name: supported-input-output-types

    *
      - Type
      - Support
    *
      - int8
      - ✅
    *
      - int16
      - ✅
    *
      - int32
      - ✅
    *
      - int64
      - ✅
    *
      - float8
      - ❌
    *
      - float16
      - ✅
    *
      - bfloat16      
      - ✅
    *
      - tensorfloat32
      - ❌
    *
      - float32
      - ✅
    *
      - float64
      - ✅