108 lines
2.8 KiB
Common Lisp
108 lines
2.8 KiB
Common Lisp
|
/*
|
||
|
|
Copyright (c) 2012 Advanced Micro Devices, Inc.
|
||
|
|
|
||
|
|
This software is provided 'as-is', without any express or implied warranty.
|
||
|
|
In no event will the authors be held liable for any damages arising from the use of this software.
|
||
|
|
Permission is granted to anyone to use this software for any purpose,
|
||
|
|
including commercial applications, and to alter it and redistribute it freely,
|
||
|
|
subject to the following restrictions:
|
||
|
|
|
||
|
|
1. The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation would be appreciated but is not required.
|
||
|
|
2. Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software.
|
||
|
|
3. This notice may not be removed or altered from any source distribution.
|
||
|
|
*/
|
||
|
|
//Originally written by Takahiro Harada
|
||
|
|
|
||
|
|
|
||
|
|
#pragma OPENCL EXTENSION cl_amd_printf : enable
|
||
|
|
#pragma OPENCL EXTENSION cl_khr_local_int32_base_atomics : enable
|
||
|
|
|
||
|
|
typedef unsigned int u32;
|
||
|
|
#define GET_GROUP_IDX get_group_id(0)
|
||
|
|
#define GET_LOCAL_IDX get_local_id(0)
|
||
|
|
#define GET_GLOBAL_IDX get_global_id(0)
|
||
|
|
#define GET_GROUP_SIZE get_local_size(0)
|
||
|
|
#define GROUP_LDS_BARRIER barrier(CLK_LOCAL_MEM_FENCE)
|
||
|
|
#define GROUP_MEM_FENCE mem_fence(CLK_LOCAL_MEM_FENCE)
|
||
|
|
#define AtomInc(x) atom_inc(&(x))
|
||
|
|
#define AtomInc1(x, out) out = atom_inc(&(x))
|
||
|
|
|
||
|
|
#define make_uint4 (uint4)
|
||
|
|
#define make_uint2 (uint2)
|
||
|
|
#define make_int2 (int2)
|
||
|
|
|
||
|
|
typedef struct
|
||
|
|
{
|
||
|
|
union
|
||
|
|
{
|
||
|
|
int4 m_data;
|
||
|
|
uint4 m_unsignedData;
|
||
|
|
float m_floatData;
|
||
|
|
};
|
||
|
|
int m_offset;
|
||
|
|
int m_n;
|
||
|
|
int m_padding[2];
|
||
|
|
} ConstBuffer;
|
||
|
|
|
||
|
|
|
||
|
|
__kernel
|
||
|
|
__attribute__((reqd_work_group_size(64,1,1)))
|
||
|
|
void FillIntKernel(__global int* dstInt, int num_elements, int value, const int offset)
|
||
|
|
{
|
||
|
|
int gIdx = GET_GLOBAL_IDX;
|
||
|
|
|
||
|
|
if( gIdx < num_elements )
|
||
|
|
{
|
||
|
|
dstInt[ offset+gIdx ] = value;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
__kernel
|
||
|
|
__attribute__((reqd_work_group_size(64,1,1)))
|
||
|
|
void FillFloatKernel(__global float* dstFloat, int num_elements, float value, const int offset)
|
||
|
|
{
|
||
|
|
int gIdx = GET_GLOBAL_IDX;
|
||
|
|
|
||
|
|
if( gIdx < num_elements )
|
||
|
|
{
|
||
|
|
dstFloat[ offset+gIdx ] = value;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
__kernel
|
||
|
|
__attribute__((reqd_work_group_size(64,1,1)))
|
||
|
|
void FillUnsignedIntKernel(__global unsigned int* dstInt, const int num, const unsigned int value, const int offset)
|
||
|
|
{
|
||
|
|
int gIdx = GET_GLOBAL_IDX;
|
||
|
|
|
||
|
|
if( gIdx < num )
|
||
|
|
{
|
||
|
|
dstInt[ offset+gIdx ] = value;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
__kernel
|
||
|
|
__attribute__((reqd_work_group_size(64,1,1)))
|
||
|
|
void FillInt2Kernel(__global int2* dstInt2, const int num, const int2 value, const int offset)
|
||
|
|
{
|
||
|
|
int gIdx = GET_GLOBAL_IDX;
|
||
|
|
|
||
|
|
if( gIdx < num )
|
||
|
|
{
|
||
|
|
dstInt2[ gIdx + offset] = make_int2( value.x, value.y );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
__kernel
|
||
|
|
__attribute__((reqd_work_group_size(64,1,1)))
|
||
|
|
void FillInt4Kernel(__global int4* dstInt4, const int num, const int4 value, const int offset)
|
||
|
|
{
|
||
|
|
int gIdx = GET_GLOBAL_IDX;
|
||
|
|
|
||
|
|
if( gIdx < num )
|
||
|
|
{
|
||
|
|
dstInt4[ offset+gIdx ] = value;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|