MCPcopy Create free account
hub / github.com/apache/tvm-ffi / PrintDLDataTypeCodeAsStr

Function PrintDLDataTypeCodeAsStr

src/ffi/dtype.cc:77–153  ·  view source on GitHub ↗

* \brief Convert a DLDataTypeCode to a string. * \param os The output stream. * \param type_code The DLDataTypeCode to convert. */

Source from the content-addressed store, hash-verified

75 * \param type_code The DLDataTypeCode to convert.
76 */
77inline void PrintDLDataTypeCodeAsStr(std::ostream& os, DLDataTypeCode type_code) { // NOLINT(*)
78 switch (static_cast<int>(type_code)) {
79 case kDLInt: {
80 os << "int";
81 break;
82 }
83 case kDLUInt: {
84 os << "uint";
85 break;
86 }
87 case kDLFloat: {
88 os << "float";
89 break;
90 }
91 case kDLOpaqueHandle: {
92 os << "handle";
93 break;
94 }
95 case kDLBfloat: {
96 os << "bfloat";
97 break;
98 }
99 case kDLFloat8_e3m4: {
100 os << "float8_e3m4";
101 break;
102 }
103 case kDLFloat8_e4m3: {
104 os << "float8_e4m3";
105 break;
106 }
107 case kDLFloat8_e4m3b11fnuz: {
108 os << "float8_e4m3b11fnuz";
109 break;
110 }
111 case kDLFloat8_e4m3fn: {
112 os << "float8_e4m3fn";
113 break;
114 }
115 case kDLFloat8_e4m3fnuz: {
116 os << "float8_e4m3fnuz";
117 break;
118 }
119 case kDLFloat8_e5m2: {
120 os << "float8_e5m2";
121 break;
122 }
123 case kDLFloat8_e5m2fnuz: {
124 os << "float8_e5m2fnuz";
125 break;
126 }
127 case kDLFloat8_e8m0fnu: {
128 os << "float8_e8m0fnu";
129 break;
130 }
131 case kDLFloat6_e2m3fn: {
132 os << "float6_e2m3fn";
133 break;
134 }

Callers

nothing calls this directly

Calls 1

Tested by

no test coverage detected