fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2025-12-24 21:50:12 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

teflon/tests: Take into account signedness when checking the output tensors

Browse source 
Otherwise, it won't be able to apply the tolerance in all cases.

Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/34481>
This commit is contained in:
Tomeu Vizoso 2024-12-06 11:41:46 +01:00 • committed by Marge Bot
parent b3746305ea
commit 554fb8af11
1 changed files with 20 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/gallium/targets/teflon/test_teflon.cpp
View file

@ -81,7 +81,24 @@ test_model(void *buf, size_t buf_size, std::string cache_dir, unsigned tolerance
}
break;
}
default: {
case kTfLiteInt8: {
int8_t *cpu = ((int8_t**)cpu_output)[i];
int8_t *npu = ((int8_t**)npu_output)[i];
if (abs(cpu[j] - npu[j]) > tolerance) {
std::cout << "CPU: ";
for (int k = 0; k < std::min(int(output_sizes[i]), 24); k++)
std::cout << std::setfill('0') << std::setw(2) << std::hex << int(cpu[k]) << " ";
std::cout << "\n";
std::cout << "NPU: ";
for (int k = 0; k < std::min(int(output_sizes[i]), 24); k++)
std::cout << std::setfill('0') << std::setw(2) << std::hex << int(npu[k]) << " ";
std::cout << "\n";
FAIL() << "Output at " << j << " from the NPU (" << std::setfill('0') << std::setw(2) << std::hex << int(npu[j]) << ") doesn't match that from the CPU (" << std::setfill('0') << std::setw(2) << std::hex << int(cpu[j]) << ").";
}
break;
}
case kTfLiteUInt8: {
uint8_t *cpu = ((uint8_t**)cpu_output)[i];
uint8_t *npu = ((uint8_t**)npu_output)[i];
if (abs(cpu[j] - npu[j]) > tolerance) {
@ -98,6 +115,8 @@ test_model(void *buf, size_t buf_size, std::string cache_dir, unsigned tolerance
}
break;
}
default:
assert(!"Unsupported data type for output tensor");
}
}
}