mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 06:34:03 -05:00
09f3aae169
* checkout files from autogen branch
* ioctl with payload
* fix am generations
* properly fix generations
This reverts commit b2a54f4f41.
* revert discovery.h
* support pragma pack(1)
* typo
* better getter
* typo
* NVCEC0_QMDV05_00_RELEASE[01]_ENABLE
* align support
* anon handling fix
---------
Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
54 lines
2.2 KiB
C
54 lines
2.2 KiB
C
/*
|
|
* NVIDIA_COPYRIGHT_BEGIN
|
|
*
|
|
* Copyright (c) 2022, 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.
|
|
*
|
|
* NVIDIA_COPYRIGHT_END
|
|
*/
|
|
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
|
|
typedef enum {
|
|
NVJITLINK_SUCCESS = 0,
|
|
NVJITLINK_ERROR_UNRECOGNIZED_OPTION,
|
|
NVJITLINK_ERROR_MISSING_ARCH,
|
|
NVJITLINK_ERROR_INVALID_INPUT,
|
|
NVJITLINK_ERROR_PTX_COMPILE,
|
|
NVJITLINK_ERROR_NVVM_COMPILE,
|
|
NVJITLINK_ERROR_INTERNAL
|
|
} nvJitLinkResult;
|
|
|
|
typedef enum {
|
|
NVJITLINK_INPUT_NONE = 0,
|
|
NVJITLINK_INPUT_CUBIN = 1,
|
|
NVJITLINK_INPUT_PTX,
|
|
NVJITLINK_INPUT_LTOIR,
|
|
NVJITLINK_INPUT_FATBIN,
|
|
NVJITLINK_INPUT_OBJECT,
|
|
NVJITLINK_INPUT_LIBRARY
|
|
} nvJitLinkInputType;
|
|
|
|
typedef struct nvJitLink* nvJitLinkHandle;
|
|
|
|
nvJitLinkResult nvJitLinkCreate(nvJitLinkHandle *handle, uint32_t numOptions, const char **options);
|
|
nvJitLinkResult nvJitLinkDestroy(nvJitLinkHandle *handle);
|
|
nvJitLinkResult nvJitLinkAddData(nvJitLinkHandle handle, nvJitLinkInputType inputType, const void *data, size_t size, const char *name);
|
|
nvJitLinkResult nvJitLinkAddFile(nvJitLinkHandle handle, nvJitLinkInputType inputType, const char *fileName);
|
|
nvJitLinkResult nvJitLinkComplete(nvJitLinkHandle handle);
|
|
nvJitLinkResult nvJitLinkGetLinkedCubinSize(nvJitLinkHandle handle, size_t *size);
|
|
nvJitLinkResult nvJitLinkGetLinkedCubin(nvJitLinkHandle handle, void *cubin);
|
|
nvJitLinkResult nvJitLinkGetLinkedPtxSize(nvJitLinkHandle handle, size_t *size);
|
|
nvJitLinkResult nvJitLinkGetLinkedPtx(nvJitLinkHandle handle, char *ptx);
|
|
nvJitLinkResult nvJitLinkGetErrorLogSize(nvJitLinkHandle handle, size_t *size);
|
|
nvJitLinkResult nvJitLinkGetErrorLog(nvJitLinkHandle handle, char *log);
|
|
nvJitLinkResult nvJitLinkGetInfoLogSize(nvJitLinkHandle handle, size_t *size);
|
|
nvJitLinkResult nvJitLinkGetInfoLog(nvJitLinkHandle handle, char *log);
|
|
nvJitLinkResult nvJitLinkVersion(unsigned int *major, unsigned int *minor);
|