kCudaUtils.h

Go to the documentation of this file.

#ifndef K_FIRESYNC_CUDA_UTILS_H
#define K_FIRESYNC_CUDA_UTILS_H

#include <kApi/Data/kArrayProvider.h>
#include <kFireSync/Cuda/kCudaDef.h>
#include <kFireSync/Cuda/kCudaAlloc.h>
#include <kFireSync/Cuda/kCudaUtils.x.h>


kFsFx(kCudaEnvironmentStatus) kCudaUtils_EnvironmentStatus();



template<class T>
kInlineFx(kStatus) kCudaUtils_AllocValue(kCudaStream stream, kAlloc allocator, T** pointer, kCudaSync sync = kCUDA_SYNC_DEFAULT)
{
    return xkCudaUtils_AllocValue(stream, allocator, pointer, sync); 
}

template<class T>
kInlineFx(kStatus) kCudaUtils_AssignValue(kCudaStream stream, T* dest, const T* source, kCudaSync sync = kCUDA_SYNC_DEFAULT)
{
    return kCudaUtils_Copy(stream, dest, source, sizeof(T), sync);
}

template<class T>
kInlineFx(kStatus) kCudaUtils_CloneValue(kCudaStream stream, kAlloc destAlloc, T** dest, const T* source, kCudaSync sync = kCUDA_SYNC_DEFAULT)
{
    return xkCudaUtils_CloneValue(stream, destAlloc, dest, source, sync); 
}

template<class T>
kInlineFx(kStatus) kCudaUtils_FreeValueRef(kAlloc allocator, T** pointer)
{
    return kAlloc_FreeRef(allocator, pointer);
}



kFsFx(kStatus) kCudaUtils_Allocate(kCudaStream stream, kAlloc alloc, kSize size, void* mem, kCudaSync sync = kCUDA_SYNC_DEFAULT);

kInlineFx(kStatus) kCudaUtils_AttachToStream(kCudaStream stream, void* pointer, kCudaSync sync = kCUDA_SYNC_DEFAULT)
{
    return kCudaStream_AttachMemory(xkCudaStream_Fallback(stream), pointer, sync); 
}
   
kInlineFx(kStatus) kCudaUtils_DetachFromStream(kCudaStream stream, void* pointer, kCudaSync sync = kCUDA_SYNC_DEFAULT)
{
    return kCudaStream_DetachMemory(xkCudaStream_Fallback(stream), pointer, sync); 
}

kFsFx(kStatus) kCudaUtils_MemSet(kCudaStream stream, void* dest, kByte fill, kSize size, kCudaSync sync = kCUDA_SYNC_DEFAULT);

kFsFx(kStatus) kCudaUtils_Copy(kCudaStream stream, void* dest, const void* src, kSize size, kCudaSync sync = kCUDA_SYNC_DEFAULT);



kInlineFx(kStatus) kCudaUtils_AttachArrayToStream(kCudaStream stream, kArrayProvider array, kCudaSync sync = kCUDA_SYNC_DEFAULT)
{
    return kCudaUtils_AttachToStream(stream, kArrayProvider_Data(array), sync);
}

kInlineFx(kStatus) kCudaUtils_DetachArrayFromStream(kCudaStream stream, kArrayProvider array, kCudaSync sync = kCUDA_SYNC_DEFAULT)
{
    return kCudaUtils_DetachFromStream(stream, kArrayProvider_Data(array), sync);
}

kFsFx(kStatus) kCudaUtils_CloneObject(kCudaStream stream, kObject* object, kObject source, kAlloc objectAllocator, kAlloc valueAllocator, kCudaSync sync = kCUDA_SYNC_DEFAULT);

kFsFx(kStatus) kCudaUtils_AssignArray(kCudaStream stream, kArrayProvider destination, kArrayProvider source, kCudaSync sync = kCUDA_SYNC_DEFAULT);

kInlineFx(kBool) kCudaUtils_IsArrayDeviceAccessible(kArrayProvider array, kBool isExclusive = kTRUE)
{
    kAllocTrait traits = kAlloc_Traits(kArrayProvider_ValueAlloc(array)); 

    return kAllocTrait_IsCudaDevice(traits) || kAllocTrait_IsCudaPinned(traits) || (kAllocTrait_IsCudaManaged(traits) && (!kAllocTrait_IsSerial(traits) || isExclusive));
}

kFsFx(kStatus) kCudaUtils_ConstructArrayFromTemplate(kCudaStream stream, kArrayProvider* outputArray, kArrayProvider templateArray, kAlloc objectAlloc = kNULL, kAlloc valueAlloc = kNULL, kCudaSync sync = kCUDA_SYNC_DEFAULT);

kFsFx(kStatus) kCudaUtils_PrepareDeviceArray(kCudaStream stream, kArrayProvider* outputArray, kArrayProvider inputArray, kBool copyData = kTRUE, kAlloc objectAlloc = kNULL, kAlloc valueAlloc = kNULL, kBool inputExclusive = kTRUE, kCudaSync sync = kCUDA_SYNC_DEFAULT);


#endif