Skip to content

Commit

Permalink
fix cuda and cholesky
Browse files Browse the repository at this point in the history
  • Loading branch information
ikbuibui committed Jul 12, 2024
1 parent dba4166 commit 0d7aaa3
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 5 deletions.
5 changes: 2 additions & 3 deletions examples/cholesky.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,7 @@

#include <iostream>

template<typename TTask>
void print_matrix(std::vector<redGrapes::IOResource<double*, TTask>> A, int nblks, int blocksize)
void print_matrix(std::vector<redGrapes::IOResource<double*>> A, int nblks, int blocksize)
{
for(int ia = 0; ia < nblks; ++ia)
{
Expand Down Expand Up @@ -71,7 +70,7 @@ int main(int argc, char* argv[])
Alin[i * N + i] += N;

// initialize tiled matrix in column-major layout
std::vector<redGrapes::IOResource<double*, RGTask>> A(nblks * nblks);
std::vector<redGrapes::IOResource<double*>> A(nblks * nblks);

// allocate each tile (also in column-major layout)
for(size_t j = 0; j < nblks; ++j)
Expand Down
4 changes: 2 additions & 2 deletions examples/cuda_mandelbrot.cu
Original file line number Diff line number Diff line change
Expand Up @@ -93,8 +93,8 @@ int main()
size_t height = 4096;
size_t area = width * height;

redGrapes::IOResource<Color*, RGTask> host_buffer;
redGrapes::IOResource<Color*, RGTask> device_buffer;
redGrapes::IOResource<Color*> host_buffer;
redGrapes::IOResource<Color*> device_buffer;

rg.emplace_task(
[area](auto host_buffer)
Expand Down

0 comments on commit 0d7aaa3

Please sign in to comment.