Commit 1eb85ac6 authored by Qianfeng Zhang's avatar Qianfeng Zhang
Browse files

Use StreamCaptureModeRelaxed for grouped fwd example to include more work into capturing scope

parent 5ef0843b
......@@ -313,6 +313,8 @@ int run(int argc, char* argv[])
HIP_CHECK_ERROR(hipStreamCreate(&stream));
HIP_CHECK_ERROR(hipGraphCreate(&graph, 0));
HIP_CHECK_ERROR(hipStreamBeginCapture(stream, hipStreamCaptureModeRelaxed));
argument =
gemm.MakeArgument(p_a,
p_b0,
......@@ -336,8 +338,6 @@ int run(int argc, char* argv[])
gemm.SetWorkSpacePointer(&argument, problem_desc_workspace_verify.GetDeviceBuffer());
HIP_CHECK_ERROR(hipStreamBeginCapture(stream, hipStreamCaptureModeGlobal));
invoker.Run(argument, StreamConfig{nullptr, false});
HIP_CHECK_ERROR(hipStreamEndCapture(stream, &graph));
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment