From 78225c56a28097f1b575879014f9f0118a2b5008 Mon Sep 17 00:00:00 2001 From: krishung5 Date: Wed, 5 Jun 2024 16:39:42 -0700 Subject: [PATCH] Add kCUDAExecutionAccelerator constant --- include/triton/backend/backend_common.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/include/triton/backend/backend_common.h b/include/triton/backend/backend_common.h index 84c889e..152af8d 100644 --- 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";