From 30e8817a97a3d9afd34d2f2a1a30e09f1efbd61d Mon Sep 17 00:00:00 2001
From: Ennis Blank <ez46emyj@stud.informatik.uni-erlangen.de>
Date: Mon, 30 Jan 2023 11:54:40 +0100
Subject: [PATCH] updated fill_tensor() function
---
src/cudaTensor.cc | 7 ++-----
1 file changed, 2 insertions(+), 5 deletions(-)
diff --git a/src/cudaTensor.cc b/src/cudaTensor.cc
index c9a940f..b653c89 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
--
GitLab