"configs/git@developer.sourcefind.cn:OpenDAS/opencompass.git" did not exist on "17b8e929dd80670c890c63de0efaa52ed35e8308"
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[]) ...@@ -313,6 +313,8 @@ int run(int argc, char* argv[])
HIP_CHECK_ERROR(hipStreamCreate(&stream)); HIP_CHECK_ERROR(hipStreamCreate(&stream));
HIP_CHECK_ERROR(hipGraphCreate(&graph, 0)); HIP_CHECK_ERROR(hipGraphCreate(&graph, 0));
HIP_CHECK_ERROR(hipStreamBeginCapture(stream, hipStreamCaptureModeRelaxed));
argument = argument =
gemm.MakeArgument(p_a, gemm.MakeArgument(p_a,
p_b0, p_b0,
...@@ -336,8 +338,6 @@ int run(int argc, char* argv[]) ...@@ -336,8 +338,6 @@ int run(int argc, char* argv[])
gemm.SetWorkSpacePointer(&argument, problem_desc_workspace_verify.GetDeviceBuffer()); gemm.SetWorkSpacePointer(&argument, problem_desc_workspace_verify.GetDeviceBuffer());
HIP_CHECK_ERROR(hipStreamBeginCapture(stream, hipStreamCaptureModeGlobal));
invoker.Run(argument, StreamConfig{nullptr, false}); invoker.Run(argument, StreamConfig{nullptr, false});
HIP_CHECK_ERROR(hipStreamEndCapture(stream, &graph)); 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