// Copyright 2023 The IREE Authors
//
// Licensed under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#ifndef IREE_BINDINGS_PYTHON_NUMPY_INTEROP_H_
#define IREE_BINDINGS_PYTHON_NUMPY_INTEROP_H_

#include "./binding.h"
#include "iree/hal/api.h"

namespace iree::python::numpy {

// Creates a numpy.dtype object for an IREE element type.
py::object DescrNewFromType(iree_hal_element_type_t t);

// Creates a numpy array from data using numpy.frombuffer and reshapes it.
// The caller must ensure the underlying memory outlives the returned array
// (e.g. via py::keep_alive on the binding).
py::object SimpleNewFromData(int nd, intptr_t const* dims,
                             py::handle dtype_descr, void* data);

}  // namespace iree::python::numpy

#endif  // IREE_BINDINGS_PYTHON_NUMPY_INTEROP_H_