Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions src/fortran/gpu_api.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -140,6 +140,11 @@ extern "C" int gpuEventSynchronize(cudaEvent_t event)
return static_cast<int>(cudaEventSynchronize(event));
}

extern "C" int gpuStreamWaitEvent(cudaStream_t stream, cudaEvent_t event)
{
return static_cast<int>(cudaStreamWaitEvent(stream, event));
}

extern "C" int gpuMemcpyAsync(void* dst, const void* src, size_t bytes,
cudaMemcpyKind kind, cudaStream_t stream)
{
Expand Down Expand Up @@ -182,6 +187,11 @@ extern "C" int gpuEventRecord(hipEvent_t event, hipStream_t stream)
return static_cast<int>(hipEventRecord(event, stream));
}

extern "C" int gpuStreamWaitEvent(hipStream_t stream, hipEvent_t event)
{
return static_cast<int>(hipStreamWaitEvent(stream, event));
}

extern "C" int gpuEventSynchronize(hipEvent_t event)
{
return static_cast<int>(hipEventSynchronize(event));
Expand Down
6 changes: 6 additions & 0 deletions src/fortran/gpu_api_interface.F90
Original file line number Diff line number Diff line change
Expand Up @@ -149,6 +149,12 @@ function gpuEventRecord(event, streamid) bind(c,name="gpuEventRecord")
integer(C_INT) :: gpuEventRecord
end function gpuEventRecord

function gpuStreamWaitEvent(streamid, event) bind(c,name="gpuStreamWaitEvent")
import
type(C_PTR),value :: event,streamid
integer(C_INT) :: StreamWaitEvent
end function gpuStreamWaitEvent

function gpuEventSynchronize(event) bind(c,name="gpuEventSynchronize")
import
type(C_PTR),value :: event
Expand Down