Skip to content

Commit

Permalink
fix batch
Browse files Browse the repository at this point in the history
  • Loading branch information
yhmtsai committed Sep 18, 2024
1 parent 4da6a9a commit bd260df
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions core/solver/batch_dispatch.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ using DeviceValueType = gko::kernels::hip::hip_type<ValueType>;

namespace gko {
namespace kernels {
namespace host {
namespace dpcpp {


template <typename T>
Expand All @@ -99,7 +99,7 @@ inline std::decay_t<T> as_device_type(T val)
}


} // namespace host
} // namespace dpcpp
} // namespace kernels
} // namespace gko

Expand Down
2 changes: 1 addition & 1 deletion hip/solver/batch_cg_kernels.hip.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ class kernel_caller {
void launch_apply_kernel(
const gko::kernels::batch_cg::storage_config& sconf, LogType& logger,
PrecType& prec, const BatchMatrixType& mat,
const value_type* const __restrict__ b_values,
const hip_value_type* const __restrict__ b_values,
hip_value_type* const __restrict__ x_values,
hip_value_type* const __restrict__ workspace_data,
const int& block_size, const size_t& shared_size) const
Expand Down

0 comments on commit bd260df

Please sign in to comment.