diff options
| -rw-r--r-- | source/slang/slang-ir-peephole.cpp | 3 | ||||
| -rw-r--r-- | tests/language-feature/types/is-on-type.slang | 45 |
2 files changed, 47 insertions, 1 deletions
diff --git a/source/slang/slang-ir-peephole.cpp b/source/slang/slang-ir-peephole.cpp index d4369da7a..88b26fbd3 100644 --- a/source/slang/slang-ir-peephole.cpp +++ b/source/slang/slang-ir-peephole.cpp @@ -960,7 +960,8 @@ struct PeepholeContext : InstPassBase { auto getTypeFromOperand = [](IRInst* operand) -> IRType* { - if (as<IRTypeType>(operand->getFullType()) || !operand->getFullType()) + if (as<IRTypeType>(operand->getFullType()) || !operand->getFullType() || + as<IRTypeKind>(operand->getFullType())) return (IRType*)operand; return operand->getFullType(); }; diff --git a/tests/language-feature/types/is-on-type.slang b/tests/language-feature/types/is-on-type.slang new file mode 100644 index 000000000..e9dd48fcf --- /dev/null +++ b/tests/language-feature/types/is-on-type.slang @@ -0,0 +1,45 @@ + +//TEST(compute):COMPARE_COMPUTE_EX(filecheck-buffer=CHECK):-slang -compute + +// Test that `is` operator works on generic type param. + +//TEST_INPUT:ubuffer(data=[0 0 0 0], stride=4):out,name=outputBuffer +RWStructuredBuffer<int> outputBuffer; + +interface I +{ + +} + +struct A : I +{ + uint i; +}; + +struct B : I +{ + float2 f2; +}; + +func test<T : I>(T t) -> int +{ + int rs = 0; + if (T is A) + { + rs++; + } + if (T is B) + { + rs+=2; + } + return rs; +} + +[shader("compute")] +[numthreads(1,1, 1)] +void computeMain() +{ + B b; + // CHECK: 2 + outputBuffer[0] = test(b); +}
\ No newline at end of file |