Updated 0.1.0 version #1

scaomath · 2025-05-19T19:24:42Z

Implemented the FVM method on a staggered MAC grid (pressure on cell centers).
Added native PyTorch implementation for applying torch.linalg and torch.fft functions directly on GridArray and GridVariable.
Added native implementation of arithmetic manipulation directly on GridVariableVector.
Added several helper functions consistent_grid to replace consistent_grid_arrays.
Removed dependence of from torch.utils._pytree import register_pytree_node
Some tests are added. Requires package pytest

scaomath added 9 commits May 10, 2025 22:37

move fft functions in equations to spectral

ae1635d

reformat

bcc68c2

removed the typing hint Array, changed to torch.Tensor directly

0a02b25

fix some typing hints

09a0be2

from . imports

d02e4bf

removed nested imports

28cc4a9

change input of NS2dSpectral to initialized class

883d61c

replace import from torch_cfd.equations with torch_cfd.spectral

4fc7924

added fvm

1e28b9c

scaomath merged commit 475c738 into main May 19, 2025

scaomath deleted the dev-0.1 branch May 19, 2025 19:25

Provide feedback