Save some WIP

Author: caugonnet

c/experimental/stf/CMakeLists.txt

@@ -29,12 +29,12 @@ endif()
 
 find_package(CUDAToolkit REQUIRED)
 set_target_properties(cccl.c.experimental.stf PROPERTIES CUDA_RUNTIME_LIBRARY STATIC)
+target_link_libraries(cccl.c.experimental.stf PUBLIC CCCL::cudax)
 target_link_libraries(cccl.c.experimental.stf PRIVATE
   CUDA::cudart_static
   CUDA::nvrtc
   CUDA::nvJitLink
   CUDA::cuda_driver
-  CCCL::cudax
 )
 # target_compile_definitions(cccl.c.experimental.stf PUBLIC CCCL_C_EXPERIMENTAL=1)
 # target_compile_definitions(cccl.c.experimental.stf PRIVATE

c/experimental/stf/include/cccl/c/experimental/stf/stf.h

@@ -18,13 +18,20 @@ typedef struct stf_ctx_handle_t* stf_ctx_handle;
 void stf_ctx_create(stf_ctx_handle* ctx);
 void stf_ctx_finalize(stf_ctx_handle ctx);
 
+// TODO stf_ctx_set_mode() + define enum with GRAPH, STREAM, ...
+// TODO stf_ctx_is_graph()
+
+cudaStream_t stf_fence(stf_ctx_handle ctx);
+
 typedef struct stf_logical_data_handle_t* stf_logical_data_handle;
 
 void stf_logical_data(stf_ctx_handle ctx, stf_logical_data_handle* ld, void* addr, size_t sz);
 void stf_logical_data_set_symbol(stf_logical_data_handle ld, const char* symbol);
 void stf_logical_data_destroy(stf_logical_data_handle ld);
 
-// TODO token
+// TODO
+// void stf_logical_data_wait(stf_logical_data_handle ld);
+// void stf_token(stf_ctx_handle ctx);
 
 typedef struct stf_task_handle_t* stf_task_handle;
 
@@ -34,8 +41,21 @@ void stf_task_add_dep(stf_task_handle t, stf_logical_data_handle ld, stf_access_
 void stf_task_start(stf_task_handle t);
 void stf_task_end(stf_task_handle t);
 cudaStream_t stf_task_get_stream(stf_task_handle t);
+void *stf_task_get(stf_task_handle t, size_t submitted_index);
 void stf_task_destroy(stf_task_handle t);
 
+typedef struct stf_kernel_desc_handle_t *stf_kernel_desc_handle;
+
+void stf_kernel_create(stf_kernel_desc_handle *d);
+void stf_kernel_destroy(stf_kernel_desc_handle d);
+// TODO stf_cuda_kernel_desc : symbol, deps, args... ?
+// void stf_kernel_set_symbol((stf_kernel_handle k, const char* symbol)
+// void stf_kernel_add_dep(stf_kernel_handle k, stf_logical_data_handle ld, stf_access_mode m);
+// void stf_kernel_start(stf_kernel_handle k);
+// void stf_kernel_set_args(stf_kernel_handle k, size_t cnt, void **args);
+// void stf_kernel_end(stf_kernel_handle k);
+// void stf_kernel_destroy(stf_kernel_handle k);
+
 #ifdef __cplusplus
 }
 #endif

c/experimental/stf/src/stf.cu

@@ -36,6 +36,12 @@ void stf_ctx_finalize(stf_ctx_handle ctx)
   delete ctx;
 }
 
+cudaStream_t stf_fence(stf_ctx_handle ctx)
+{
+   assert(ctx);
+   return ctx->ctx.fence();
+}
+
 void stf_logical_data(stf_ctx_handle ctx, stf_logical_data_handle* ld, void* addr, size_t sz)
 {
   assert(ld);

c/experimental/stf/test/test_task.cpp

@@ -13,7 +13,7 @@
 
 #include <cccl/c/experimental/stf/stf.h>
 
-C2H_TEST("basic stf task", "[task]")
+C2H_TEST("empty stf tasks", "[task]")
 {
   size_t N = 1000000;