Skip to content
Merged
Changes from 1 commit
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
Prev Previous commit
Next Next commit
Add missing file
  • Loading branch information
caugonnet committed Jan 23, 2025
commit ed79e03476db5dd3702b33c19eb4157711c32e86
42 changes: 42 additions & 0 deletions cudax/test/stf/dot/sections_2.cu
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
//===----------------------------------------------------------------------===//
//
// Part of CUDASTF in CUDA C++ Core Libraries,
// under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// SPDX-FileCopyrightText: Copyright (c) 2022-2024 NVIDIA CORPORATION & AFFILIATES.
//
//===----------------------------------------------------------------------===//

/**
* @file
* @brief This test makes sure we can generate a dot file with sections
*/

#include <cuda/experimental/stf.cuh>

using namespace cuda::experimental::stf;

int main()
{
// TODO (miscco): Make it work for windows
#if !_CCCL_COMPILER(MSVC)
context ctx;
auto lA = ctx.logical_token().set_symbol("A");
auto lB = ctx.logical_token().set_symbol("B");
auto lC = ctx.logical_token().set_symbol("C");
ctx.dot_push_section("foo");
for (size_t i = 0; i < 2; i++)
{
auto guard = ctx.dot_section("bar");
ctx.task(lA.read(), lB.rw()).set_symbol("t1")->*[](cudaStream_t, auto, auto) {};
for (size_t j = 0; j < 2; j++) {
auto inner_guard = ctx.dot_section("baz");
ctx.task(lA.read(), lC.rw()).set_symbol("t2")->*[](cudaStream_t, auto, auto) {};
ctx.task(lB.read(), lC.read(), lA.rw()).set_symbol("t3")->*[](cudaStream_t, auto, auto, auto) {};
}
}
ctx.dot_pop_section();
ctx.finalize();
#endif // !_CCCL_COMPILER(MSVC)
}
Loading