TheRock/rocm-libraries/projects/hipdnn/tools/DescriptorGenerator/templates/constants.hpp.j2

{% from "macros.j2" import vector_const_name %}
// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT

#pragma once

#include <array>
#include <cstdint>

namespace hipdnn_tests::constants
{

// Standard {{ op.name }} constants for testing get/set of valid operations.
// These represent "any valid {{ op.name | lower }}" — specific values are not significant.

{% for tf in op.tensor_fields %}
{% set uid = op.test_data.tensor_uids.get(tf.name, loop.index) %}
{% set tc = op.test_data.tensor_configs.get(tf.name) %}
constexpr int64_t {{ op.tensor_const_prefix }}TENSOR_{{ tf.name | upper }}_UID = {{ uid }};
{% if tc %}
constexpr std::array<int64_t, {{ tc.dims | length }}> {{ op.tensor_const_prefix }}TENSOR_{{ tf.name | upper }}_DIMS = {{ '{' }}{{ tc.dims | join(', ') }}{{ '}' }};
constexpr std::array<int64_t, {{ tc.strides | length }}> {{ op.tensor_const_prefix }}TENSOR_{{ tf.name | upper }}_STRIDES = {{ '{' }}{{ tc.strides | join(', ') }}{{ '}' }};
{% endif %}

{% endfor %}
{% for df in op.data_fields if df.is_vector and df.required %}
{% if df.name in op.test_data.field_values %}
{% set fv = op.test_data.field_values[df.name] %}
{% set const_name = vector_const_name(df, op) %}
constexpr std::array<int64_t, {{ fv | length }}> {{ const_name }} = {{ '{' }}{{ fv | join(', ') }}{{ '}' }};
{% endif %}
{% endfor %}
} // namespace hipdnn_tests::constants