Backends/SiaNet.Backend.TensorSharp/Cuda/KernelOps/ReduceAllOp.cs
/// <summary>
/// Invokes the reduce all pass2.
/// </summary>
/// <param name="context">The context.</param>
/// <param name="cudaContext">The cuda context.</param>
/// <param name="ptx">The PTX.</param>
/// <param name="kernelName">Name of the kernel.</param>
/// <param name="grid">The grid.</param>
/// <param name="block">The block.</param>
/// <param name="smemSize">Size of the smem.</param>
/// <param name="index32">if set to <c>true</c> [index32].</param>
/// <param name="args">The arguments.</param>
Invokes the reduce all pass2.
Invokes the reduce all pass2 .
public static void InvokeReduceAllPass2(TSCudaContext context, CudaContext cudaContext, byte[] ptx, string kernelName, dim3 grid, dim3 block, uint smemSize, bool index32, params object[] args)
        {
            var config = new ApplySpecialization(index32).GetConfig();
            var kernel = context.KernelCache.Get(cudaContext, ptx, kernelName);
            kernel.GridDimensions = grid;
            kernel.BlockDimensions = block;
            kernel.DynamicSharedMemory = smemSize;
            kernel.Run(args);            
        }
public static void InvokeReduceAllPass2 ( TSCudaContext context , CudaContext cudaContext , byte [ ] ptx , string kernelName , dim3 grid , dim3 block , uint smemSize , bool index32 , params object [ ] args ) { var config = new ApplySpecialization ( index32 ) . GetConfig ( ) ; var kernel = context . KernelCache . Get ( cudaContext , ptx , kernelName ) ; kernel . GridDimensions = grid ; kernel . BlockDimensions = block ; kernel . DynamicSharedMemory = smemSize ; kernel . Run ( args ) ; }
Invokes the reduce all pass2.
Invokes the reduce all pass2 .
context
TSCudaContext
None
cudaContext
CudaContext
The cuda context.
The cuda context .
ptx
byte[]
None
kernelName
string
Name of the kernel.
Name of the kernel .
grid
dim3
None
block
dim3
None
smemSize
uint
Size of the smem.
Size of the smem .
index32
bool
if set to true
if set to true
args
None