(value)
| 169 | |
| 170 | |
| 171 | def get_memory(value): |
| 172 | from paddle.base.core import DataType |
| 173 | |
| 174 | numel = value.numel() |
| 175 | mapping = { |
| 176 | DataType.BOOL: 1, |
| 177 | DataType.INT8: 1, |
| 178 | DataType.INT16: 2, |
| 179 | DataType.INT32: 4, |
| 180 | DataType.INT64: 8, |
| 181 | DataType.UINT8: 1, |
| 182 | DataType.UINT16: 2, |
| 183 | DataType.UINT32: 4, |
| 184 | DataType.UINT64: 8, |
| 185 | DataType.FLOAT32: 4, |
| 186 | DataType.FLOAT64: 8, |
| 187 | } |
| 188 | dtype = mapping[value.type().dtype] |
| 189 | return dtype * numel |
| 190 | |
| 191 | |
| 192 | def analysis_io(program: paddle.pir.Program): |
no test coverage detected