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
Show 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, ...@@ -237,7 +237,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded,
.setVirtual() .setVirtual()
.setId('A') // after add .setId('A') // after add
.setAlignment(16) .setAlignment(16)
.setDataType(dataType) .setDataType(CUDNN_DATA_FLOAT)
.build(), .build(),
cudnn_frontend::TensorBuilder() cudnn_frontend::TensorBuilder()
.setDim(4, y_dim_padded) .setDim(4, y_dim_padded)
...@@ -245,7 +245,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded, ...@@ -245,7 +245,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded,
.setVirtual() .setVirtual()
.setId('B') // after bias .setId('B') // after bias
.setAlignment(16) .setAlignment(16)
.setDataType(dataType) .setDataType(CUDNN_DATA_FLOAT)
.build(), .build(),
cudnn_frontend::TensorBuilder() cudnn_frontend::TensorBuilder()
.setDim(4, y_dim_padded) .setDim(4, y_dim_padded)
...@@ -253,7 +253,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded, ...@@ -253,7 +253,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded,
.setId('C') // after conv .setId('C') // after conv
.setAlignment(16) .setAlignment(16)
.setVirtual() .setVirtual()
.setDataType(dataType) .setDataType(CUDNN_DATA_FLOAT)
.build(), .build(),
cudnn_frontend::TensorBuilder() cudnn_frontend::TensorBuilder()
.setDim(4, y_dim_padded) .setDim(4, y_dim_padded)
...@@ -268,7 +268,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded, ...@@ -268,7 +268,7 @@ create_conv_bias_add_act_descriptors(int64_t* x_dim_padded,
.setId('D') // after optional add .setId('D') // after optional add
.setAlignment(16) .setAlignment(16)
.setVirtual() .setVirtual()
.setDataType(dataType) .setDataType(CUDNN_DATA_FLOAT)
.build()); .build());
} }
...@@ -358,7 +358,7 @@ create_dconv_descriptors(int64_t* x_dim_padded, ...@@ -358,7 +358,7 @@ create_dconv_descriptors(int64_t* x_dim_padded,
.setVirtual() .setVirtual()
.setId('A') // after dconv .setId('A') // after dconv
.setAlignment(16) .setAlignment(16)
.setDataType(dataType) .setDataType(CUDNN_DATA_FLOAT)
.build(), .build(),
cudnn_frontend::TensorBuilder() cudnn_frontend::TensorBuilder()
.setDim(4, x_dim_padded) .setDim(4, x_dim_padded)
...@@ -366,7 +366,7 @@ create_dconv_descriptors(int64_t* x_dim_padded, ...@@ -366,7 +366,7 @@ create_dconv_descriptors(int64_t* x_dim_padded,
.setVirtual() .setVirtual()
.setId('B') // after drelu .setId('B') // after drelu
.setAlignment(16) .setAlignment(16)
.setDataType(dataType) .setDataType(CUDNN_DATA_FLOAT)
.build()); .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