Commit c899dc8b authored by zhangyue's avatar zhangyue
Browse files

issue/209: delete N assert

parent 280d942c
......@@ -54,6 +54,8 @@ __device__ void launchOp(
size_t output_index,
Args... args) {
static_assert(N == Op::num_inputs, "template N is not equal to Op::num_inputs!\n");
#pragma unroll
// Copy inputs to buf
for (size_t i = 0; i < N; i++) {
......@@ -87,9 +89,6 @@ __global__ void elementwiseKernel(
const void *const *inputs,
Args... args) {
// Only support 3 mode elementwise
static_assert(N < 4, "elementwise Kernel support mode < 4 calculate");
int cid = core_id();
int ncores = core_num();
if (cid >= ncores) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment