From 4da0a3a5369142078fd7c681e3f0f1817bd6e2f3 Mon Sep 17 00:00:00 2001
From: skal <pascal.massimino@gmail.com>
Date: Wed, 11 Feb 2026 10:51:06 +0100
Subject: add --save-intermediates to train.py and cnn_test

---
 src/gpu/texture_readback.h | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'src/gpu/texture_readback.h')

diff --git a/src/gpu/texture_readback.h b/src/gpu/texture_readback.h
index 1bf770f..8230e13 100644
--- a/src/gpu/texture_readback.h
+++ b/src/gpu/texture_readback.h
@@ -20,4 +20,14 @@ std::vector<uint8_t> read_texture_pixels(
     int width,
     int height);
 
+// Read RGBA16Float texture and convert to BGRA8Unorm for saving
+// Converts [-1,1] float range to [0,255] uint8 range
+// Returns: width * height * 4 bytes (BGRA8)
+std::vector<uint8_t> texture_readback_fp16_to_u8(
+    WGPUDevice device,
+    WGPUQueue queue,
+    WGPUTexture texture,
+    int width,
+    int height);
+
 #endif // !defined(STRIP_ALL)
-- 
cgit v1.2.3