diff options
Diffstat (limited to 'src/backends')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/DebugTestImpl.cpp | 3 | ||||
-rw-r--r-- | src/backends/reference/workloads/Debug.cpp | 5 |
2 files changed, 6 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/DebugTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/DebugTestImpl.cpp index fa9a825313..e2fa8dbf5a 100644 --- a/src/backends/backendsCommon/test/layerTests/DebugTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/DebugTestImpl.cpp @@ -70,7 +70,10 @@ LayerTestResult<T, Dim> DebugTestImpl( if (toFile) { + // Given that this is dependent on an ExNet switch, we need to explicitly set the directory that the + // files are stored in as this happens within the ExNet flow fs::path tmpDir = fs::temp_directory_path(); + armnnUtils::Filesystem::CreateDirectory("/ArmNNIntermediateLayerOutputs"); std::string full_path = tmpDir.generic_string() + "/ArmNNIntermediateLayerOutputs/" + layerName + ".numpy"; ExecuteWorkload(*workload, memoryManager); diff --git a/src/backends/reference/workloads/Debug.cpp b/src/backends/reference/workloads/Debug.cpp index fdadfef590..f4f9792fc1 100644 --- a/src/backends/reference/workloads/Debug.cpp +++ b/src/backends/reference/workloads/Debug.cpp @@ -103,9 +103,10 @@ void Debug(const TensorInfo& inputInfo, { if (outputsToFile) { - auto rootPathToFile = armnnUtils::Filesystem::CreateDirectory("/ArmNNIntermediateLayerOutputs"); - std::ofstream out(rootPathToFile + layerName + ".numpy"); + fs::path tmpDir = fs::temp_directory_path(); + std::ofstream out(tmpDir.generic_string() + "/ArmNNIntermediateLayerOutputs/" + layerName + ".numpy"); PrintOutput<T>(inputInfo, inputData, guid, layerName, slotIndex, out); + out.close(); } else { |