hkasivis
5e7210980e
* libhsakmt: Update hsakmt_fmm_get_handle to support address range Currently, hsakmt_fmm_get_handle works only if the address is allocated (staring) value. Update it so it can find the handle if address falls in the valid allocated range. This is useful for AMD infinity storage feature where data needs to be transferred to any memory within in the allocated range Signed-off-by: Harish Kasiviswanathan <Harish.Kasiviswanathan@amd.com> * libhsakmt: Introduce AMD Infinity Storage (AIS) API Add hsaKmtAisReadWriteFile() API to support AMD Infinity Storage. The API moves data directly from GPU VRAM to a file. v2: Add in/out ioctl arguments to provide more status information to user space. Modify hsaKmt API also accordingly. Signed-off-by: Harish Kasiviswanathan <Harish.Kasiviswanathan@amd.com> * rocr: Initial implementation of AMD Infinity Storage (AIS) Implement first two API: hsa_amd_ais_file_write and hsa_amd_ais_file_read v2: Change API from hsa_amd_ to hsa_amd_ais_ Change API to take in handle instead of fd for compatibility accross different platforms Original Author: Chris Freehill <Chris.Freehill@amd.com> Signed-off-by: Harish Kasiviswanathan <Harish.Kasiviswanathan@amd.com> --------- Signed-off-by: Harish Kasiviswanathan <Harish.Kasiviswanathan@amd.com>
78 řádky
2.5 KiB
C
78 řádky
2.5 KiB
C
/*
|
|
* Copyright © 2023 Advanced Micro Devices, Inc.
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person
|
|
* obtaining a copy of this software and associated documentation
|
|
* files (the "Software"), to deal in the Software without
|
|
* restriction, including without limitation the rights to use, copy,
|
|
* modify, merge, publish, distribute, sublicense, and/or sell copies
|
|
* of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice (including
|
|
* the next paragraph) shall be included in all copies or substantial
|
|
* portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
|
* HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
* DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
|
|
#include "libhsakmt.h"
|
|
#include "hsakmt/linux/kfd_ioctl.h"
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include "fmm.h"
|
|
|
|
|
|
HSAKMT_STATUS HSAKMTAPI hsaKmtAisReadWriteFile(void *MemoryAddress,
|
|
HSAuint64 MemorySizeInBytes,
|
|
HSAint32 fd,
|
|
HSAint64 file_offset,
|
|
HsaAisFlags AisFlags,
|
|
HSAuint64 *SizeCopiedInBytes,
|
|
HSAint32 *status)
|
|
{
|
|
CHECK_KFD_OPEN();
|
|
|
|
struct kfd_ioctl_ais_args args = {0};
|
|
uint64_t handle, size_offset = MemorySizeInBytes;
|
|
int ret;
|
|
|
|
/* Support is only for dGPUs */
|
|
|
|
|
|
if (!hsakmt_fmm_get_handle(MemoryAddress, &handle, &size_offset)) {
|
|
pr_err("Address/size out of range: %p/%lu\n", MemoryAddress, MemorySizeInBytes);
|
|
return HSAKMT_STATUS_INVALID_PARAMETER;
|
|
}
|
|
|
|
args.in.handle = handle;
|
|
args.in.fd = fd;
|
|
args.in.file_offset = file_offset;
|
|
args.in.size = MemorySizeInBytes;
|
|
if (AisFlags == HSA_AIS_WRITE)
|
|
args.in.op = KFD_IOC_AIS_WRITE;
|
|
else if (AisFlags == HSA_AIS_READ)
|
|
args.in.op = KFD_IOC_AIS_READ;
|
|
else {
|
|
pr_err("Invalid AisFlags: %d\n", AisFlags);
|
|
return HSAKMT_STATUS_INVALID_PARAMETER;
|
|
}
|
|
|
|
args.in.handle_offset = size_offset;
|
|
ret = hsakmt_ioctl(hsakmt_kfd_fd, AMDKFD_IOC_AIS_OP, &args);
|
|
|
|
if (SizeCopiedInBytes)
|
|
*SizeCopiedInBytes = args.out.size_copied;
|
|
if (status)
|
|
*status = args.out.status;
|
|
|
|
return (ret < 0) ? HSAKMT_STATUS_ERROR : HSAKMT_STATUS_SUCCESS;
|
|
}
|