Skip to content

GitLab

Unverified Commit ab1a93a7 authored by Thor Johnsen's avatar Thor Johnsen Committed by GitHub
Browse files

Merge pull request #1301 from NVIDIA/bug_fix_in_fast_bottleneck 

Change data type for virtual tensors to float
parents 39fc7ccf 51e81314
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
apex/contrib/csrc/bottleneck/bottleneck.cpp
View file @ ab1a93a7
......@@ -237,7 +237,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded,
.setVirtual()
.setId('A') // after add
.setAlignment(16)
.setDataType(dataType)
.setDataType(CUDNN_DATA_FLOAT)
.build(),
cudnn_frontend::TensorBuilder()
.setDim(4, y_dim_padded)
......@@ -245,7 +245,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded,
.setVirtual()
.setId('B') // after bias
.setAlignment(16)
.setDataType(dataType)
.setDataType(CUDNN_DATA_FLOAT)
.build(),
cudnn_frontend::TensorBuilder()
.setDim(4, y_dim_padded)
......@@ -253,7 +253,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded,
.setId('C') // after conv
.setAlignment(16)
.setVirtual()
.setDataType(dataType)
.setDataType(CUDNN_DATA_FLOAT)
.build(),
cudnn_frontend::TensorBuilder()
.setDim(4, y_dim_padded)
......@@ -268,7 +268,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded,
.setId('D') // after optional add
.setAlignment(16)
.setVirtual()
.setDataType(dataType)
.setDataType(CUDNN_DATA_FLOAT)
.build());
}
......@@ -358,7 +358,7 @@ create_dconv_descriptors(int64_t* x_dim_padded,
.setVirtual()
.setId('A') // after dconv
.setAlignment(16)
.setDataType(dataType)
.setDataType(CUDNN_DATA_FLOAT)
.build(),
cudnn_frontend::TensorBuilder()
.setDim(4, x_dim_padded)
......@@ -366,7 +366,7 @@ create_dconv_descriptors(int64_t* x_dim_padded,
.setVirtual()
.setId('B') // after drelu
.setAlignment(16)
.setDataType(dataType)
.setDataType(CUDNN_DATA_FLOAT)
.build());
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment