diff options
| author | Yong He <yonghe@outlook.com> | 2024-06-13 16:29:58 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-13 16:29:58 -0700 |
| commit | cfef0c6f66c9d36ae2899c8c2790c3fe422a7700 (patch) | |
| tree | 0e116769009ae7e4b3b16d2ef28f275bea5e1fd7 /tests/autodiff | |
| parent | 2407966e899f9e4f490b23a92fc06d5da20544cc (diff) | |
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 <yonghe@Yongs-Mac-mini.local>
Diffstat (limited to 'tests/autodiff')
| -rw-r--r-- | tests/autodiff/custom-intrinsic.slang | 4 | ||||
| -rw-r--r-- | tests/autodiff/reverse-matrix-ops.slang | 1 |
2 files changed, 3 insertions, 2 deletions
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<float> 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>)] T myexp(T x); @@ -33,6 +33,7 @@ namespace myintrinsiclib __generic<T : IDFloat> __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<T : IDFloat> __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<float> outputBuffer; |