InfiniTensor/include/cuda/cuda_expand.h

#pragma once

#include "operators/unary.h"
#include "utils/small_array.h"
namespace infini {
void expandKernel(int dType, void *input, void *output, int nDims,
                  int outputsize, SmallArray inputShape,
                  SmallArray outputShape);

void expandRowKernel(int dType, void *input, void *output, int n_rows,
                     int row_len);
}; // namespace infini