cupva_device.h

Fully qualified name: src/device/vpu_runtime/include/cupva_device.h

File members: src/device/vpu_runtime/include/cupva_device.h

/*
 * Copyright (c) 2020-2023 NVIDIA Corporation.  All rights reserved.
 *
 * NVIDIA Corporation and its licensors retain all intellectual property
 * and proprietary rights in and to this software, related documentation
 * and any modifications thereto.  Any use, reproduction, disclosure or
 * distribution of this software and related documentation without an express
 * license agreement from NVIDIA Corporation is strictly prohibited.
 */
#ifndef CUPVA_DEVICE_H
#define CUPVA_DEVICE_H

#if CUPVA_DEVICE != 1
#    error "cupva_device.h should only be used in VPU device code."
#endif

#if CUPVA_BUILD_MODE == CUPVA_NATIVE
#    include <assert.h>
#    include <vpu_native.h>
#endif

#include <cupva_device/device_core.h>
#include <cupva_device/dynamic_dataflow.h>
#include <cupva_device/gatherscatter_dataflow.h>
#include <cupva_device/sampler.h>
#include <cupva_device/sequence_dataflow.h>
#include <cupva_device/tensor_dataflow.h>

#if CUPVA_BUILD_MODE == CUPVA_NATIVE
#    include <cupva_device_native.h>
#else // CUPVA_BUILD_MODE != CUPVA_NATIVE
#    include <cupva_device_vpu.h>
#endif // CUPVA_BUILD_MODE == CUPVA_NATIVE

#endif // CUPVA_DEVICE_H