Spaces:
Running
Running
| //------------------------------------------------------------------------------ | |
| // silu | |
| //------------------------------------------------------------------------------ | |
| kernel void kernel_silu( | |
| global float * src0, | |
| ulong offset0, | |
| global float * dst, | |
| ulong offsetd | |
| ) { | |
| src0 = (global float*)((global char*)src0 + offset0); | |
| dst = (global float*)((global char*)dst + offsetd); | |
| float x = src0[get_global_id(0)]; | |
| dst[get_global_id(0)] = x / (1.0f + exp(-x)); | |
| } | |
| kernel void kernel_silu_4( | |
| global float4 * src0, | |
| ulong offset0, | |
| global float4 * dst, | |
| ulong offsetd | |
| ) { | |
| src0 = (global float4*)((global char*)src0 + offset0); | |
| dst = (global float4*)((global char*)dst + offsetd); | |
| float4 x = src0[get_global_id(0)]; | |
| dst[get_global_id(0)] = x / (1.0f + exp(-x)); | |
| } | |