Skip to content

GitLab

"...composable_kernel_rocm.git" did not exist on "770d2b77253b5bfbcc794d4133e7ecada63cdd44"
Commit aa9e2e56 authored by Jehandad Khan's avatar Jehandad Khan
Browse files

upgrade to ROCm 5.1 and fix formatting

parent 8dad1124
Show whitespace changes
Inline Side-by-side
Showing with 20 additions and 21 deletions
Dockerfile
View file @ aa9e2e56
FROM ubuntu:18.04 FROM ubuntu:18.04
ARG ROCMVERSION=5.0 ARG ROCMVERSION=5.1
ARG OSDB_BKC_VERSION ARG OSDB_BKC_VERSION
RUN set -xe RUN set -xe
......
library/src/utility/conv_fwd_util.cpp
View file @ aa9e2e56
...@@ -77,8 +77,8 @@ ConvParams::ConvParams(ck::index_t n_dim, ...@@ -77,8 +77,8 @@ ConvParams::ConvParams(ck::index_t n_dim,
conv_filter_dilations.size() != num_dim_spatial || conv_filter_dilations.size() != num_dim_spatial ||
input_left_pads.size() != num_dim_spatial || input_right_pads.size() != num_dim_spatial) input_left_pads.size() != num_dim_spatial || input_right_pads.size() != num_dim_spatial)
{ {
throw(std::runtime_error( throw(
"ConvParams::GetOutputSpatialLengths: " std::runtime_error("ConvParams::GetOutputSpatialLengths: "
"parameter size is different from number of declared dimensions!")); "parameter size is different from number of declared dimensions!"));
} }
} }
...@@ -91,8 +91,8 @@ std::vector<ck::index_t> ConvParams::GetOutputSpatialLengths() const ...@@ -91,8 +91,8 @@ std::vector<ck::index_t> ConvParams::GetOutputSpatialLengths() const
conv_filter_dilations.size() != num_dim_spatial || conv_filter_dilations.size() != num_dim_spatial ||
input_left_pads.size() != num_dim_spatial || input_right_pads.size() != num_dim_spatial) input_left_pads.size() != num_dim_spatial || input_right_pads.size() != num_dim_spatial)
{ {
throw(std::runtime_error( throw(
"ConvParams::GetOutputSpatialLengths: " std::runtime_error("ConvParams::GetOutputSpatialLengths: "
"parameter size is different from number of declared dimensions!")); "parameter size is different from number of declared dimensions!"));
} }
...@@ -101,8 +101,7 @@ std::vector<ck::index_t> ConvParams::GetOutputSpatialLengths() const ...@@ -101,8 +101,7 @@ std::vector<ck::index_t> ConvParams::GetOutputSpatialLengths() const
{ {
// XEff = (X - 1) * conv_dilation_w + 1; // XEff = (X - 1) * conv_dilation_w + 1;
// Wo = (Wi + in_left_pad_w + in_right_pad_w - XEff) / conv_stride_w + 1; // Wo = (Wi + in_left_pad_w + in_right_pad_w - XEff) / conv_stride_w + 1;
const ck::index_t idx_eff = const ck::index_t idx_eff = (filter_spatial_lengths[i] - 1) * conv_filter_dilations[i] + 1;
(filter_spatial_lengths[i] - 1) * conv_filter_dilations[i] + 1;
out_spatial_len[i] = out_spatial_len[i] =
(input_spatial_lengths[i] + input_left_pads[i] + input_right_pads[i] - idx_eff) / (input_spatial_lengths[i] + input_left_pads[i] + input_right_pads[i] - idx_eff) /
conv_filter_strides[i] + conv_filter_strides[i] +
......
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