diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 0c1689bde..f9139ac5b 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -24,377 +24,377 @@ jobs: matrix: include: - - name: osx-clang-runtime9 - os: macos-11 - compiler: clang - clang-runtime: '9' - - - name: osx-clang-runtime10 - os: macos-11 - compiler: clang - clang-runtime: '10' - - - name: osx-clang-runtime11 - os: macos-latest - compiler: clang - clang-runtime: '11' - - - name: osx-clang-runtime12 - os: macos-latest - compiler: clang - clang-runtime: '12' - - - name: osx-clang-runtime13 - os: macos-latest - compiler: clang - clang-runtime: '13' - - - name: osx-clang-runtime14 - os: macos-latest - compiler: clang - clang-runtime: '14' - - - name: osx-clang-runtime15 - os: macos-latest - compiler: clang - clang-runtime: '15' + # - name: osx-clang-runtime9 + # os: macos-11 + # compiler: clang + # clang-runtime: '9' + + # - name: osx-clang-runtime10 + # os: macos-11 + # compiler: clang + # clang-runtime: '10' + + # - name: osx-clang-runtime11 + # os: macos-latest + # compiler: clang + # clang-runtime: '11' + + # - name: osx-clang-runtime12 + # os: macos-latest + # compiler: clang + # clang-runtime: '12' + + # - name: osx-clang-runtime13 + # os: macos-latest + # compiler: clang + # clang-runtime: '13' + + # - name: osx-clang-runtime14 + # os: macos-latest + # compiler: clang + # clang-runtime: '14' + + # - name: osx-clang-runtime15 + # os: macos-latest + # compiler: clang + # clang-runtime: '15' - name: osx-clang-runtime16 os: macos-latest compiler: clang clang-runtime: '16' - - name: win-msvc-runtime14 - os: windows-latest - compiler: msvc - clang-runtime: '14' - - - name: win-msvc-runtime15 - os: windows-latest - compiler: msvc - clang-runtime: '15' - - - name: win-msvc-runtime16 - os: windows-latest - compiler: msvc - clang-runtime: '16' - - - name: ubu20-gcc7-runtime8 - os: ubuntu-20.04 - compiler: gcc-7 - clang-runtime: '8' - - - name: ubu20-gcc7-runtime11-analyzers - os: ubuntu-20.04 - compiler: gcc-7 - clang-runtime: '11' - coverage: true - cuda: true - extra_cmake_options: '-DLLVM_ENABLE_WERROR=On -DENABLE_ENZYME_BACKEND=On' - #clang-format: true - - - name: ubu20-gcc7-runtime11-benchmarks - os: ubuntu-20.04 - compiler: gcc-7 - clang-runtime: '11' - extra_cmake_options: '-DCLAD_ENABLE_BENCHMARKS=On -DENABLE_ENZYME_BACKEND=On' - benchmark: true - - - name: ubu20-gcc8-runtime11-coverity - os: ubuntu-20.04 - compiler: gcc-8 - clang-runtime: '11' - coverity: true - - - name: ubu20-gcc9-runtime9 - os: ubuntu-20.04 - compiler: gcc-9 - clang-runtime: '9' - - - name: ubu20-clang8-runtime9 - os: ubuntu-20.04 - compiler: clang-8 - clang-runtime: '9' - - - name: ubu20-clang9-runtime9 - os: ubuntu-20.04 - compiler: clang-9 - clang-runtime: '9' - - - name: ubu20-gcc7-runtime10 - os: ubuntu-20.04 - compiler: gcc-7 - clang-runtime: '10' - - - name: ubu20-gcc8-runtime10 - os: ubuntu-20.04 - compiler: gcc-8 - clang-runtime: '10' - - - name: ubu20-gcc9-runtime10 - os: ubuntu-20.04 - compiler: gcc-9 - clang-runtime: '10' - - - name: ubu20-clang8-runtime10 - os: ubuntu-20.04 - compiler: clang-8 - clang-runtime: '10' - - - name: ubu20-clang9-runtime10 - os: ubuntu-20.04 - compiler: clang-9 - clang-runtime: '10' - - - name: ubu20-gcc7-runtime11 - os: ubuntu-20.04 - compiler: gcc-7 - clang-runtime: '11' - - - name: ubu20-gcc8-runtime11 - os: ubuntu-20.04 - compiler: gcc-8 - clang-runtime: '11' - - - name: ubu20-gcc9-runtime11 - os: ubuntu-20.04 - compiler: gcc-9 - clang-runtime: '11' - - - name: ubu20-clang8-runtime11 - os: ubuntu-20.04 - compiler: clang-8 - clang-runtime: '11' - - - name: ubu20-clang9-runtime11 - os: ubuntu-20.04 - compiler: clang-9 - clang-runtime: '11' - - - name: ubu20-gcc7-runtime12 - os: ubuntu-20.04 - compiler: gcc-7 - clang-runtime: '12' - - - name: ubu20-gcc8-runtime12 - os: ubuntu-20.04 - compiler: gcc-8 - clang-runtime: '12' - - - name: ubu20-gcc9-runtime12 - os: ubuntu-20.04 - compiler: gcc-9 - clang-runtime: '12' - - name: ubu20-clang8-runtime12 - os: ubuntu-20.04 - compiler: clang-8 - clang-runtime: '12' - - - name: ubu20-clang9-runtime12 - os: ubuntu-20.04 - compiler: clang-9 - clang-runtime: '12' - - - name: ubu20-clang9-runtime10-cuda - os: ubuntu-20.04 - compiler: clang-9 - clang-runtime: '10' - cuda: true - - - name: ubu20-clang9-runtime15 - os: ubuntu-20.04 - compiler: clang-9 - clang-runtime: '15' - - - name: ubu20-clang9-runtime16 - os: ubuntu-20.04 - compiler: clang-9 - clang-runtime: '16' - - - name: ubu22-gcc9-runtime11 - os: ubuntu-22.04 - compiler: gcc-9 - clang-runtime: '11' - - - name: ubu22-gcc10-runtime11 - os: ubuntu-22.04 - compiler: gcc-10 - clang-runtime: '11' - - - name: ubu22-gcc11-runtime11 - os: ubuntu-22.04 - compiler: gcc-11 - clang-runtime: '11' - - - name: ubu22-clang11-runtime11 - os: ubuntu-22.04 - compiler: 'clang-11' - clang-runtime: '11' - - - name: ubu22-clang12-runtime11 - os: ubuntu-22.04 - compiler: clang-12 - clang-runtime: '11' - - - name: ubu22-clang13-runtime11 - os: ubuntu-22.04 - compiler: clang-13 - clang-runtime: '11' - - - name: ubu22-clang14-runtime11 - os: ubuntu-22.04 - compiler: clang-14 - clang-runtime: '11' - - - name: ubu22-clang15-runtime11 - os: ubuntu-22.04 - compiler: clang-15 - clang-runtime: '11' - - - name: ubu22-gcc9-runtime12 - os: ubuntu-22.04 - compiler: gcc-9 - clang-runtime: '12' - - - name: ubu22-gcc10-runtime12 - os: ubuntu-22.04 - compiler: gcc-10 - clang-runtime: '12' - - - name: ubu22-gcc11-runtime12 - os: ubuntu-22.04 - compiler: gcc-11 - clang-runtime: '12' - - - name: ubu22-clang11-runtime12 - os: ubuntu-22.04 - compiler: 'clang-11' - clang-runtime: '12' - - - name: ubu22-clang12-runtime12 - os: ubuntu-22.04 - compiler: clang-12 - clang-runtime: '12' - - - name: ubu22-clang13-runtime12 - os: ubuntu-22.04 - compiler: clang-13 - clang-runtime: '12' - - - name: ubu22-clang14-runtime12 - os: ubuntu-22.04 - compiler: clang-14 - clang-runtime: '12' - - - name: ubu22-clang15-runtime12 - os: ubuntu-22.04 - compiler: clang-15 - clang-runtime: '12' - - - name: ubu22-gcc10-runtime13 - os: ubuntu-22.04 - compiler: gcc-10 - clang-runtime: '13' - - - name: ubu22-gcc11-runtime13 - os: ubuntu-22.04 - compiler: gcc-11 - clang-runtime: '13' - - - name: ubu22-gcc12-runtime13 - os: ubuntu-22.04 - compiler: gcc-12 - clang-runtime: '13' - - - name: ubu22-clang11-runtime13 - os: ubuntu-22.04 - compiler: clang-11 - clang-runtime: '13' - - - name: ubu22-clang12-runtime13 - os: ubuntu-22.04 - compiler: clang-12 - clang-runtime: '13' - - - name: ubu22-clang13-runtime13 - os: ubuntu-22.04 - compiler: clang-13 - clang-runtime: '13' - - - name: ubu22-clang14-runtime13 - os: ubuntu-22.04 - compiler: clang-14 - clang-runtime: '13' - - - name: ubu22-clang15-runtime13 - os: ubuntu-22.04 - compiler: clang-15 - clang-runtime: '13' - - - name: ubu22-gcc10-runtime14 - os: ubuntu-22.04 - compiler: gcc-10 - clang-runtime: '14' - - - name: ubu22-gcc11-runtime14 - os: ubuntu-22.04 - compiler: gcc-11 - clang-runtime: '14' - - - name: ubu22-gcc12-runtime14 - os: ubuntu-22.04 - compiler: gcc-12 - clang-runtime: '14' - - - name: ubu22-gcc12-runtime15 - os: ubuntu-22.04 - compiler: gcc-12 - clang-runtime: '15' - - - name: ubu22-gcc12-runtime16 - os: ubuntu-22.04 - compiler: gcc-12 - clang-runtime: '16' - - - name: ubu22-clang11-runtime14 - os: ubuntu-22.04 - compiler: clang-11 - clang-runtime: '14' - - - name: ubu22-clang12-runtime14 - os: ubuntu-22.04 - compiler: clang-12 - clang-runtime: '14' - - - name: ubu22-clang13-runtime14 - os: ubuntu-22.04 - compiler: clang-13 - clang-runtime: '14' - - - name: ubu22-clang14-runtime14 - os: ubuntu-22.04 - compiler: clang-14 - clang-runtime: '14' - - - name: ubu22-clang14-runtime15 - os: ubuntu-22.04 - compiler: clang-14 - clang-runtime: '15' - - - name: ubu22-clang15-runtime14 - os: ubuntu-22.04 - compiler: clang-15 - clang-runtime: '14' - - - name: ubu22-clang15-runtime15 - os: ubuntu-22.04 - compiler: clang-15 - clang-runtime: '15' - - - name: ubu22-clang15-runtime16 - os: ubuntu-22.04 - compiler: clang-15 - clang-runtime: '16' + # - name: win-msvc-runtime14 + # os: windows-latest + # compiler: msvc + # clang-runtime: '14' + + # - name: win-msvc-runtime15 + # os: windows-latest + # compiler: msvc + # clang-runtime: '15' + + # - name: win-msvc-runtime16 + # os: windows-latest + # compiler: msvc + # clang-runtime: '16' + + # - name: ubu20-gcc7-runtime8 + # os: ubuntu-20.04 + # compiler: gcc-7 + # clang-runtime: '8' + + # - name: ubu20-gcc7-runtime11-analyzers + # os: ubuntu-20.04 + # compiler: gcc-7 + # clang-runtime: '11' + # coverage: true + # cuda: true + # extra_cmake_options: '-DLLVM_ENABLE_WERROR=On -DENABLE_ENZYME_BACKEND=On' + # #clang-format: true + + # - name: ubu20-gcc7-runtime11-benchmarks + # os: ubuntu-20.04 + # compiler: gcc-7 + # clang-runtime: '11' + # extra_cmake_options: '-DCLAD_ENABLE_BENCHMARKS=On -DENABLE_ENZYME_BACKEND=On' + # benchmark: true + + # - name: ubu20-gcc8-runtime11-coverity + # os: ubuntu-20.04 + # compiler: gcc-8 + # clang-runtime: '11' + # coverity: true + + # - name: ubu20-gcc9-runtime9 + # os: ubuntu-20.04 + # compiler: gcc-9 + # clang-runtime: '9' + + # - name: ubu20-clang8-runtime9 + # os: ubuntu-20.04 + # compiler: clang-8 + # clang-runtime: '9' + + # - name: ubu20-clang9-runtime9 + # os: ubuntu-20.04 + # compiler: clang-9 + # clang-runtime: '9' + + # - name: ubu20-gcc7-runtime10 + # os: ubuntu-20.04 + # compiler: gcc-7 + # clang-runtime: '10' + + # - name: ubu20-gcc8-runtime10 + # os: ubuntu-20.04 + # compiler: gcc-8 + # clang-runtime: '10' + + # - name: ubu20-gcc9-runtime10 + # os: ubuntu-20.04 + # compiler: gcc-9 + # clang-runtime: '10' + + # - name: ubu20-clang8-runtime10 + # os: ubuntu-20.04 + # compiler: clang-8 + # clang-runtime: '10' + + # - name: ubu20-clang9-runtime10 + # os: ubuntu-20.04 + # compiler: clang-9 + # clang-runtime: '10' + + # - name: ubu20-gcc7-runtime11 + # os: ubuntu-20.04 + # compiler: gcc-7 + # clang-runtime: '11' + + # - name: ubu20-gcc8-runtime11 + # os: ubuntu-20.04 + # compiler: gcc-8 + # clang-runtime: '11' + + # - name: ubu20-gcc9-runtime11 + # os: ubuntu-20.04 + # compiler: gcc-9 + # clang-runtime: '11' + + # - name: ubu20-clang8-runtime11 + # os: ubuntu-20.04 + # compiler: clang-8 + # clang-runtime: '11' + + # - name: ubu20-clang9-runtime11 + # os: ubuntu-20.04 + # compiler: clang-9 + # clang-runtime: '11' + + # - name: ubu20-gcc7-runtime12 + # os: ubuntu-20.04 + # compiler: gcc-7 + # clang-runtime: '12' + + # - name: ubu20-gcc8-runtime12 + # os: ubuntu-20.04 + # compiler: gcc-8 + # clang-runtime: '12' + + # - name: ubu20-gcc9-runtime12 + # os: ubuntu-20.04 + # compiler: gcc-9 + # clang-runtime: '12' + # - name: ubu20-clang8-runtime12 + # os: ubuntu-20.04 + # compiler: clang-8 + # clang-runtime: '12' + + # - name: ubu20-clang9-runtime12 + # os: ubuntu-20.04 + # compiler: clang-9 + # clang-runtime: '12' + + # - name: ubu20-clang9-runtime10-cuda + # os: ubuntu-20.04 + # compiler: clang-9 + # clang-runtime: '10' + # cuda: true + + # - name: ubu20-clang9-runtime15 + # os: ubuntu-20.04 + # compiler: clang-9 + # clang-runtime: '15' + + # - name: ubu20-clang9-runtime16 + # os: ubuntu-20.04 + # compiler: clang-9 + # clang-runtime: '16' + + # - name: ubu22-gcc9-runtime11 + # os: ubuntu-22.04 + # compiler: gcc-9 + # clang-runtime: '11' + + # - name: ubu22-gcc10-runtime11 + # os: ubuntu-22.04 + # compiler: gcc-10 + # clang-runtime: '11' + + # - name: ubu22-gcc11-runtime11 + # os: ubuntu-22.04 + # compiler: gcc-11 + # clang-runtime: '11' + + # - name: ubu22-clang11-runtime11 + # os: ubuntu-22.04 + # compiler: 'clang-11' + # clang-runtime: '11' + + # - name: ubu22-clang12-runtime11 + # os: ubuntu-22.04 + # compiler: clang-12 + # clang-runtime: '11' + + # - name: ubu22-clang13-runtime11 + # os: ubuntu-22.04 + # compiler: clang-13 + # clang-runtime: '11' + + # - name: ubu22-clang14-runtime11 + # os: ubuntu-22.04 + # compiler: clang-14 + # clang-runtime: '11' + + # - name: ubu22-clang15-runtime11 + # os: ubuntu-22.04 + # compiler: clang-15 + # clang-runtime: '11' + + # - name: ubu22-gcc9-runtime12 + # os: ubuntu-22.04 + # compiler: gcc-9 + # clang-runtime: '12' + + # - name: ubu22-gcc10-runtime12 + # os: ubuntu-22.04 + # compiler: gcc-10 + # clang-runtime: '12' + + # - name: ubu22-gcc11-runtime12 + # os: ubuntu-22.04 + # compiler: gcc-11 + # clang-runtime: '12' + + # - name: ubu22-clang11-runtime12 + # os: ubuntu-22.04 + # compiler: 'clang-11' + # clang-runtime: '12' + + # - name: ubu22-clang12-runtime12 + # os: ubuntu-22.04 + # compiler: clang-12 + # clang-runtime: '12' + + # - name: ubu22-clang13-runtime12 + # os: ubuntu-22.04 + # compiler: clang-13 + # clang-runtime: '12' + + # - name: ubu22-clang14-runtime12 + # os: ubuntu-22.04 + # compiler: clang-14 + # clang-runtime: '12' + + # - name: ubu22-clang15-runtime12 + # os: ubuntu-22.04 + # compiler: clang-15 + # clang-runtime: '12' + + # - name: ubu22-gcc10-runtime13 + # os: ubuntu-22.04 + # compiler: gcc-10 + # clang-runtime: '13' + + # - name: ubu22-gcc11-runtime13 + # os: ubuntu-22.04 + # compiler: gcc-11 + # clang-runtime: '13' + + # - name: ubu22-gcc12-runtime13 + # os: ubuntu-22.04 + # compiler: gcc-12 + # clang-runtime: '13' + + # - name: ubu22-clang11-runtime13 + # os: ubuntu-22.04 + # compiler: clang-11 + # clang-runtime: '13' + + # - name: ubu22-clang12-runtime13 + # os: ubuntu-22.04 + # compiler: clang-12 + # clang-runtime: '13' + + # - name: ubu22-clang13-runtime13 + # os: ubuntu-22.04 + # compiler: clang-13 + # clang-runtime: '13' + + # - name: ubu22-clang14-runtime13 + # os: ubuntu-22.04 + # compiler: clang-14 + # clang-runtime: '13' + + # - name: ubu22-clang15-runtime13 + # os: ubuntu-22.04 + # compiler: clang-15 + # clang-runtime: '13' + + # - name: ubu22-gcc10-runtime14 + # os: ubuntu-22.04 + # compiler: gcc-10 + # clang-runtime: '14' + + # - name: ubu22-gcc11-runtime14 + # os: ubuntu-22.04 + # compiler: gcc-11 + # clang-runtime: '14' + + # - name: ubu22-gcc12-runtime14 + # os: ubuntu-22.04 + # compiler: gcc-12 + # clang-runtime: '14' + + # - name: ubu22-gcc12-runtime15 + # os: ubuntu-22.04 + # compiler: gcc-12 + # clang-runtime: '15' + + # - name: ubu22-gcc12-runtime16 + # os: ubuntu-22.04 + # compiler: gcc-12 + # clang-runtime: '16' + + # - name: ubu22-clang11-runtime14 + # os: ubuntu-22.04 + # compiler: clang-11 + # clang-runtime: '14' + + # - name: ubu22-clang12-runtime14 + # os: ubuntu-22.04 + # compiler: clang-12 + # clang-runtime: '14' + + # - name: ubu22-clang13-runtime14 + # os: ubuntu-22.04 + # compiler: clang-13 + # clang-runtime: '14' + + # - name: ubu22-clang14-runtime14 + # os: ubuntu-22.04 + # compiler: clang-14 + # clang-runtime: '14' + + # - name: ubu22-clang14-runtime15 + # os: ubuntu-22.04 + # compiler: clang-14 + # clang-runtime: '15' + + # - name: ubu22-clang15-runtime14 + # os: ubuntu-22.04 + # compiler: clang-15 + # clang-runtime: '14' + + # - name: ubu22-clang15-runtime15 + # os: ubuntu-22.04 + # compiler: clang-15 + # clang-runtime: '15' + + # - name: ubu22-clang15-runtime16 + # os: ubuntu-22.04 + # compiler: clang-15 + # clang-runtime: '16' steps: - uses: actions/checkout@v3 @@ -748,7 +748,7 @@ jobs: if: ${{ failure() }} uses: mxschmitt/action-tmate@v3 # When debugging increase to a suitable value! - timeout-minutes: ${{ github.event.pull_request && 1 || 20 }} + timeout-minutes: ${{ github.event.pull_request && 20 || 20 }} - name: Prepare code coverage report if: ${{ success() && (matrix.coverage == true) }} run: | diff --git a/include/clad/Differentiator/BuiltinDerivatives.h b/include/clad/Differentiator/BuiltinDerivatives.h index d7dd9d859..af6950945 100644 --- a/include/clad/Differentiator/BuiltinDerivatives.h +++ b/include/clad/Differentiator/BuiltinDerivatives.h @@ -13,6 +13,7 @@ namespace custom_derivatives{} #include "clad/Differentiator/ArrayRef.h" #include "clad/Differentiator/CladConfig.h" +#include #include namespace clad { @@ -139,32 +140,34 @@ CUDA_HOST_DEVICE void fma_pullback(T1 a, T2 b, T3 c, T4 d_y, *d_c += d_y; } -template -CUDA_HOST_DEVICE ValueAndPushforward -min_pushforward(const T& a, const T& b, const T& d_a, const T& d_b) { +template +CUDA_HOST_DEVICE ValueAndPushforward +min_pushforward(const T1& a, const T2& b, const T1& d_a, const T2& d_b) { return {::std::min(a, b), a < b ? d_a : d_b}; } -template -CUDA_HOST_DEVICE ValueAndPushforward -max_pushforward(const T& a, const T& b, const T& d_a, const T& d_b) { +template +CUDA_HOST_DEVICE ValueAndPushforward +max_pushforward(const T1& a, const T2& b, const T1& d_a, const T2& d_b) { return {::std::max(a, b), a < b ? d_b : d_a}; } -template -CUDA_HOST_DEVICE void min_pullback(const T& a, const T& b, U d_y, - clad::array_ref d_a, - clad::array_ref d_b) { +template +CUDA_HOST_DEVICE void min_pullback(const T1& a, const T2& b, U d_y, + clad::array_ref d_a, + clad::array_ref d_b) { if (a < b) *d_a += d_y; else *d_b += d_y; } -template -CUDA_HOST_DEVICE void max_pullback(const T& a, const T& b, U d_y, - clad::array_ref d_a, - clad::array_ref d_b) { +template +CUDA_HOST_DEVICE void max_pullback(const T1& a, const T2& b, U d_y, + clad::array_ref d_a, + clad::array_ref d_b) { if (a < b) *d_b += d_y; else @@ -172,19 +175,20 @@ CUDA_HOST_DEVICE void max_pullback(const T& a, const T& b, U d_y, } #if __cplusplus >= 201703L -template -CUDA_HOST_DEVICE ValueAndPushforward -clamp_pushforward(const T& v, const T& lo, const T& hi, const T& d_v, - const T& d_lo, const T& d_hi) { +template +CUDA_HOST_DEVICE ValueAndPushforward +clamp_pushforward(const T1& v, const T2& lo, const T3& hi, const T1& d_v, + const T2& d_lo, const T3& d_hi) { return {::std::clamp(v, lo, hi), v < lo ? d_lo : hi < v ? d_hi : d_v}; } -template -CUDA_HOST_DEVICE void clamp_pullback(const T& v, const T& lo, const T& hi, +template +CUDA_HOST_DEVICE void clamp_pullback(const T1& v, const T2& lo, const T3& hi, const U& d_y, - clad::array_ref d_v, - clad::array_ref d_lo, - clad::array_ref d_hi) { + clad::array_ref d_v, + clad::array_ref d_lo, + clad::array_ref d_hi) { if (v < lo) *d_lo += d_y; else if (hi < v) diff --git a/lib/Differentiator/ReverseModeVisitor.cpp b/lib/Differentiator/ReverseModeVisitor.cpp index 4e5d7020e..639d7c064 100644 --- a/lib/Differentiator/ReverseModeVisitor.cpp +++ b/lib/Differentiator/ReverseModeVisitor.cpp @@ -1768,6 +1768,20 @@ Expr* getArraySizeExpr(const ArrayType* AT, ASTContext& context, BuildOp(UnaryOperatorKind::UO_AddrOf, baseDiff.getExpr())); std::string customPullback = clad::utils::ComputeEffectiveFnName(FD) + "_pullback"; + if (customPullback.find("max") != std::string::npos || + customPullback.find("min") != std::string::npos || + customPullback.find("clamp") != std::string::npos) { + // print signature of FD + llvm ::outs() << "Function signature: \n"; + FD->print(llvm::outs(), 0, false); + // print the name of pullback and its arguments + llvm ::outs() << "Pullback function name: " << customPullback << "\n"; + for (auto arg : pullbackCallArgs) { + llvm ::outs() << "Pullback function argument: "; + arg->printPrettyControlled(llvm ::outs(), nullptr, + m_Context.getPrintingPolicy(), 0); + } + } OverloadedDerivedFn = m_Builder.BuildCallToCustomDerivativeOrNumericalDiff( customPullback, pullbackCallArgs, getCurrentScope(),