PaddlePaddle / PaddleScience

PaddleScience is SDK and library for developing AI-driven scientific computing applications based on PaddlePaddle.
http://paddlescience-docs.rtfd.io/
Apache License 2.0
289 stars 178 forks source link

【Paddle Toolkit Development Competition No.5】Paddle 适配 torch-harmonics #1013

Open co63oc opened 2 weeks ago

co63oc commented 2 weeks ago

PR types

Others

PR changes

Others

Describe

Paddle 适配 torch-harmonics RFC https://github.com/PaddlePaddle/community/pull/1002

安装依赖triton,triton依赖torch,会同时安装 torch

examples/train_sfno.py 运行比较 Paddle image

Torch image

单测59项运行完成 image

Notebooks/支持运行文件 conditioning_sht.ipynb getting_started.ipynb helmholtz.ipynb plot_spherical_harmonics.ipynb quadrature.ipynb shallow_water_equations.ipynb train_sfno.ipynb

Notebooks/gradient_analysis.ipynb 中SGD使用complex64类型参数不支持,不支持运行

paddle-bot[bot] commented 2 weeks ago

Thanks for your contribution!

lijialin03 commented 1 week ago

pytest时有一个循环引用的报错: image

lijialin03 commented 1 week ago

paddle3.0下,test_convolution 报错,大概看了一下可能是因为reshape的shape不对,比如应该是.reshape([-1, 1, 1])但是现在是.reshape(-1, 1, 1) image

co63oc commented 1 week ago

paddle3.0下,test_convolution 报错,大概看了一下可能是因为reshape的shape不对,比如应该是.reshape([-1, 1, 1])但是现在是.reshape(-1, 1, 1)

paddle_aux对reshape的参数是有修改的应该是可以,我再看下

co63oc commented 1 week ago

image 安装的3.0.0.dev20241111版本,test_convolution.py没有问题 image utils/paddle_aux.py 处理reshape的部分,在源文件中引入paddle_aux 会替换reshape image

lijialin03 commented 1 week ago

image 安装的3.0.0.dev20241111版本,test_convolution.py没有问题 image utils/paddle_aux.py 处理reshape的部分,在源文件中引入paddle_aux 会替换reshape image

确实,是triton版本的问题,更新到 3.1.0 test全部跑通

co63oc commented 1 week ago

确实,是triton版本的问题,更新到 3.1.0 test全部跑通

好的

co63oc commented 6 days ago

已创建私有仓库 https://github.com/PFCCLab/paddle_harmonics

HydrogenSulfate commented 4 days ago

已创建私有仓库 https://github.com/PFCCLab/paddle_harmonics

好的