Error: torch_xla/csrc/convert_ops.cpp:83 : Unsupported XLA type 10

最新推荐文章于 2024-03-07 16:28:05 发布

讨饭

最新推荐文章于 2024-03-07 16:28:05 发布

阅读量377

点赞数

分类专栏： Python错误信息文章标签： python 深度学习 pytorch

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/qq_40418553/article/details/126332861

版权

Python错误信息专栏收录该内容

12 篇文章 0 订阅

订阅专栏

本文讲述了在使用张量处理时，如何将类型为torch.float16的张量h转换为bfloat16类型，以便在TPU上进行高效运算，重点在于h=torch.sum(h,dim=[2,3])的代码调整过程。

摘要由CSDN通过智能技术生成

我的错误出现在

h = torch.sum(h, dim=[2, 3])

而h的类型为

tensor([], device='xla:1', dtype=torch.float16, grad_fn=<ReluBackward0>)

因为xla在TPU上运算需要使用 bfloat16

所以更改h的类型为bfloat16

h = h.type(torch.bfloat16)
h = torch.sum(h, dim=[2, 3])

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

讨饭 CSDN认证博客专家 CSDN认证企业博客

码龄7年

19: 原创

107万+: 周排名

173万+: 总排名

7万+: 访问

: 等级

299: 积分

4: 粉丝

39: 获赞

22: 评论

98: 收藏

私信

关注

热门文章

分类专栏

最新评论

Python socket 中的recv函数 | 发送长消息
Him_Hei: 此方法无论数据大小1到无限大recv时皆无需输入接收的大小(因为用的是bytes的big方法以16进制表达，而且bytes大小固定是4，具体实现不懂反正100000也是4位bytes)，就是(recv(1024或其它数字))，而且无需md5自带数据检验功能
Python socket 中的recv函数 | 发送长消息
Him_Hei: 我刚刚解决了这个题，思路是在socket的基础上自己写一个send和recv函数: import socket def recv(client: socket): _size = client.recv(4) # 从发送端获得大小 size = int.from_bytes(_size, byteorder="big") # 转化为int数值 client.send(_size) # 发送大小给发送端校验并准备接收数据 d = client.recv(size) # 等待接受大小数据校验失败的结果或者数据的到来 if d == b"SIZE_ERR": # 如果发送端大小校验失败 raise BufferError("Size Error") # 抛出异常(此时recv函数中断被迫返回, 等于brace) client.send(d) # 发送接受到的数据给发送端校验 if client.recv(4) == b"PASS": # 接受校验结果如果PASS return d # 返回数据 else: raise BufferError("Data Error") # 否则抛出异常(此时recv函数中断被迫返回, 等于brace) def send(client, d: bytes): size = len(d) # 获得数据大小 _size = size.to_bytes(4, byteorder="big") # 转化为4字节大小 client.send(_size) # 发送大小给接收端 if client.recv(4) == _size: # 校验接收端获取到的数据大小是否正确，如果正确则继续发送数据 client.send(d) # 发送数据给接收端 if client.recv(size) == d: # 校验接收端获取到的数据是否正确，如果正确则继续发送校验结果 client.send(b"PASS") # 发送PASS校验结果 else: client.send(b"FAIL") # 否则发送FAIL校验结果 else: # 大小校验错误反转 d = b"SIZE_ERROR" client.send(d) # 发送SIZE_ERROR(此行执行前的接受端阻塞在”d = client.recv(size)"语句中等待校验结果 raise BufferError("Size Error") # 大小校验失败抛出异常给到引用代码处理各位上眼
BLS公钥发送 | bn256模块 | pickle
老黄G: 这个库能实现聚合签名吗
Python socket 中的recv函数 | 发送长消息
梧桐叶宇: while len(data) < length: to_recv = min(MAX_PACKET_SIZE, length - len(data)) packet = conn.recv(to_recv) if not packet: break data += packet
iClord.exe 系统错误. 由于找不到iUtils.dll，无法继续执行代码。重新安装程序可能会解决此问题。
CSDN-Ada助手: 恭喜您第20篇博客的发布！您的文章标题“iClord.exe系统错误. 由于找不到iUtils.dll，无法继续执行代码。重新安装程序可能会解决此问题。”让我对您的技术造诣和热情赞叹不已。尽管在技术领域中遇到问题是常有的事情，但您的解决方案总是令人信服。我想提出下一步的创作建议，希望能够在您的博客中加入更多技术细节和实用的解决方案。这将进一步丰富您的文章内容，为读者提供更多有价值的信息。期待您的下一篇博客，再次感谢您的分享！

最新文章

目录

评论

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

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