Skip to content

Commit

Permalink
fix compilation
Browse files Browse the repository at this point in the history
  • Loading branch information
upsj committed Jun 1, 2023
1 parent a99f364 commit 3deb1fa
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 3 deletions.
12 changes: 10 additions & 2 deletions core/device_hooks/hip_hooks.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -64,13 +64,21 @@ void HipAllocator::deallocate(void* dev_ptr) const GKO_NOT_COMPILED(hip);
std::shared_ptr<HipExecutor> HipExecutor::create(
int device_id, std::shared_ptr<Executor> master, bool device_reset,
allocation_mode alloc_mode, GKO_HIP_STREAM_STRUCT* stream)
GKO_NOT_COMPILED(hip);
{
return std::shared_ptr<HipExecutor>(
new HipExecutor(device_id, std::move(master),
std::make_shared<HipAllocator>(), stream));
}


std::shared_ptr<HipExecutor> HipExecutor::create(
int device_id, std::shared_ptr<Executor> master,
std::shared_ptr<HipAllocatorBase> alloc, GKO_HIP_STREAM_STRUCT* stream)
GKO_NOT_COMPILED(hip);
{
return std::shared_ptr<HipExecutor>(
new HipExecutor(device_id, std::move(master),
std::make_shared<HipAllocator>(), stream));
}


void HipExecutor::populate_exec_info(const machine_topology* mach_topo)
Expand Down
2 changes: 1 addition & 1 deletion test/utils/executor.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,7 @@ inline void init_executor(std::shared_ptr<gko::ReferenceExecutor> ref,
throw std::runtime_error{"No suitable CUDA devices"};
}
exec = gko::CudaExecutor::create(
0, ref, std::make_shared<gko::CudaAllocator>(stream), stream);
0, ref, std::make_shared<gko::CudaAllocator>(), stream);
}
}

Expand Down

0 comments on commit 3deb1fa

Please sign in to comment.