Skip to content

Commit

Permalink
m_team_scratch_pool -> m_team_scratch_event
Browse files Browse the repository at this point in the history
  • Loading branch information
masterleinad committed Mar 22, 2023
1 parent 33a5d60 commit c5d2c3d
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions core/src/SYCL/Kokkos_SYCL_Instance.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,7 @@ int SYCLInternal::acquire_team_scratch_space() {
&m_current_team_scratch, m_n_team_scratch - 1,
desul::MemoryOrderRelaxed(), desul::MemoryScopeDevice());

m_team_scratch_pool[current_team_scratch].wait_and_throw();
m_team_scratch_event[current_team_scratch].wait_and_throw();

return current_team_scratch;
}
Expand Down Expand Up @@ -172,7 +172,7 @@ sycl::device_ptr<void> SYCLInternal::resize_team_scratch_space(

void SYCLInternal::register_team_scratch_event(int scratch_pool_id,
sycl::event event) {
m_team_scratch_pool[scratch_pool_id] = event;
m_team_scratch_event[scratch_pool_id] = event;
}

uint32_t SYCLInternal::impl_get_instance_id() const { return m_instance_id; }
Expand Down
2 changes: 1 addition & 1 deletion core/src/SYCL/Kokkos_SYCL_Instance.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ class SYCLInternal {
mutable int64_t m_team_scratch_current_size[m_n_team_scratch] = {};
mutable sycl::device_ptr<void> m_team_scratch_ptr[m_n_team_scratch] = {};
mutable int m_current_team_scratch = 0;
mutable sycl::event m_team_scratch_pool[m_n_team_scratch] = {};
mutable sycl::event m_team_scratch_event[m_n_team_scratch] = {};
mutable std::mutex m_team_scratch_mutex;

uint32_t m_instance_id = Kokkos::Tools::Experimental::Impl::idForInstance<
Expand Down

0 comments on commit c5d2c3d

Please sign in to comment.