Add kCUDAExecutionAccelerator constant

triton-inference-server · Jun 5, 2024 · 78225c5 · 78225c5
1 parent b5dab15
commit 78225c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/include/triton/backend/backend_common.h b/include/triton/backend/backend_common.h
@@ -1,4 +1,4 @@
-// Copyright 2020-2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// Copyright 2020-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 //
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions
@@ -279,6 +279,7 @@ constexpr int WILDCARD_DIM = -1;
 
 constexpr char kTensorRTExecutionAccelerator[] = "tensorrt";
 constexpr char kOpenVINOExecutionAccelerator[] = "openvino";
+constexpr char kCUDAExecutionAccelerator[] = "cuda";
 constexpr char kGPUIOExecutionAccelerator[] = "gpu_io";
 constexpr char kAutoMixedPrecisionExecutionAccelerator[] =
     "auto_mixed_precision";