Skip to content

Commit

Permalink
[cuda] Port over allocator and buffer implementation
Browse files Browse the repository at this point in the history
This commit ports over existing CUDA driver allocator and buffer
implementation. The main logic is kept as-is, with one noticeable
changes--context wrapper is dropped and fields in it are directly
put in various API calls. This is to make supporting multiple
device and stream easier later. Other changes are just polishing
on comments and errors.
  • Loading branch information
antiagainst committed Jun 7, 2023
1 parent e56c1ae commit 6c237e7
Show file tree
Hide file tree
Showing 5 changed files with 221 additions and 181 deletions.
4 changes: 4 additions & 0 deletions experimental/cuda2/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,10 @@ iree_cc_library(
"api.h"
SRCS
"api.h"
"cuda_allocator.c"
"cuda_allocator.h"
"cuda_buffer.c"
"cuda_buffer.h"
"cuda_driver.c"
DEPS
::dynamic_symbols
Expand Down
Loading

0 comments on commit 6c237e7

Please sign in to comment.