|
18 | 18 |
|
19 | 19 | #ifdef MLIR_CRUNNERUTILS_DEFINE_FUNCTIONS
|
20 | 20 |
|
21 |
| -#define AART |
22 |
| - |
23 | 21 | #include <algorithm>
|
24 | 22 | #include <cassert>
|
25 | 23 | #include <cctype>
|
@@ -174,82 +172,8 @@ class SparseTensorStorage : public SparseTensorStorageBase {
|
174 | 172 | }
|
175 | 173 | // Then setup the tensor.
|
176 | 174 | traverse(tensor, sparsity, 0, nnz, 0);
|
177 |
| -#ifdef AART |
178 |
| - dump(); |
179 |
| -#endif |
180 | 175 | }
|
181 | 176 |
|
182 |
| -#ifdef AART |
183 |
| - void dump() { |
184 |
| - fprintf(stderr, "++++++++++ rank=%lu +++++++++++\n", sizes.size()); |
185 |
| - if constexpr (std::is_same_v<P, uint64_t>) |
186 |
| - fprintf(stderr, "PTR64 "); |
187 |
| - else if constexpr (std::is_same_v<P, uint32_t>) |
188 |
| - fprintf(stderr, "PTR32 "); |
189 |
| - else if constexpr (std::is_same_v<P, uint16_t>) |
190 |
| - fprintf(stderr, "PTR16 "); |
191 |
| - else if constexpr (std::is_same_v<P, uint8_t>) |
192 |
| - fprintf(stderr, "PTR8 "); |
193 |
| - if constexpr (std::is_same_v<I, uint64_t>) |
194 |
| - fprintf(stderr, "INDX64 "); |
195 |
| - else if constexpr (std::is_same_v<I, uint32_t>) |
196 |
| - fprintf(stderr, "INDX32 "); |
197 |
| - else if constexpr (std::is_same_v<I, uint16_t>) |
198 |
| - fprintf(stderr, "INDX16 "); |
199 |
| - else if constexpr (std::is_same_v<I, uint8_t>) |
200 |
| - fprintf(stderr, "INDX8 "); |
201 |
| - if constexpr (std::is_same_v<V, double>) |
202 |
| - fprintf(stderr, "VALF64\n"); |
203 |
| - else if constexpr (std::is_same_v<V, float>) |
204 |
| - fprintf(stderr, "VALF32\n"); |
205 |
| - else if constexpr (std::is_same_v<V, int64_t>) |
206 |
| - fprintf(stderr, "VALI64\n"); |
207 |
| - else if constexpr (std::is_same_v<V, int32_t>) |
208 |
| - fprintf(stderr, "VALI32\n"); |
209 |
| - else if constexpr (std::is_same_v<V, int16_t>) |
210 |
| - fprintf(stderr, "VALI16\n"); |
211 |
| - else if constexpr (std::is_same_v<V, int8_t>) |
212 |
| - fprintf(stderr, "VALI8\n"); |
213 |
| - for (uint64_t r = 0; r < sizes.size(); r++) { |
214 |
| - fprintf(stderr, "dim %lu #%lu\n", r, sizes[r]); |
215 |
| - fprintf(stderr, " positions[%lu] #%lu :", r, pointers[r].size()); |
216 |
| - for (uint64_t i = 0; i < pointers[r].size(); i++) |
217 |
| - if constexpr (std::is_same_v<P, uint64_t>) |
218 |
| - fprintf(stderr, " %lu", pointers[r][i]); |
219 |
| - else if constexpr (std::is_same_v<P, uint32_t>) |
220 |
| - fprintf(stderr, " %u", pointers[r][i]); |
221 |
| - else if constexpr (std::is_same_v<P, uint16_t>) |
222 |
| - fprintf(stderr, " %u", pointers[r][i]); |
223 |
| - else if constexpr (std::is_same_v<P, uint8_t>) |
224 |
| - fprintf(stderr, " %u", pointers[r][i]); |
225 |
| - fprintf(stderr, "\n indices[%lu] #%lu :", r, indices[r].size()); |
226 |
| - for (uint64_t i = 0; i < indices[r].size(); i++) |
227 |
| - if constexpr (std::is_same_v<I, uint64_t>) |
228 |
| - fprintf(stderr, " %lu", indices[r][i]); |
229 |
| - else if constexpr (std::is_same_v<I, uint32_t>) |
230 |
| - fprintf(stderr, " %u", indices[r][i]); |
231 |
| - else if constexpr (std::is_same_v<I, uint16_t>) |
232 |
| - fprintf(stderr, " %u", indices[r][i]); |
233 |
| - else if constexpr (std::is_same_v<I, uint8_t>) |
234 |
| - fprintf(stderr, " %u", indices[r][i]); |
235 |
| - fprintf(stderr, "\n"); |
236 |
| - } |
237 |
| - fprintf(stderr, "values #%lu :", values.size()); |
238 |
| - for (uint64_t i = 0; i < values.size(); i++) |
239 |
| - if constexpr (std::is_same_v<V, double>) |
240 |
| - fprintf(stderr, " %lf", values[i]); |
241 |
| - else if constexpr (std::is_same_v<V, float>) |
242 |
| - fprintf(stderr, " %f", values[i]); |
243 |
| - else if constexpr (std::is_same_v<V, int32_t>) |
244 |
| - fprintf(stderr, " %d", values[i]); |
245 |
| - else if constexpr (std::is_same_v<V, int16_t>) |
246 |
| - fprintf(stderr, " %d", values[i]); |
247 |
| - else if constexpr (std::is_same_v<V, int8_t>) |
248 |
| - fprintf(stderr, " %d", values[i]); |
249 |
| - fprintf(stderr, "\n+++++++++++++++++++++++++++++\n"); |
250 |
| - } |
251 |
| -#endif |
252 |
| - |
253 | 177 | virtual ~SparseTensorStorage() {}
|
254 | 178 |
|
255 | 179 | uint64_t getRank() const { return sizes.size(); }
|
@@ -490,16 +414,6 @@ void *openTensorC(char *filename, uint64_t *idata) {
|
490 | 414 | // Close the file and return sorted tensor.
|
491 | 415 | fclose(file);
|
492 | 416 | tensor->sort(); // sort lexicographically
|
493 |
| -#if 1 |
494 |
| - const std::vector<Element> &elements = tensor->getElements(); |
495 |
| - for (uint64_t k = 1; k < nnz; k++) { |
496 |
| - uint64_t same = 0; |
497 |
| - for (uint64_t r = 0; r < rank; r++) |
498 |
| - if (elements[k].indices[r] == elements[k - 1].indices[r]) |
499 |
| - same++; |
500 |
| - assert(same < rank && "duplicate element"); |
501 |
| - } |
502 |
| -#endif |
503 | 417 | return tensor;
|
504 | 418 | }
|
505 | 419 |
|
|
0 commit comments