InfiniTensor/include/cuda/cuda_utility.h

11 lines
226 B
C++

#include "core/tensor.h"
namespace infini {
void cudaPrintFloat(float *x, int len);
void cudaPrintTensor(const Tensor &tensor) {
cudaPrintFloat(tensor->getRawDataPtr<float *>(), tensor->size());
}
} // namespace infini