From 4c6b0a2831a7edd1419bd0b2e6edd089080e07be Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Wed, 31 Jul 2024 17:35:08 -0700
Subject: Allow generic type deduction from ParameterBlock arguments. (#4766)

* Allow generic type deduction from ParameterBlock arguments.

* Fix test.

* Update expected failure list.

---------

Co-authored-by: Yong He <yhe@nvidia.com>
---
 .../generics/parameter-block-unify.slang           | 33 ++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 tests/language-feature/generics/parameter-block-unify.slang

(limited to 'tests/language-feature')

diff --git a/tests/language-feature/generics/parameter-block-unify.slang b/tests/language-feature/generics/parameter-block-unify.slang
new file mode 100644
index 000000000..b549f555b
--- /dev/null
+++ b/tests/language-feature/generics/parameter-block-unify.slang
@@ -0,0 +1,33 @@
+//TEST(compute):COMPARE_COMPUTE_EX(filecheck-buffer=CHECK):-slang -compute -shaderobj -output-using-type
+//TEST(compute, vulkan):COMPARE_COMPUTE_EX(filecheck-buffer=CHECK):-vk -compute -shaderobj -output-using-type
+//TEST(compute, vulkan):COMPARE_COMPUTE_EX(filecheck-buffer=CHECK):-mtl -compute -shaderobj -output-using-type -render-features argument-buffer-tier-2
+
+struct TestStruct<Format:__BuiltinIntegerType, let count : int>
+{
+    Format f;
+};
+
+Format testFunction<Format : __BuiltinIntegerType, let count : int>(TestStruct<Format, count> data)
+{
+    return data.f + __int_cast<Format>(count);
+}
+
+//TEST_INPUT: set testBlock = new TestStruct<int, 12>{1}
+ParameterBlock<TestStruct<int, 12>> testBlock;
+
+//TEST_INPUT: set testBlock2 = new TestStruct<int, 12>{2}
+ConstantBuffer<TestStruct<int, 2>> testBlock2;
+
+//TEST_INPUT: set outputBuffer = out ubuffer(data=[0 0 0 0], stride=4)
+RWStructuredBuffer<int> outputBuffer;
+
+[numthreads(1, 1, 1)]
+void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
+{
+    // CHECK: 13
+    outputBuffer[0] = testFunction(testBlock);
+    // CHECK: 13
+    outputBuffer[1] = testFunction<int, 12>(testBlock);
+    // CHECK: 4
+    outputBuffer[2] = testFunction(testBlock2);
+}
\ No newline at end of file
-- 
cgit v1.2.3