Skip to content

Commit

Permalink
#1849: NOMERGE: Add driver/runtime version check for CUDA
Browse files Browse the repository at this point in the history
  • Loading branch information
JacobDomagala committed Jun 19, 2022
1 parent ad8f3a2 commit 157f66a
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions tests/unit/active/test_async_op_cuda.cc
Original file line number Diff line number Diff line change
Expand Up @@ -180,6 +180,14 @@ struct CUDAGroup {
};

TEST_F(TestAsyncOp, test_async_op_cuda) {
int driverVer;
cudaDriverGetVersion(&driverVer);

int runtimeVer;
cudaRuntimeGetVersion(&runtimeVer);

fmt::print("CUDA driver version is {} and runtime {}\n", driverVer, runtimeVer);

auto const this_node = theContext()->getNode();
auto p = theObjGroup()->makeCollective<CUDAGroup>("test_async_op_cuda");
auto ep = theTerm()->makeEpochRooted(term::UseDS{true});
Expand Down

0 comments on commit 157f66a

Please sign in to comment.