derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
81 lines
2.9 KiB
C++
81 lines
2.9 KiB
C++
// Copyright © Advanced Micro Devices, Inc., or its affiliates.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <hipdnn_frontend/Graph.hpp>
|
|
#include <hipdnn_frontend/attributes/ConvolutionFpropAttributes.hpp>
|
|
#include <hipdnn_frontend/attributes/PointwiseAttributes.hpp>
|
|
#include <string>
|
|
|
|
#include "harness/GraphDescription.hpp"
|
|
|
|
using namespace hipdnn_frontend;
|
|
using namespace hipdnn_frontend::graph;
|
|
|
|
// NOLINTBEGIN(readability-identifier-naming) -- gtest macro-generated names
|
|
|
|
TEST(TestGraphDescription, SingleConvFprop)
|
|
{
|
|
Graph graph;
|
|
graph.set_io_data_type(DataType::FLOAT).set_compute_data_type(DataType::FLOAT);
|
|
|
|
auto x = std::make_shared<TensorAttributes>();
|
|
x->set_dim({1, 16, 16, 16}).set_stride({4096, 256, 16, 1}).set_data_type(DataType::FLOAT);
|
|
|
|
auto w = std::make_shared<TensorAttributes>();
|
|
w->set_dim({1, 16, 3, 3}).set_stride({144, 9, 3, 1}).set_data_type(DataType::FLOAT);
|
|
|
|
graph.conv_fprop(x, w, ConvFpropAttributes());
|
|
|
|
auto desc = hipdnn_integration_tests::describeGraph(graph);
|
|
EXPECT_NE(desc.find("ConvFprop"), std::string::npos) << "desc=" << desc;
|
|
EXPECT_NE(desc.find("io="), std::string::npos) << "desc=" << desc;
|
|
EXPECT_NE(desc.find("compute="), std::string::npos) << "desc=" << desc;
|
|
}
|
|
|
|
TEST(TestGraphDescription, ConvPlusPointwiseRelu)
|
|
{
|
|
Graph graph;
|
|
graph.set_io_data_type(DataType::HALF).set_compute_data_type(DataType::FLOAT);
|
|
|
|
auto x = std::make_shared<TensorAttributes>();
|
|
x->set_dim({1, 16, 16, 16}).set_stride({4096, 256, 16, 1}).set_data_type(DataType::HALF);
|
|
|
|
auto w = std::make_shared<TensorAttributes>();
|
|
w->set_dim({1, 16, 3, 3}).set_stride({144, 9, 3, 1}).set_data_type(DataType::HALF);
|
|
|
|
auto y = graph.conv_fprop(x, w, ConvFpropAttributes());
|
|
|
|
PointwiseAttributes pw;
|
|
pw.set_mode(PointwiseMode::RELU_FWD);
|
|
graph.pointwise(y, pw);
|
|
|
|
auto desc = hipdnn_integration_tests::describeGraph(graph);
|
|
EXPECT_NE(desc.find("ConvFprop"), std::string::npos) << "desc=" << desc;
|
|
EXPECT_NE(desc.find("Pointwise:RELU_FWD"), std::string::npos) << "desc=" << desc;
|
|
EXPECT_NE(desc.find('+'), std::string::npos) << "desc=" << desc;
|
|
}
|
|
|
|
TEST(TestGraphDescription, DataTypeAnnotations)
|
|
{
|
|
Graph graph;
|
|
graph.set_io_data_type(DataType::HALF)
|
|
.set_compute_data_type(DataType::FLOAT)
|
|
.set_intermediate_data_type(DataType::HALF);
|
|
|
|
auto x = std::make_shared<TensorAttributes>();
|
|
x->set_dim({1, 16, 16, 16}).set_stride({4096, 256, 16, 1}).set_data_type(DataType::HALF);
|
|
|
|
PointwiseAttributes pw;
|
|
pw.set_mode(PointwiseMode::RELU_FWD);
|
|
graph.pointwise(x, pw);
|
|
|
|
auto desc = hipdnn_integration_tests::describeGraph(graph);
|
|
EXPECT_NE(desc.find("io=fp16"), std::string::npos) << "desc=" << desc;
|
|
EXPECT_NE(desc.find("compute=fp32"), std::string::npos) << "desc=" << desc;
|
|
EXPECT_NE(desc.find("intermediate=fp16"), std::string::npos) << "desc=" << desc;
|
|
}
|
|
|
|
// NOLINTEND(readability-identifier-naming)
|