Skip to content
Snippets Groups Projects
Commit d9c094f6 authored by Maxence Naud's avatar Maxence Naud
Browse files

Merge branch 'fix_207-producer-clone' into 'dev'

Change code lines order to avoid pointer error

See merge request eclipse/aidge/aidge_backend_cuda!59
parents 75102058 0c730402
No related branches found
No related tags found
2 merge requests!61v0.4.0,!59Change code lines order to avoid pointer error
Pipeline #61259 passed
...@@ -37,7 +37,7 @@ TEST_CASE("CUDA test") { ...@@ -37,7 +37,7 @@ TEST_CASE("CUDA test") {
} }
// Allocate device memory // 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_a), sizeof(float) * N);
cudaMalloc(reinterpret_cast<void**>(&d_b), sizeof(float) * N); cudaMalloc(reinterpret_cast<void**>(&d_b), sizeof(float) * N);
cudaMalloc(reinterpret_cast<void**>(&d_out), sizeof(float) * N); cudaMalloc(reinterpret_cast<void**>(&d_out), sizeof(float) * N);
...@@ -46,9 +46,9 @@ TEST_CASE("CUDA test") { ...@@ -46,9 +46,9 @@ TEST_CASE("CUDA test") {
cudaMemcpy(d_a, a, sizeof(float) * N, cudaMemcpyHostToDevice); cudaMemcpy(d_a, a, sizeof(float) * N, cudaMemcpyHostToDevice);
cudaMemcpy(d_b, b, 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); vector_add(d_out, d_a, d_b, N);
// Transfer data back to host memory // Transfer data back to host memory
cudaMemcpy(out, d_out, sizeof(float) * N, cudaMemcpyDeviceToHost); cudaMemcpy(out, d_out, sizeof(float) * N, cudaMemcpyDeviceToHost);
...@@ -71,7 +71,6 @@ TEST_CASE("CUDA test") { ...@@ -71,7 +71,6 @@ TEST_CASE("CUDA test") {
TEST_CASE("Tensor creation", "[Connector]") { TEST_CASE("Tensor creation", "[Connector]") {
SECTION("from const array") { SECTION("from const array") {
Tensor x; Tensor x;
x.setBackend("cuda");
x = Array3D<int,2,2,2>{ x = Array3D<int,2,2,2>{
{ {
{ {
...@@ -83,6 +82,7 @@ TEST_CASE("Tensor creation", "[Connector]") { ...@@ -83,6 +82,7 @@ TEST_CASE("Tensor creation", "[Connector]") {
{7, 8} {7, 8}
} }
}}; }};
x.setBackend("cuda");
REQUIRE(x.nbDims() == 3); REQUIRE(x.nbDims() == 3);
REQUIRE(x.dims()[0] == 2); REQUIRE(x.dims()[0] == 2);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment