Add /proc/gpu#/local_memory API for getting VRAM size

1 files changed, 27 insertions, 0 deletions

diff --git a/nvdebug.h b/nvdebug.h
index ff35f70..b13d311 100644
--- a/nvdebug.h
+++ b/nvdebug.h
@@ -1145,7 +1145,34 @@ typedef union {
 } page_tbl_entry_v0_t;
 */
 
+/* VRAM Information
 
+  If ECC is disabled:
+    bytes = (magnitude << scale) * 1024 * 1024
+  If ECC is enabled:
+    bytes = ((magnitude << scale) * 1024 * 1024) / 16 * 15
+
+  Support: Pascal, Volta, Turing, [more?]
+ */
+#define NV_FB_MMU_LOCAL_MEMORY_RANGE 0x00100ce0
+typedef union  {
+        struct {
+                uint32_t scale:4;
+                uint32_t mag:6;
+                uint32_t:20;
+                bool is_ecc:1;
+                uint32_t:1;
+        } __attribute__((packed));
+        uint32_t raw;
+} memory_range_t;
+
+static inline uint64_t memory_range_to_bytes(memory_range_t range) {
+        // ECC takes a byte out of available memory for parity data
+        if (range.is_ecc)
+                return ((range.mag << range.scale) * 1024ull * 1024ull) / 16 * 15;
+        else
+                return (range.mag << range.scale) * 1024ull * 1024ull;
+}
 
 /* Begin nvdebug types and functions */