v4.0 update. (#2371)
This commit is contained in:
@@ -446,7 +446,7 @@ public:
|
||||
|
||||
Status
|
||||
construct_graph(bool launch_with_pdl) {
|
||||
#if ((__CUDACC_VER_MAJOR__ >= 12) && (__CUDACC_VER_MINOR__ >= 4))
|
||||
#if (__CUDACC_VER_MAJOR__ > 12 || (__CUDACC_VER_MAJOR__ == 12 && __CUDACC_VER_MINOR__ >= 6))
|
||||
Status status = Status::kSuccess;
|
||||
|
||||
// Destroy existing graph, if created
|
||||
|
||||
@@ -47,7 +47,7 @@ void launch_full_barrier(
|
||||
cudaStream_t stream,
|
||||
bool launch_with_pdl) {
|
||||
|
||||
#if (__CUDACC_VER_MAJOR__ > 12 || (__CUDACC_VER_MAJOR__ == 12 && __CUDACC_VER_MINOR__ >= 4))
|
||||
#if (__CUDACC_VER_MAJOR__ > 12 || (__CUDACC_VER_MAJOR__ == 12 && __CUDACC_VER_MINOR__ >= 6))
|
||||
// Legacy (kernel) launch with PDL
|
||||
cudaLaunchAttribute attributes[1];
|
||||
attributes[0].id = cudaLaunchAttributeProgrammaticStreamSerialization;
|
||||
|
||||
Reference in New Issue
Block a user