saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-25 22:26:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update softmax_grad_impl.cu: add constexpr (#15794)

Browse source 
### Description
Add a "constexpr" keyword to fix a static analysis warning
This commit is contained in:
Changming Sun 2023-05-04 08:10:17 -07:00 committed by GitHub
parent df7424e11f
commit 34fcdd83c8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
orttraining/orttraining/training_ops/cuda/math/softmax_grad_impl.cu
View file

@ -268,7 +268,7 @@ Status SoftmaxGradImpl(cudaStream_t stream, cudnnHandle_t cudnn_handle, T* input
#define CASE_LOG2_ELEMENTS(log2_elements_value) \
case log2_elements_value: { \
if (log2_elements_value < start_to_use_register_efficient_func) { \
if constexpr (log2_elements_value < start_to_use_register_efficient_func) { \
LAUNCH_KERNEL(log2_elements_value, softmax_warp_backward); \
} else { \
LAUNCH_KERNEL(log2_elements_value, softmax_warp_backward_register_efficicent); \