无法在 C++ 中使用带有 -O3 选项的 clang 进行尾递归

如何解决无法在 C++ 中使用带有 -O3 选项的 clang 进行尾递归

我无法告诉 clang 编译器在我的 C++ 代码中执行尾递归优化。我看到了这篇文章 Which,if any,C++ compilers do tail-recursion optimization?,其中的建议是在 clang 中使用 -O3 标志,它会这样做。但尽管如此,对于以下素性测试代码的大量输入,我还是遇到了堆栈溢出问题。

$keys = array_unique(array_merge(array_keys($array1),array_keys($array2)));

$result = array();
foreach ($keys as $key) {
    $result[$key]['dates'] = array_merge($array1[$key]['dates'] ?? [],$array2[$key]['dates'] ?? []);
}
print_r($result);

我现在正在阅读 SICP 的第 1 章,并且有一个关于使用 O(√n) 算法检查素性的练习 (1.22)。尾递归由 Scheme 标准强制执行,因此对于大输入没有问题,但我在 C++ 中做同样的问题,对于大数字,最小除数函数消耗堆栈。我正在使用 GMP 库进行大整数运算。

是否可以在这样的程序中强制执行尾递归?

谢谢

解决方法

mpz_is_divisible 有点奇怪。一方面,您忘记释放 rem 持有的内存。即使您添加 mpz_clear(rem) 调用,您也不会在 smallest_divisor 中获得 TCO。您需要做的是将其标记为 __attribute__((noinline)) 以让 Clang 进行优化(尽管 GCC 在没有它的情况下会这样做)。非常奇怪,也没有实际意义,因为 mpz_is_divisible 只是 mpz_divisible_p 的更差版本。

void smallest_divisor(mpz_t n,mpz_t div,mpz_t div_sqr,mpz_t smallest_div) {    
    mpz_sqr(div_sqr,div);
    if (mpz_cmp(div_sqr,n) > 0) {
        mpz_set(smallest_div,n);
        return;
    }
    if (mpz_divisible_p(n,div)) {
        mpz_set(smallest_div,div);
        return;
    }
    mpz_add_ui(div,div,1);
    smallest_divisor(n,div_sqr,smallest_div); // gets TCO
}

Godbolt(从主干 GCC 中窃取 GMP,可能需要使用当前日期更新某些路径才能使其工作)

我还大量清理了代码以生成新版本。特别是,我的 FP 直觉告诉我 smallest_divisor 真的不应该是它自己的函数。它是属于 mpz_prime_test_basic 内部的递归工作者。我们可以使用 this answer 来编写本地递归定义。我在 Godbolt 上找不到 gmpxx.h,所以我也写了一个 mpz_class 的克隆。然后我们就可以写

bool is_prime(mpz_t n) noexcept {
    mpz div(2L);
    fix{[](auto rec,mpz_t n,mpz_t div_sqr) noexcept -> void {
        mpz_mul(div_sqr,div);
        if(mpz_cmp(div_sqr,n) > 0) mpz_set(div,n);
        else if(!mpz_divisible_p(n,div)) {
            mpz_add_ui(div,1);
            rec(n,div_sqr);
        }
    }}(n,mpz());
    return mpz_cmp(div,n) == 0;
}

因为我们实际上并没有改变调用之间的参数,只是在它们后面发生了变化,所以我们也可以只写

bool is_prime(mpz_t n) noexcept {
    mpz div(2L),div_sqr;
    fix{[&](auto rec) noexcept -> void {
        mpz_mul(div_sqr,1);
            rec();
        }
    }}();
    return mpz_cmp(div,n) == 0;
}

Godbolt(同样的警告)

,

原则上,所有主要的 C++ 编译器都会进行尾调用优化。您的代码应该从中受益。

查看从 Godbolt with Clang 11 上的代码获得的程序集后,-O3 标志似乎确实导致尾代码优化。您的递归函数编译为:

smallest_divisor(test,test,test): # @smallest_divisor(test,test)
  push rax
.LBB1_1: # =>This Inner Loop Header: Depth=1  <------ TCO Loop 
  mov edi,2
  call mpz_pow_ui(test,unsigned int)
  call mpz_cmp(test,test)
  test eax,eax
  jg .LBB1_4                                  <------ go to return sequence
  call mpz_init(test)
  call mpz_tdiv_r(test,test)
  xor edi,edi
  call mpz_cmp_si(test,int)
  test eax,eax
  je .LBB1_4
  mov edi,1
  call mpz_add_ui(test,unsigned int)
  jmp .LBB1_1                                <--- no recursive call but loop
.LBB1_4:                                     <--- this is the end of recursion
  pop rax
  jmp mpz_set(test,test) # TAILCALL         <-- no ret,but another tail call :-)

我无法测试您的代码,因为我没有该库,也没有时间投资获取和安装它。

根据您描述的症状,我可以看到以下可能的根本原因:

  • 库中有一些 UB,导致段错误
  • 由于编译器设置,优化被禁用。例如,如果您使用 XCode,则需要转到菜单 Product > Scheme > Edit Scheme...,然后选择“Release”作为构建配置。

在 XCode 中尝试过汇编(产品 > 执行操作 > 汇编),生成的汇编代码显示了 TCO:

__Z16smallest_divisor4testS_S_S_:       ## @_Z16smallest_divisor4testS_S_S_
Lfunc_begin1:
    .loc    67 29 0                 ## Test0105-1/main.cpp:29:0
    .cfi_startproc
## %bb.0:
    pushq   %rbp
    .cfi_def_cfa_offset 16
    .cfi_offset %rbp,-16
    movq    %rsp,%rbp
    .cfi_def_cfa_register %rbp
LBB1_1:                                 ## =>This Inner Loop Header: Depth=1
Ltmp3:
    .loc    67 24 4 prologue_end    ## Test0105-1/main.cpp:24:4
    movl    $2,%edi
    callq   __Z10mpz_pow_ui4testS_j
   ....
    movl    $1,%edi
    callq   __Z10mpz_add_ui4testS_j
Ltmp13:
    .loc    67 0 3 is_stmt 0        ## Test0105-1/main.cpp:0:3
    jmp LBB1_1                             <<<<<<<<<<<< Tail call optimization (loop)
LBB1_4:
    popq    %rbp
Ltmp14:
    jmp __Z7mpz_set4testS_      ## TAILCALL

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


使用本地python环境可以成功执行 import pandas as pd import matplotlib.pyplot as plt # 设置字体 plt.rcParams[&#39;font.sans-serif&#39;] = [&#39;SimHei&#39;] # 能正确显示负号 p
错误1:Request method ‘DELETE‘ not supported 错误还原:controller层有一个接口,访问该接口时报错:Request method ‘DELETE‘ not supported 错误原因:没有接收到前端传入的参数,修改为如下 参考 错误2:cannot r
错误1:启动docker镜像时报错:Error response from daemon: driver failed programming external connectivity on endpoint quirky_allen 解决方法:重启docker -&gt; systemctl r
错误1:private field ‘xxx‘ is never assigned 按Altʾnter快捷键,选择第2项 参考:https://blog.csdn.net/shi_hong_fei_hei/article/details/88814070 错误2:启动时报错,不能找到主启动类 #
报错如下,通过源不能下载,最后警告pip需升级版本 Requirement already satisfied: pip in c:\users\ychen\appdata\local\programs\python\python310\lib\site-packages (22.0.4) Coll
错误1:maven打包报错 错误还原:使用maven打包项目时报错如下 [ERROR] Failed to execute goal org.apache.maven.plugins:maven-resources-plugin:3.2.0:resources (default-resources)
错误1:服务调用时报错 服务消费者模块assess通过openFeign调用服务提供者模块hires 如下为服务提供者模块hires的控制层接口 @RestController @RequestMapping(&quot;/hires&quot;) public class FeignControl
错误1:运行项目后报如下错误 解决方案 报错2:Failed to execute goal org.apache.maven.plugins:maven-compiler-plugin:3.8.1:compile (default-compile) on project sb 解决方案:在pom.
参考 错误原因 过滤器或拦截器在生效时,redisTemplate还没有注入 解决方案:在注入容器时就生效 @Component //项目运行时就注入Spring容器 public class RedisBean { @Resource private RedisTemplate&lt;String
使用vite构建项目报错 C:\Users\ychen\work&gt;npm init @vitejs/app @vitejs/create-app is deprecated, use npm init vite instead C:\Users\ychen\AppData\Local\npm-
参考1 参考2 解决方案 # 点击安装源 协议选择 http:// 路径填写 mirrors.aliyun.com/centos/8.3.2011/BaseOS/x86_64/os URL类型 软件库URL 其他路径 # 版本 7 mirrors.aliyun.com/centos/7/os/x86
报错1 [root@slave1 data_mocker]# kafka-console-consumer.sh --bootstrap-server slave1:9092 --topic topic_db [2023-12-19 18:31:12,770] WARN [Consumer clie
错误1 # 重写数据 hive (edu)&gt; insert overwrite table dwd_trade_cart_add_inc &gt; select data.id, &gt; data.user_id, &gt; data.course_id, &gt; date_format(
错误1 hive (edu)&gt; insert into huanhuan values(1,&#39;haoge&#39;); Query ID = root_20240110071417_fe1517ad-3607-41f4-bdcf-d00b98ac443e Total jobs = 1
报错1:执行到如下就不执行了,没有显示Successfully registered new MBean. [root@slave1 bin]# /usr/local/software/flume-1.9.0/bin/flume-ng agent -n a1 -c /usr/local/softwa
虚拟及没有启动任何服务器查看jps会显示jps,如果没有显示任何东西 [root@slave2 ~]# jps 9647 Jps 解决方案 # 进入/tmp查看 [root@slave1 dfs]# cd /tmp [root@slave1 tmp]# ll 总用量 48 drwxr-xr-x. 2
报错1 hive&gt; show databases; OK Failed with exception java.io.IOException:java.lang.RuntimeException: Error in configuring object Time taken: 0.474 se
报错1 [root@localhost ~]# vim -bash: vim: 未找到命令 安装vim yum -y install vim* # 查看是否安装成功 [root@hadoop01 hadoop]# rpm -qa |grep vim vim-X11-7.4.629-8.el7_9.x
修改hadoop配置 vi /usr/local/software/hadoop-2.9.2/etc/hadoop/yarn-site.xml # 添加如下 &lt;configuration&gt; &lt;property&gt; &lt;name&gt;yarn.nodemanager.res