fix todo

zhouwg · Jun 18, 2024 · 4d70039 · 4d70039
1 parent 9456bba
commit 4d70039
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ggml-qnn.cpp b/ggml-qnn.cpp
@@ -1995,8 +1995,7 @@ class ggml_qnn_tensor_readwrite
         if (is_npu) {
             qnn_instance * instance = ctx->instance;
             uint8_t *qnn_buffer = static_cast<uint8_t *>(instance->alloc_rpcmem(
-                    ggml_nbytes(tensor), 4)); // TODO: should we get the align param from device here?
-            if (!qnn_buffer) {
+                    ggml_nbytes(tensor), alignof(void*)));
                 QNN_LOG_WARN("alloc rpcmem failure, %s\n", strerror(errno));
                 QNN_LOG_DEBUG("tensor%p name %s", _qnn_tensor, QNN_TENSOR_GET_NAME(*_qnn_tensor));
                 _context = nullptr;