pytorch 1.4.0 ROCM 3.1 ubuntu 18.4 编译 实占踩坑记

为了一尝平民的法拉力, AMD Radeon VII (雷七)的美味(其实还是没钱,上不起2080ti), 自行在ubuntu上编译 pytorch,

大体步骤参照 如下blog处理:

https://blog.csdn.net/duoganniang1006/article/details/103654781

基本上问题不大,在最后的编译步骤中踩坑了:

某个文件编译不了,查找google, 只发现一个仁兄跟我有类似情况,但是没有解决方案。作为cpp程序员,不能在此小问题之前掉价。

在源码的 caffe2/operators/hip/relu_op.hip:55/105 两行有如下代码:

 

__float2half2_rn(xx.x >0.0f ?xx.x:0.0f , xx.y >0.0f ?xx.y:0.0f)

 

这样类似的语句,加粗部分因为推断的类型有岐义,所以无法编译。需要作如下修改

__float2half2_rn(xx.x >0.0f ? (float)xx.x:0.0f , xx.y >0.0f ?(float)xx.y:0.0f)

 

 

遂自己行修改,强行编译。

 

 

 

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值