From 93cd1fa0340db13af085132bf535bc49c2112fbf Mon Sep 17 00:00:00 2001
From: nimlgen <138685161+nimlgen@users.noreply.github.com>
Date: Thu, 5 Sep 2024 16:06:25 +0300
Subject: [PATCH] qcom as_buffer (#6371)

---
 tinygrad/runtime/ops_qcom.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tinygrad/runtime/ops_qcom.py b/tinygrad/runtime/ops_qcom.py
index 2c667772a6..81d07efc3d 100644
--- a/tinygrad/runtime/ops_qcom.py
+++ b/tinygrad/runtime/ops_qcom.py
@@ -313,6 +313,10 @@ class QCOMAllocator(HCQAllocator):
     self.device.synchronize()
     ctypes.memmove(from_mv(dest), src.va_addr, dest.nbytes)
 
+  def as_buffer(self, src:HCQBuffer) -> memoryview:
+    self.device.synchronize()
+    return to_mv(src.va_addr, src.size)
+
   def _free(self, opaque, options:BufferOptions):
     self.device.synchronize()
     self.device._gpu_free(opaque)