p95768021
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

style:fix style

This commit is contained in:
OdinaryWord 2023-12-14 13:35:08 +08:00
parent 2af4c1276b
commit 046b2d68d8
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/kernels/cuda/gather.cc
View File

@ -24,10 +24,11 @@ class GatherCuda : public CudaKernelWithoutConfig {
} else if (op->getDType() == DataType::Float16) { } else if (op->getDType() == DataType::Float16) {
gather_kernel<half>((half *)inputData, (half *)outputData, metaData, gather_kernel<half>((half *)inputData, (half *)outputData, metaData,
op->getOutput()->size()); op->getOutput()->size());
} } else if (op->getDType() == DataType::Int8) {
else if (op->getDType() == DataType::Int8) { gather_kernel<int8_t>((int8_t *)inputData, (int8_t *)outputData,
gather_kernel<int8_t>((int8_t *)inputData, (int8_t *)outputData, metaData, metaData, op->getOutput()->size());
op->getOutput()->size()); } else {
IT_ASSERT(false);
} }
} }
}; };