Skip to content
Snippets Groups Projects

Change code lines order to avoid pointer error

Merged Maxence Naud requested to merge fix_207-producer-clone into dev
1 file
+ 4
4
Compare changes
  • Side-by-side
  • Inline
@@ -37,7 +37,7 @@ TEST_CASE("CUDA test") {
}
// Allocate device memory
float *d_a, *d_b, *d_out;
float *d_a, *d_b, *d_out;
cudaMalloc(reinterpret_cast<void**>(&d_a), sizeof(float) * N);
cudaMalloc(reinterpret_cast<void**>(&d_b), sizeof(float) * N);
cudaMalloc(reinterpret_cast<void**>(&d_out), sizeof(float) * N);
@@ -46,9 +46,9 @@ TEST_CASE("CUDA test") {
cudaMemcpy(d_a, a, sizeof(float) * N, cudaMemcpyHostToDevice);
cudaMemcpy(d_b, b, sizeof(float) * N, cudaMemcpyHostToDevice);
// Executing kernel
// Executing kernel
vector_add(d_out, d_a, d_b, N);
// Transfer data back to host memory
cudaMemcpy(out, d_out, sizeof(float) * N, cudaMemcpyDeviceToHost);
@@ -71,7 +71,6 @@ TEST_CASE("CUDA test") {
TEST_CASE("Tensor creation", "[Connector]") {
SECTION("from const array") {
Tensor x;
x.setBackend("cuda");
x = Array3D<int,2,2,2>{
{
{
@@ -83,6 +82,7 @@ TEST_CASE("Tensor creation", "[Connector]") {
{7, 8}
}
}};
x.setBackend("cuda");
REQUIRE(x.nbDims() == 3);
REQUIRE(x.dims()[0] == 2);
Loading