diff --git a/src/cudaTensor.cc b/src/cudaTensor.cc index c9a940f7afcb38cfd9774126875d76e39c3f2be2..b653c8978888c4ec70cfae9e8d02e2f24e661f2c 100644 --- a/src/cudaTensor.cc +++ b/src/cudaTensor.cc @@ -214,9 +214,6 @@ float& CudaTensor::operator[](const shape_t& access) { } - - - void CudaTensor::load_data(float* data, int n) { cudaError_t cudaStatus = cudaMemcpy( (void*) dat, data, n*sizeof(float), cudaMemcpyHostToDevice @@ -235,7 +232,7 @@ std::ostream& CudaTensor::print_shape(std::ostream& os) { while (it != shape.end()) { os << ", " << *(it++); } - os << ")"; + os << ")" <<std::endl; return os; } @@ -308,7 +305,7 @@ void CudaTensor::compare(Tensor& rhs) { void CudaTensor::fillTensor() { float* data = (float*) malloc(size()*sizeof(float)); for (int i = 0; i < size(); i++) { - data[i] = static_cast <float> (rand()) / (static_cast <float> (RAND_MAX/256)); + data[i] = static_cast <float> (rand()/float(RAND_MAX)*24.f+1.f); } load_data(data, size()); } \ No newline at end of file