Files
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
Flatten subprojects into a single base monorepo
2026-09-02 16:34:47 -07:00

480 lines
23 KiB
C++

// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT
#include "hipdnn_backend.h"
#include <cstring>
#include <gtest/gtest.h>
class IntegrationVariantPackDescriptorApi : public ::testing::Test
{
protected:
hipdnnBackendDescriptor_t _varpack;
void SetUp() override
{
EXPECT_EQ(hipdnnBackendCreateDescriptor(HIPDNN_BACKEND_VARIANT_PACK_DESCRIPTOR, &_varpack),
HIPDNN_STATUS_SUCCESS);
ASSERT_NE(_varpack, nullptr);
}
void TearDown() override
{
EXPECT_EQ(hipdnnBackendDestroyDescriptor(_varpack), HIPDNN_STATUS_SUCCESS);
}
};
TEST_F(IntegrationVariantPackDescriptorApi, ValidSetAttributesAndFinalize)
{
std::array<void*, 3> devPtrs = {reinterpret_cast<void*>(0x1234),
reinterpret_cast<void*>(0x5678),
reinterpret_cast<void*>(0x9abc)};
std::array<int64_t, 3> uids = {1, 2, 3};
void* workspace = reinterpret_cast<void*>(0xdeadbeef);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_VOID_PTR,
3,
static_cast<const void*>(devPtrs.data())),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(
_varpack, HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS, HIPDNN_TYPE_INT64, 3, uids.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_WORKSPACE,
HIPDNN_TYPE_VOID_PTR,
1,
static_cast<const void*>(&workspace)),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendFinalize(_varpack), HIPDNN_STATUS_SUCCESS);
}
TEST_F(IntegrationVariantPackDescriptorApi, OverrideAttributesSetGetAndFinalize)
{
std::array<void*, 2> devPtrs
= {reinterpret_cast<void*>(0x1234), reinterpret_cast<void*>(0x5678)};
std::array<int64_t, 2> uids = {1, 2};
void* workspace = reinterpret_cast<void*>(0xdeadbeef);
std::array<int64_t, 2> overrideUids = {1, 2};
std::array<int64_t, 2> overrideLengths = {4, 4};
std::array<int64_t, 8> overrideShapes = {1, 3, 2, 2, 1, 3, 2, 2};
std::array<int64_t, 8> overrideStrides = {12, 4, 2, 1, 12, 4, 2, 1};
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_VOID_PTR,
static_cast<int64_t>(devPtrs.size()),
static_cast<const void*>(devPtrs.data())),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(uids.size()),
uids.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_WORKSPACE,
HIPDNN_TYPE_VOID_PTR,
1,
static_cast<const void*>(&workspace)),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_OVERRIDE_UNIQUE_IDS_EXT,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(overrideUids.size()),
overrideUids.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_OVERRIDE_SHAPES_EXT,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(overrideShapes.size()),
overrideShapes.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_OVERRIDE_STRIDES_EXT,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(overrideStrides.size()),
overrideStrides.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_OVERRIDE_LENGTHS_EXT,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(overrideLengths.size()),
overrideLengths.data()),
HIPDNN_STATUS_SUCCESS);
ASSERT_EQ(hipdnnBackendFinalize(_varpack), HIPDNN_STATUS_SUCCESS);
std::array<int64_t, 2> retrievedUids{};
std::array<int64_t, 2> retrievedLengths{};
std::array<int64_t, 8> retrievedShapes{};
std::array<int64_t, 8> retrievedStrides{};
int64_t elementCount = 0;
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_OVERRIDE_UNIQUE_IDS_EXT,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(retrievedUids.size()),
&elementCount,
retrievedUids.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(elementCount, static_cast<int64_t>(overrideUids.size()));
EXPECT_EQ(retrievedUids, overrideUids);
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_OVERRIDE_LENGTHS_EXT,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(retrievedLengths.size()),
&elementCount,
retrievedLengths.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(elementCount, static_cast<int64_t>(overrideLengths.size()));
EXPECT_EQ(retrievedLengths, overrideLengths);
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_OVERRIDE_SHAPES_EXT,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(retrievedShapes.size()),
&elementCount,
retrievedShapes.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(elementCount, static_cast<int64_t>(overrideShapes.size()));
EXPECT_EQ(retrievedShapes, overrideShapes);
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_OVERRIDE_STRIDES_EXT,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(retrievedStrides.size()),
&elementCount,
retrievedStrides.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(elementCount, static_cast<int64_t>(overrideStrides.size()));
EXPECT_EQ(retrievedStrides, overrideStrides);
}
TEST_F(IntegrationVariantPackDescriptorApi, OverrideAttributesRejectZeroLengthOnFinalize)
{
std::array<void*, 1> devPtrs = {reinterpret_cast<void*>(0x1234)};
std::array<int64_t, 1> uids = {1};
void* workspace = reinterpret_cast<void*>(0xdeadbeef);
std::array<int64_t, 1> overrideUids = {1};
std::array<int64_t, 1> overrideLengths = {0};
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_VOID_PTR,
static_cast<int64_t>(devPtrs.size()),
static_cast<const void*>(devPtrs.data())),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(uids.size()),
uids.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_WORKSPACE,
HIPDNN_TYPE_VOID_PTR,
1,
static_cast<const void*>(&workspace)),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_OVERRIDE_UNIQUE_IDS_EXT,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(overrideUids.size()),
overrideUids.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_OVERRIDE_LENGTHS_EXT,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(overrideLengths.size()),
overrideLengths.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendFinalize(_varpack), HIPDNN_STATUS_BAD_PARAM);
}
TEST_F(IntegrationVariantPackDescriptorApi, ValidSetAttributesAndGetAttributesBeforeFinalize)
{
std::array<void*, 3> devPtrs = {reinterpret_cast<void*>(0x1234),
reinterpret_cast<void*>(0x5678),
reinterpret_cast<void*>(0x9abc)};
std::array<int64_t, 3> uids = {1, 2, 3};
void* workspace = reinterpret_cast<void*>(0xdeadbeef);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_VOID_PTR,
3,
static_cast<const void*>(devPtrs.data())),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(
_varpack, HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS, HIPDNN_TYPE_INT64, 3, uids.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_WORKSPACE,
HIPDNN_TYPE_VOID_PTR,
1,
static_cast<const void*>(&workspace)),
HIPDNN_STATUS_SUCCESS);
std::array<void*, 3> retrievedDevPtrs;
int64_t elementCount = 0;
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_INT64,
3,
&elementCount,
static_cast<void*>(retrievedDevPtrs.data())),
HIPDNN_STATUS_NOT_INITIALIZED);
}
TEST_F(IntegrationVariantPackDescriptorApi, InvalidSetAttributes)
{
std::array<void*, 3> devPtrs = {reinterpret_cast<void*>(0x1234),
reinterpret_cast<void*>(0x5678),
reinterpret_cast<void*>(0x9abc)};
std::array<int64_t, 3> uids = {1, 2, 3};
void* workspace = reinterpret_cast<void*>(0xdeadbeef);
// HIPDNN_STATUS_BAD_PARAM since HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS expects HIPDNN_TYPE_VOID_PTR
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_INT64,
3,
static_cast<const void*>(devPtrs.data())),
HIPDNN_STATUS_BAD_PARAM);
// HIPDNN_STATUS_BAD_PARAM since HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS expects HIPDNN_TYPE_INT64
EXPECT_EQ(
hipdnnBackendSetAttribute(
_varpack, HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS, HIPDNN_TYPE_VOID_PTR, 3, uids.data()),
HIPDNN_STATUS_BAD_PARAM);
// HIPDNN_STATUS_BAD_PARAM since HIPDNN_ATTR_VARIANT_PACK_WORKSPACE expects HIPDNN_TYPE_VOID_PTR
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_WORKSPACE,
HIPDNN_TYPE_VOID_PTR,
2,
static_cast<const void*>(&workspace)),
HIPDNN_STATUS_BAD_PARAM);
// HIPDNN_STATUS_BAD_PARAM since null ptr passed
EXPECT_EQ(
hipdnnBackendSetAttribute(
_varpack, HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS, HIPDNN_TYPE_VOID_PTR, 3, nullptr),
HIPDNN_STATUS_BAD_PARAM_NULL_POINTER);
}
TEST_F(IntegrationVariantPackDescriptorApi, InvalidGetAttributes)
{
std::array<void*, 3> retrievedDevPtrs;
int64_t elementCount = 0;
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_INT64,
3,
&elementCount,
static_cast<void*>(retrievedDevPtrs.data())),
HIPDNN_STATUS_NOT_INITIALIZED);
}
// Invalid finalize since no devPtrs, or ids were set
TEST_F(IntegrationVariantPackDescriptorApi, InvalidFinalizeEmpty)
{
EXPECT_EQ(hipdnnBackendFinalize(_varpack), HIPDNN_STATUS_BAD_PARAM);
}
// Invalid finalize since devPtrs, and ids do not have the same size.
TEST_F(IntegrationVariantPackDescriptorApi, InvalidFinalizeParams)
{
std::array<void*, 3> devPtrs = {reinterpret_cast<void*>(0x1234),
reinterpret_cast<void*>(0x5678),
reinterpret_cast<void*>(0x9abc)};
std::array<int64_t, 2> uids = {1, 2};
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_VOID_PTR,
devPtrs.size(),
static_cast<const void*>(devPtrs.data())),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS,
HIPDNN_TYPE_INT64,
uids.size(),
uids.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendFinalize(_varpack), HIPDNN_STATUS_BAD_PARAM);
}
TEST_F(IntegrationVariantPackDescriptorApi, InvalidNullDescriptor)
{
hipdnnBackendDescriptor_t nullDescriptor = nullptr;
int64_t dummyData = 42;
auto status = hipdnnBackendSetAttribute(
nullDescriptor, HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS, HIPDNN_TYPE_INT64, 1, &dummyData);
EXPECT_EQ(status, HIPDNN_STATUS_BAD_PARAM_NULL_POINTER);
int64_t retrievedData = 0;
int64_t elementCount = 0;
status = hipdnnBackendGetAttribute(nullDescriptor,
HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS,
HIPDNN_TYPE_INT64,
1,
&elementCount,
&retrievedData);
EXPECT_EQ(status, HIPDNN_STATUS_BAD_PARAM_NULL_POINTER);
}
class IntegrationVariantPackDescriptorApiFinalized : public ::testing::Test
{
protected:
hipdnnBackendDescriptor_t _varpack;
std::array<void*, 3> _devPtrs = {reinterpret_cast<void*>(0x1234),
reinterpret_cast<void*>(0x5678),
reinterpret_cast<void*>(0x9abc)};
std::array<int64_t, 3> _uids = {1, 2, 3};
void* _workspace = reinterpret_cast<void*>(0xdeadbeef);
void SetUp() override
{
EXPECT_EQ(hipdnnBackendCreateDescriptor(HIPDNN_BACKEND_VARIANT_PACK_DESCRIPTOR, &_varpack),
HIPDNN_STATUS_SUCCESS);
ASSERT_NE(_varpack, nullptr);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_VOID_PTR,
static_cast<int64_t>(_devPtrs.size()),
static_cast<const void*>(_devPtrs.data())),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS,
HIPDNN_TYPE_INT64,
static_cast<int64_t>(_uids.size()),
_uids.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_WORKSPACE,
HIPDNN_TYPE_VOID_PTR,
1,
static_cast<const void*>(&_workspace)),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(hipdnnBackendFinalize(_varpack), HIPDNN_STATUS_SUCCESS);
}
void TearDown() override
{
EXPECT_EQ(hipdnnBackendDestroyDescriptor(_varpack), HIPDNN_STATUS_SUCCESS);
}
};
TEST_F(IntegrationVariantPackDescriptorApiFinalized, ValidGetAttributes)
{
std::array<void*, 3> retrievedDevPtrs;
std::array<int64_t, 3> retrievedUids;
void* retrievedWorkspace = nullptr;
int64_t elementCount = 0;
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_VOID_PTR,
3,
&elementCount,
static_cast<void*>(retrievedDevPtrs.data())),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(elementCount, 3);
EXPECT_EQ(std::memcmp(static_cast<const void*>(retrievedDevPtrs.data()),
static_cast<const void*>(_devPtrs.data()),
_devPtrs.size() * sizeof(void*)),
0);
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS,
HIPDNN_TYPE_INT64,
3,
&elementCount,
retrievedUids.data()),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(elementCount, 3);
EXPECT_EQ(std::memcmp(retrievedUids.data(), _uids.data(), _uids.size() * sizeof(int64_t)), 0);
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_WORKSPACE,
HIPDNN_TYPE_VOID_PTR,
1,
&elementCount,
static_cast<void*>(&retrievedWorkspace)),
HIPDNN_STATUS_SUCCESS);
EXPECT_EQ(elementCount, 1);
EXPECT_EQ(retrievedWorkspace, _workspace);
}
TEST_F(IntegrationVariantPackDescriptorApiFinalized, InvalidGetAttributes)
{
std::array<void*, 3> retrievedDevPtrs;
std::array<int64_t, 3> retrievedUids;
void* retrievedWorkspace = nullptr;
int64_t elementCount = 0;
// HIPDNN_STATUS_BAD_PARAM since HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS expects HIPDNN_TYPE_VOID_PTR
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_INT64,
3,
&elementCount,
static_cast<void*>(retrievedDevPtrs.data())),
HIPDNN_STATUS_BAD_PARAM);
// HIPDNN_STATUS_BAD_PARAM since HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS expects HIPDNN_TYPE_INT64
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_UNIQUE_IDS,
HIPDNN_TYPE_VOID_PTR,
3,
&elementCount,
retrievedUids.data()),
HIPDNN_STATUS_BAD_PARAM);
// HIPDNN_STATUS_BAD_PARAM since HIPDNN_ATTR_VARIANT_PACK_WORKSPACE expects HIPDNN_TYPE_VOID_PTR
// but elementCount should be 1
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_WORKSPACE,
HIPDNN_TYPE_VOID_PTR,
2,
&elementCount,
static_cast<void*>(&retrievedWorkspace)),
HIPDNN_STATUS_BAD_PARAM);
// HIPDNN_STATUS_BAD_PARAM since arrayOfElements cannot be nullptr
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_VOID_PTR,
3,
&elementCount,
nullptr),
HIPDNN_STATUS_BAD_PARAM_NULL_POINTER);
// HIPDNN_STATUS_BAD_PARAM since elementCount cannot be nullptr
EXPECT_EQ(hipdnnBackendGetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_VOID_PTR,
3,
nullptr,
static_cast<void*>(retrievedDevPtrs.data())),
HIPDNN_STATUS_BAD_PARAM_NULL_POINTER);
}
TEST_F(IntegrationVariantPackDescriptorApiFinalized, InvalidSetAttributes)
{
EXPECT_EQ(hipdnnBackendSetAttribute(_varpack,
HIPDNN_ATTR_VARIANT_PACK_DATA_POINTERS,
HIPDNN_TYPE_VOID_PTR,
static_cast<int64_t>(_devPtrs.size()),
static_cast<const void*>(_devPtrs.data())),
HIPDNN_STATUS_NOT_INITIALIZED);
}