From cfef0c6f66c9d36ae2899c8c2790c3fe422a7700 Mon Sep 17 00:00:00 2001 From: Yong He Date: Thu, 13 Jun 2024 16:29:58 -0700 Subject: Metal: misc fixes and enable more tests. (#4374) * Fix and enable tests for metal. * Fix. * Fix. * Fix tests. * Fix warnings. * Fix. --------- Co-authored-by: Yong He --- tests/autodiff/custom-intrinsic.slang | 4 +++- tests/autodiff/reverse-matrix-ops.slang | 1 - 2 files changed, 3 insertions(+), 2 deletions(-) (limited to 'tests/autodiff') diff --git a/tests/autodiff/custom-intrinsic.slang b/tests/autodiff/custom-intrinsic.slang index 7d82887e2..2ebd01099 100644 --- a/tests/autodiff/custom-intrinsic.slang +++ b/tests/autodiff/custom-intrinsic.slang @@ -1,6 +1,5 @@ //TEST(compute, vulkan):COMPARE_COMPUTE_EX:-vk -compute -shaderobj -output-using-type //TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj -output-using-type -//DISABLE_TEST(compute):COMPARE_COMPUTE:-slang -shaderobj -mtl //TEST_INPUT:ubuffer(data=[0 0 0 0 0], stride=4):out,name=outputBuffer RWStructuredBuffer outputBuffer; @@ -17,6 +16,7 @@ namespace myintrinsiclib __target_intrinsic(cuda, "$P_exp($0)") __target_intrinsic(cpp, "$P_exp($0)") __target_intrinsic(spirv, "12 resultType resultId glsl450 27 _0") + __target_intrinsic(metal, "exp($0)") [ForwardDerivative(d_myexp)] T myexp(T x); @@ -33,6 +33,7 @@ namespace myintrinsiclib __generic __target_intrinsic(hlsl, "sin($0)") __target_intrinsic(glsl, "sin($0)") + __target_intrinsic(metal, "sin($0)") __target_intrinsic(cuda, "$P_sin($0)") __target_intrinsic(cpp, "$P_sin($0)") __target_intrinsic(spirv, "12 resultType resultId glsl450 13 _0") @@ -51,6 +52,7 @@ namespace myintrinsiclib __generic __target_intrinsic(hlsl, "cos($0)") __target_intrinsic(glsl, "cos($0)") + __target_intrinsic(metal, "cos($0)") __target_intrinsic(cuda, "$P_cos($0)") __target_intrinsic(cpp, "$P_cos($0)") __target_intrinsic(spirv, "12 resultType resultId glsl450 14 _0") diff --git a/tests/autodiff/reverse-matrix-ops.slang b/tests/autodiff/reverse-matrix-ops.slang index 4f4a6f675..e7be41811 100644 --- a/tests/autodiff/reverse-matrix-ops.slang +++ b/tests/autodiff/reverse-matrix-ops.slang @@ -1,6 +1,5 @@ //TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj -output-using-type //TEST(compute, vulkan):COMPARE_COMPUTE_EX:-vk -compute -shaderobj -output-using-type -//DISABLE_TEST(compute):COMPARE_COMPUTE:-slang -shaderobj -mtl //TEST_INPUT:ubuffer(data=[0 0 0 0 0 0 0 0 0 0 0], stride=4):out,name=outputBuffer RWStructuredBuffer outputBuffer; -- cgit v1.2.3