From 41e7e565eb3dfa13562cbfa3e8641874c2c6d66c Mon Sep 17 00:00:00 2001 From: Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com> Date: Fri, 4 Apr 2025 14:12:27 -0400 Subject: Add a loop analysis step to infer the exit values of loop phi parameters. (#6696) * Initial loop analysis pass * More changes for a single-pass implication propagation * Update slang-ir-autodiff-loop-analysis.cpp * Cleanup + new system for loop analysis * Fixup bugs in loop analysis * Remove some relation types to simplify the analysis. Add test * Remove unused * Address comments * Fix issue with continue loops * Update reverse-loop-exit-value-inference-1.slang * Update reverse-continue-loop.slang --- tests/autodiff/reverse-loop.slang | 46 --------------------------------------- 1 file changed, 46 deletions(-) delete mode 100644 tests/autodiff/reverse-loop.slang (limited to 'tests/autodiff/reverse-loop.slang') diff --git a/tests/autodiff/reverse-loop.slang b/tests/autodiff/reverse-loop.slang deleted file mode 100644 index 18b672860..000000000 --- a/tests/autodiff/reverse-loop.slang +++ /dev/null @@ -1,46 +0,0 @@ -//TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj -output-using-type -//TEST(compute, vulkan):COMPARE_COMPUTE_EX:-vk -compute -shaderobj -output-using-type -//TEST(compute):COMPARE_COMPUTE_EX:-cpu -compute -output-using-type -shaderobj -//DISABLE_TEST:SIMPLE(filecheck=CHK):-target glsl -stage compute -entry computeMain -report-checkpoint-intermediates - -//TEST_INPUT:ubuffer(data=[0 0 0 0 0], stride=4):out,name=outputBuffer -RWStructuredBuffer outputBuffer; - -typedef DifferentialPair dpfloat; -typedef float.Differential dfloat; - -//CHK: note: checkpointing context of 24 bytes associated with function: 'test_simple_loop' -[Differentiable] -float test_simple_loop(float y) -{ - //CHK: note: 20 bytes (FixedArray ) used to checkpoint the following item: - float t = y; - - //CHK: note: 4 bytes (int32_t) used for a loop counter here: - for (int i = 0; i < 3; i++) - { - t = t * t; - } - - return t; -} - -[numthreads(1, 1, 1)] -void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID) -{ - { - dpfloat dpa = dpfloat(1.0, 0.0); - - __bwd_diff(test_simple_loop)(dpa, 1.0f); - outputBuffer[0] = dpa.d; // Expect: 8.0 - } - - { - dpfloat dpa = dpfloat(0.4, 0.0); - - __bwd_diff(test_simple_loop)(dpa, 1.0f); - outputBuffer[1] = dpa.d; // Expect: 0.0131072 - } -} - -//CHK-NOT: note \ No newline at end of file -- cgit v1.2.3