mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-03-23 00:27:38 +00:00
069500464d
* [Compiler] Addressing new compiler warnings Clang enables new lifetime warnings in production and we see build errors due to this with the staging compiler. The attributes added in this PR are suggested by the compiler. However, I'm not very familiar with the code base, so the changes may be incorrect. * Update some more instances * Adds file-level ignores via clang diagnostic pragma The number of instances was large, so I decided to use file-level scope to disable the warning via pragma clang diagnostic ignored. It also showed this warning coming from the gtest dependency. For that, I did add the respective command line flag to the CMake variables. I don't know if this is acceptable or not. * This adds the remaining instances For a build on gfx90a. * fix clang format * Adding couple more instances from gfx1200 build * Fixed another few instances --------- Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com> Co-authored-by: illsilin_amdeng <Illia.Silin@amd.com>
115 lines
3.9 KiB
C++
115 lines
3.9 KiB
C++
// Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#pragma once
|
|
|
|
#include <cstdlib>
|
|
#include <numeric>
|
|
#include <iterator>
|
|
#include <vector>
|
|
|
|
#include "ck/ck.hpp"
|
|
|
|
#include "ck/library/utility/numeric.hpp"
|
|
|
|
namespace ck {
|
|
namespace utils {
|
|
namespace conv {
|
|
|
|
struct ConvParam
|
|
{
|
|
ConvParam();
|
|
ConvParam(ck::index_t n_dim,
|
|
ck::index_t group_count,
|
|
ck::index_t n_batch,
|
|
ck::index_t n_out_channels,
|
|
ck::index_t n_in_channels,
|
|
const std::vector<ck::index_t>& filters_len,
|
|
const std::vector<ck::index_t>& input_len,
|
|
const std::vector<ck::index_t>& strides,
|
|
const std::vector<ck::index_t>& dilations,
|
|
const std::vector<ck::index_t>& left_pads,
|
|
const std::vector<ck::index_t>& right_pads);
|
|
|
|
ConvParam(ck::long_index_t n_dim,
|
|
ck::long_index_t group_count,
|
|
ck::long_index_t n_batch,
|
|
ck::long_index_t n_out_channels,
|
|
ck::long_index_t n_in_channels,
|
|
const std::vector<ck::long_index_t>& filters_len,
|
|
const std::vector<ck::long_index_t>& input_len,
|
|
const std::vector<ck::long_index_t>& strides,
|
|
const std::vector<ck::long_index_t>& dilations,
|
|
const std::vector<ck::long_index_t>& left_pads,
|
|
const std::vector<ck::long_index_t>& right_pads);
|
|
|
|
ck::long_index_t num_dim_spatial_;
|
|
ck::long_index_t G_;
|
|
ck::long_index_t N_;
|
|
ck::long_index_t K_;
|
|
ck::long_index_t C_;
|
|
|
|
std::vector<ck::long_index_t> filter_spatial_lengths_;
|
|
std::vector<ck::long_index_t> input_spatial_lengths_;
|
|
std::vector<ck::long_index_t> output_spatial_lengths_;
|
|
|
|
std::vector<ck::long_index_t> conv_filter_strides_;
|
|
std::vector<ck::long_index_t> conv_filter_dilations_;
|
|
|
|
std::vector<ck::long_index_t> input_left_pads_;
|
|
std::vector<ck::long_index_t> input_right_pads_;
|
|
|
|
std::vector<ck::long_index_t> GetOutputSpatialLengths() const;
|
|
|
|
std::size_t GetFlops() const;
|
|
|
|
template <typename InDataType>
|
|
std::size_t GetInputByte() const
|
|
{
|
|
// sizeof(InDataType) * (G * N * C * <input spatial lengths product>) +
|
|
return sizeof(InDataType) *
|
|
(G_ * N_ * C_ *
|
|
ck::accumulate_n<std::size_t>(
|
|
std::begin(input_spatial_lengths_), num_dim_spatial_, 1, std::multiplies<>()));
|
|
}
|
|
|
|
template <typename WeiDataType>
|
|
std::size_t GetWeightByte() const
|
|
{
|
|
// sizeof(WeiDataType) * (G * K * C * <filter spatial lengths product>) +
|
|
return sizeof(WeiDataType) *
|
|
(G_ * K_ * C_ *
|
|
ck::accumulate_n<std::size_t>(
|
|
std::begin(filter_spatial_lengths_), num_dim_spatial_, 1, std::multiplies<>()));
|
|
}
|
|
|
|
template <typename OutDataType>
|
|
std::size_t GetOutputByte() const
|
|
{
|
|
// sizeof(OutDataType) * (G * N * K * <output spatial lengths product>);
|
|
return sizeof(OutDataType) * (G_ * N_ * K_ *
|
|
std::accumulate(std::begin(output_spatial_lengths_),
|
|
std::end(output_spatial_lengths_),
|
|
static_cast<std::size_t>(1),
|
|
std::multiplies<std::size_t>()));
|
|
}
|
|
|
|
template <typename InDataType, typename WeiDataType, typename OutDataType>
|
|
std::size_t GetByte() const
|
|
{
|
|
return GetInputByte<InDataType>() + GetWeightByte<WeiDataType>() +
|
|
GetOutputByte<OutDataType>();
|
|
}
|
|
};
|
|
|
|
std::string get_conv_param_parser_helper_msg();
|
|
|
|
ConvParam parse_conv_param(int num_dim_spatial, int arg_idx, char* const argv[]);
|
|
|
|
} // namespace conv
|
|
} // namespace utils
|
|
} // namespace ck
|
|
|
|
std::ostream& operator<<([[clang::lifetimebound]] std::ostream& os,
|
|
const ck::utils::conv::ConvParam& p);
|