├── .github ├── PULL_REQUEST_TEMPLATE.md └── workflows │ ├── cla.yml │ ├── lint.yml │ └── mkdocs.yml ├── .gitignore ├── .pre-commit-config.yaml ├── AUTHORS ├── CHANGELOG.md ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── assets ├── XLB_diff.png ├── airfoil.gif ├── airfoil.png ├── building.png ├── car.png ├── cavity.gif ├── logo-transparent.png └── wind_turbine.gif ├── docs ├── index.md └── javascripts │ └── mathjax.js ├── examples ├── cfd │ ├── data │ │ └── turbulent_channel_dns_data.json │ ├── flow_past_sphere_3d.py │ ├── lid_driven_cavity_2d.py │ ├── lid_driven_cavity_2d_distributed.py │ ├── turbulent_channel_3d.py │ └── windtunnel_3d.py ├── ibm │ ├── airfoil_ibm.py │ ├── sphere_ibm.py │ ├── wind_turbine_ibm.py │ └── windtunnel_ibm.py ├── out_of_core │ ├── assets │ │ └── nvidia_new.stl │ ├── autodiff_lbm.py │ ├── ds │ │ ├── __init__.py │ │ └── ooc_grid.py │ ├── lid_driven_cavity.py │ ├── operators │ │ ├── __init__.py │ │ ├── clamp_field.py │ │ ├── color_mapper.py │ │ ├── gradient_descent.py │ │ ├── initialize_target_density.py │ │ ├── l2_loss.py │ │ ├── mesh_renderer.py │ │ ├── q_criterion.py │ │ ├── soa_copy.py │ │ ├── transform_mesh.py │ │ ├── trilinear_interpolation.py │ │ └── uniform_initializer.py │ └── subroutine │ │ ├── __init__.py │ │ ├── autodiff_stepper_subroutine.py │ │ ├── gradient_descent.py │ │ ├── initialize_field.py │ │ ├── prepare_fields.py │ │ ├── render_q_criterion.py │ │ ├── rho_loss_subroutine.py │ │ ├── stepper_subroutine.py │ │ ├── subroutine.py │ │ └── volume_saver_subroutine.py └── performance │ └── mlups_3d.py ├── mkdocs.yml ├── requirements.txt ├── ruff.toml ├── setup.py ├── tests ├── __init__.py ├── boundary_conditions │ ├── bc_equilibrium │ │ ├── test_bc_equilibrium_jax.py │ │ └── test_bc_equilibrium_warp.py │ ├── bc_fullway_bounce_back │ │ ├── test_bc_fullway_bounce_back_jax.py │ │ └── test_bc_fullway_bounce_back_warp.py │ └── mask │ │ ├── test_bc_indices_masker_jax.py │ │ └── test_bc_indices_masker_warp.py ├── grids │ ├── __init__.py │ ├── test_grid_jax.py │ └── test_grid_warp.py └── kernels │ ├── __init__.py │ ├── collision │ ├── test_bgk_collision_jax.py │ └── test_bgk_collision_warp.py │ ├── equilibrium │ ├── test_equilibrium_jax.py │ └── test_equilibrium_warp.py │ ├── macroscopic │ ├── test_macroscopic_jax.py │ └── test_macroscopic_warp.py │ └── stream │ ├── test_stream_jax.py │ └── test_stream_warp.py └── xlb ├── __init__.py ├── compute_backend.py ├── default_config.py ├── distribute ├── __init__.py └── distribute.py ├── experimental ├── __init__.py └── ooc │ ├── __init__.py │ ├── ooc_array.py │ ├── out_of_core.py │ ├── tiles │ ├── __init__.py │ ├── compressed_tile.py │ ├── dense_tile.py │ ├── dynamic_array.py │ └── tile.py │ └── utils.py ├── grid ├── __init__.py ├── grid.py ├── jax_grid.py └── warp_grid.py ├── grid_backend.py ├── helper ├── __init__.py ├── check_boundary_overlaps.py ├── ibm_helper.py ├── initializers.py └── nse_solver.py ├── operator ├── __init__.py ├── boundary_condition │ ├── __init__.py │ ├── bc_do_nothing.py │ ├── bc_equilibrium.py │ ├── bc_extrapolation_outflow.py │ ├── bc_fullway_bounce_back.py │ ├── bc_grads_approximation.py │ ├── bc_halfway_bounce_back.py │ ├── bc_regularized.py │ ├── bc_zouhe.py │ ├── boundary_condition.py │ ├── boundary_condition_registry.py │ └── helper_functions_bc.py ├── boundary_masker │ ├── __init__.py │ ├── indices_boundary_masker.py │ └── mesh_boundary_masker.py ├── collision │ ├── __init__.py │ ├── bgk.py │ ├── collision.py │ ├── forced_collision.py │ ├── kbc.py │ └── smagorinsky_les_bgk.py ├── equilibrium │ ├── __init__.py │ ├── equilibrium.py │ └── quadratic_equilibrium.py ├── force │ ├── __init__.py │ ├── exact_difference_force.py │ └── momentum_transfer.py ├── macroscopic │ ├── __init__.py │ ├── first_moment.py │ ├── macroscopic.py │ ├── second_moment.py │ └── zero_moment.py ├── operator.py ├── parallel_operator.py ├── postprocess │ ├── __init__.py │ ├── grid_to_point.py │ ├── q_criterion.py │ └── vorticity.py ├── precision_caster │ ├── __init__.py │ └── precision_caster.py ├── stepper │ ├── __init__.py │ ├── ibm_stepper.py │ ├── nse_stepper.py │ └── stepper.py └── stream │ ├── __init__.py │ └── stream.py ├── physics_type.py ├── precision_policy.py ├── precision_policy └── precision_policy.py ├── utils ├── __init__.py └── utils.py └── velocity_set ├── __init__.py ├── d2q9.py ├── d3q19.py ├── d3q27.py └── velocity_set.py /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/workflows/cla.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/.github/workflows/cla.yml -------------------------------------------------------------------------------- /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.github/workflows/mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/.github/workflows/mkdocs.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /AUTHORS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/AUTHORS -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/README.md -------------------------------------------------------------------------------- /assets/XLB_diff.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/assets/XLB_diff.png -------------------------------------------------------------------------------- /assets/airfoil.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/assets/airfoil.gif -------------------------------------------------------------------------------- /assets/airfoil.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/assets/airfoil.png -------------------------------------------------------------------------------- /assets/building.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/assets/building.png -------------------------------------------------------------------------------- /assets/car.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/assets/car.png -------------------------------------------------------------------------------- /assets/cavity.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/assets/cavity.gif -------------------------------------------------------------------------------- /assets/logo-transparent.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/assets/logo-transparent.png -------------------------------------------------------------------------------- /assets/wind_turbine.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/assets/wind_turbine.gif -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/javascripts/mathjax.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/docs/javascripts/mathjax.js -------------------------------------------------------------------------------- /examples/cfd/data/turbulent_channel_dns_data.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/cfd/data/turbulent_channel_dns_data.json -------------------------------------------------------------------------------- /examples/cfd/flow_past_sphere_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/cfd/flow_past_sphere_3d.py -------------------------------------------------------------------------------- /examples/cfd/lid_driven_cavity_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/cfd/lid_driven_cavity_2d.py -------------------------------------------------------------------------------- /examples/cfd/lid_driven_cavity_2d_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/cfd/lid_driven_cavity_2d_distributed.py -------------------------------------------------------------------------------- /examples/cfd/turbulent_channel_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/cfd/turbulent_channel_3d.py -------------------------------------------------------------------------------- /examples/cfd/windtunnel_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/cfd/windtunnel_3d.py -------------------------------------------------------------------------------- /examples/ibm/airfoil_ibm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/ibm/airfoil_ibm.py -------------------------------------------------------------------------------- /examples/ibm/sphere_ibm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/ibm/sphere_ibm.py -------------------------------------------------------------------------------- /examples/ibm/wind_turbine_ibm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/ibm/wind_turbine_ibm.py -------------------------------------------------------------------------------- /examples/ibm/windtunnel_ibm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/ibm/windtunnel_ibm.py -------------------------------------------------------------------------------- /examples/out_of_core/assets/nvidia_new.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/assets/nvidia_new.stl -------------------------------------------------------------------------------- /examples/out_of_core/autodiff_lbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/autodiff_lbm.py -------------------------------------------------------------------------------- /examples/out_of_core/ds/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/ds/__init__.py -------------------------------------------------------------------------------- /examples/out_of_core/ds/ooc_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/ds/ooc_grid.py -------------------------------------------------------------------------------- /examples/out_of_core/lid_driven_cavity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/lid_driven_cavity.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/__init__.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/clamp_field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/clamp_field.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/color_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/color_mapper.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/gradient_descent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/gradient_descent.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/initialize_target_density.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/initialize_target_density.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/l2_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/l2_loss.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/mesh_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/mesh_renderer.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/q_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/q_criterion.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/soa_copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/soa_copy.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/transform_mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/transform_mesh.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/trilinear_interpolation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/trilinear_interpolation.py -------------------------------------------------------------------------------- /examples/out_of_core/operators/uniform_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/operators/uniform_initializer.py -------------------------------------------------------------------------------- /examples/out_of_core/subroutine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/subroutine/__init__.py -------------------------------------------------------------------------------- /examples/out_of_core/subroutine/autodiff_stepper_subroutine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/subroutine/autodiff_stepper_subroutine.py -------------------------------------------------------------------------------- /examples/out_of_core/subroutine/gradient_descent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/subroutine/gradient_descent.py -------------------------------------------------------------------------------- /examples/out_of_core/subroutine/initialize_field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/subroutine/initialize_field.py -------------------------------------------------------------------------------- /examples/out_of_core/subroutine/prepare_fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/subroutine/prepare_fields.py -------------------------------------------------------------------------------- /examples/out_of_core/subroutine/render_q_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/subroutine/render_q_criterion.py -------------------------------------------------------------------------------- /examples/out_of_core/subroutine/rho_loss_subroutine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/subroutine/rho_loss_subroutine.py -------------------------------------------------------------------------------- /examples/out_of_core/subroutine/stepper_subroutine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/subroutine/stepper_subroutine.py -------------------------------------------------------------------------------- /examples/out_of_core/subroutine/subroutine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/subroutine/subroutine.py -------------------------------------------------------------------------------- /examples/out_of_core/subroutine/volume_saver_subroutine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/out_of_core/subroutine/volume_saver_subroutine.py -------------------------------------------------------------------------------- /examples/performance/mlups_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/examples/performance/mlups_3d.py -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/requirements.txt -------------------------------------------------------------------------------- /ruff.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/ruff.toml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/boundary_conditions/bc_equilibrium/test_bc_equilibrium_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/boundary_conditions/bc_equilibrium/test_bc_equilibrium_jax.py -------------------------------------------------------------------------------- /tests/boundary_conditions/bc_equilibrium/test_bc_equilibrium_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/boundary_conditions/bc_equilibrium/test_bc_equilibrium_warp.py -------------------------------------------------------------------------------- /tests/boundary_conditions/bc_fullway_bounce_back/test_bc_fullway_bounce_back_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/boundary_conditions/bc_fullway_bounce_back/test_bc_fullway_bounce_back_jax.py -------------------------------------------------------------------------------- /tests/boundary_conditions/bc_fullway_bounce_back/test_bc_fullway_bounce_back_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/boundary_conditions/bc_fullway_bounce_back/test_bc_fullway_bounce_back_warp.py -------------------------------------------------------------------------------- /tests/boundary_conditions/mask/test_bc_indices_masker_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/boundary_conditions/mask/test_bc_indices_masker_jax.py -------------------------------------------------------------------------------- /tests/boundary_conditions/mask/test_bc_indices_masker_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/boundary_conditions/mask/test_bc_indices_masker_warp.py -------------------------------------------------------------------------------- /tests/grids/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/grids/test_grid_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/grids/test_grid_jax.py -------------------------------------------------------------------------------- /tests/grids/test_grid_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/grids/test_grid_warp.py -------------------------------------------------------------------------------- /tests/kernels/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/kernels/collision/test_bgk_collision_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/kernels/collision/test_bgk_collision_jax.py -------------------------------------------------------------------------------- /tests/kernels/collision/test_bgk_collision_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/kernels/collision/test_bgk_collision_warp.py -------------------------------------------------------------------------------- /tests/kernels/equilibrium/test_equilibrium_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/kernels/equilibrium/test_equilibrium_jax.py -------------------------------------------------------------------------------- /tests/kernels/equilibrium/test_equilibrium_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/kernels/equilibrium/test_equilibrium_warp.py -------------------------------------------------------------------------------- /tests/kernels/macroscopic/test_macroscopic_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/kernels/macroscopic/test_macroscopic_jax.py -------------------------------------------------------------------------------- /tests/kernels/macroscopic/test_macroscopic_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/kernels/macroscopic/test_macroscopic_warp.py -------------------------------------------------------------------------------- /tests/kernels/stream/test_stream_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/kernels/stream/test_stream_jax.py -------------------------------------------------------------------------------- /tests/kernels/stream/test_stream_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/tests/kernels/stream/test_stream_warp.py -------------------------------------------------------------------------------- /xlb/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/__init__.py -------------------------------------------------------------------------------- /xlb/compute_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/compute_backend.py -------------------------------------------------------------------------------- /xlb/default_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/default_config.py -------------------------------------------------------------------------------- /xlb/distribute/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/distribute/__init__.py -------------------------------------------------------------------------------- /xlb/distribute/distribute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/distribute/distribute.py -------------------------------------------------------------------------------- /xlb/experimental/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /xlb/experimental/ooc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/experimental/ooc/__init__.py -------------------------------------------------------------------------------- /xlb/experimental/ooc/ooc_array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/experimental/ooc/ooc_array.py -------------------------------------------------------------------------------- /xlb/experimental/ooc/out_of_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/experimental/ooc/out_of_core.py -------------------------------------------------------------------------------- /xlb/experimental/ooc/tiles/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /xlb/experimental/ooc/tiles/compressed_tile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/experimental/ooc/tiles/compressed_tile.py -------------------------------------------------------------------------------- /xlb/experimental/ooc/tiles/dense_tile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/experimental/ooc/tiles/dense_tile.py -------------------------------------------------------------------------------- /xlb/experimental/ooc/tiles/dynamic_array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/experimental/ooc/tiles/dynamic_array.py -------------------------------------------------------------------------------- /xlb/experimental/ooc/tiles/tile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/experimental/ooc/tiles/tile.py -------------------------------------------------------------------------------- /xlb/experimental/ooc/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/experimental/ooc/utils.py -------------------------------------------------------------------------------- /xlb/grid/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/grid/__init__.py -------------------------------------------------------------------------------- /xlb/grid/grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/grid/grid.py -------------------------------------------------------------------------------- /xlb/grid/jax_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/grid/jax_grid.py -------------------------------------------------------------------------------- /xlb/grid/warp_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/grid/warp_grid.py -------------------------------------------------------------------------------- /xlb/grid_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/grid_backend.py -------------------------------------------------------------------------------- /xlb/helper/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/helper/__init__.py -------------------------------------------------------------------------------- /xlb/helper/check_boundary_overlaps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/helper/check_boundary_overlaps.py -------------------------------------------------------------------------------- /xlb/helper/ibm_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/helper/ibm_helper.py -------------------------------------------------------------------------------- /xlb/helper/initializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/helper/initializers.py -------------------------------------------------------------------------------- /xlb/helper/nse_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/helper/nse_solver.py -------------------------------------------------------------------------------- /xlb/operator/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/__init__.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/__init__.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/bc_do_nothing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/bc_do_nothing.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/bc_equilibrium.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/bc_equilibrium.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/bc_extrapolation_outflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/bc_extrapolation_outflow.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/bc_fullway_bounce_back.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/bc_fullway_bounce_back.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/bc_grads_approximation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/bc_grads_approximation.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/bc_halfway_bounce_back.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/bc_halfway_bounce_back.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/bc_regularized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/bc_regularized.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/bc_zouhe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/bc_zouhe.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/boundary_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/boundary_condition.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/boundary_condition_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/boundary_condition_registry.py -------------------------------------------------------------------------------- /xlb/operator/boundary_condition/helper_functions_bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_condition/helper_functions_bc.py -------------------------------------------------------------------------------- /xlb/operator/boundary_masker/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_masker/__init__.py -------------------------------------------------------------------------------- /xlb/operator/boundary_masker/indices_boundary_masker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_masker/indices_boundary_masker.py -------------------------------------------------------------------------------- /xlb/operator/boundary_masker/mesh_boundary_masker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/boundary_masker/mesh_boundary_masker.py -------------------------------------------------------------------------------- /xlb/operator/collision/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/collision/__init__.py -------------------------------------------------------------------------------- /xlb/operator/collision/bgk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/collision/bgk.py -------------------------------------------------------------------------------- /xlb/operator/collision/collision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/collision/collision.py -------------------------------------------------------------------------------- /xlb/operator/collision/forced_collision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/collision/forced_collision.py -------------------------------------------------------------------------------- /xlb/operator/collision/kbc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/collision/kbc.py -------------------------------------------------------------------------------- /xlb/operator/collision/smagorinsky_les_bgk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/collision/smagorinsky_les_bgk.py -------------------------------------------------------------------------------- /xlb/operator/equilibrium/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/equilibrium/__init__.py -------------------------------------------------------------------------------- /xlb/operator/equilibrium/equilibrium.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/equilibrium/equilibrium.py -------------------------------------------------------------------------------- /xlb/operator/equilibrium/quadratic_equilibrium.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/equilibrium/quadratic_equilibrium.py -------------------------------------------------------------------------------- /xlb/operator/force/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/force/__init__.py -------------------------------------------------------------------------------- /xlb/operator/force/exact_difference_force.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/force/exact_difference_force.py -------------------------------------------------------------------------------- /xlb/operator/force/momentum_transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/force/momentum_transfer.py -------------------------------------------------------------------------------- /xlb/operator/macroscopic/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/macroscopic/__init__.py -------------------------------------------------------------------------------- /xlb/operator/macroscopic/first_moment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/macroscopic/first_moment.py -------------------------------------------------------------------------------- /xlb/operator/macroscopic/macroscopic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/macroscopic/macroscopic.py -------------------------------------------------------------------------------- /xlb/operator/macroscopic/second_moment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/macroscopic/second_moment.py -------------------------------------------------------------------------------- /xlb/operator/macroscopic/zero_moment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/macroscopic/zero_moment.py -------------------------------------------------------------------------------- /xlb/operator/operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/operator.py -------------------------------------------------------------------------------- /xlb/operator/parallel_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/parallel_operator.py -------------------------------------------------------------------------------- /xlb/operator/postprocess/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/postprocess/__init__.py -------------------------------------------------------------------------------- /xlb/operator/postprocess/grid_to_point.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/postprocess/grid_to_point.py -------------------------------------------------------------------------------- /xlb/operator/postprocess/q_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/postprocess/q_criterion.py -------------------------------------------------------------------------------- /xlb/operator/postprocess/vorticity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/postprocess/vorticity.py -------------------------------------------------------------------------------- /xlb/operator/precision_caster/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/precision_caster/__init__.py -------------------------------------------------------------------------------- /xlb/operator/precision_caster/precision_caster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/precision_caster/precision_caster.py -------------------------------------------------------------------------------- /xlb/operator/stepper/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/stepper/__init__.py -------------------------------------------------------------------------------- /xlb/operator/stepper/ibm_stepper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/stepper/ibm_stepper.py -------------------------------------------------------------------------------- /xlb/operator/stepper/nse_stepper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/stepper/nse_stepper.py -------------------------------------------------------------------------------- /xlb/operator/stepper/stepper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/stepper/stepper.py -------------------------------------------------------------------------------- /xlb/operator/stream/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/stream/__init__.py -------------------------------------------------------------------------------- /xlb/operator/stream/stream.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/operator/stream/stream.py -------------------------------------------------------------------------------- /xlb/physics_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/physics_type.py -------------------------------------------------------------------------------- /xlb/precision_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/precision_policy.py -------------------------------------------------------------------------------- /xlb/precision_policy/precision_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/precision_policy/precision_policy.py -------------------------------------------------------------------------------- /xlb/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/utils/__init__.py -------------------------------------------------------------------------------- /xlb/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/utils/utils.py -------------------------------------------------------------------------------- /xlb/velocity_set/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/velocity_set/__init__.py -------------------------------------------------------------------------------- /xlb/velocity_set/d2q9.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/velocity_set/d2q9.py -------------------------------------------------------------------------------- /xlb/velocity_set/d3q19.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/velocity_set/d3q19.py -------------------------------------------------------------------------------- /xlb/velocity_set/d3q27.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/velocity_set/d3q27.py -------------------------------------------------------------------------------- /xlb/velocity_set/velocity_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Autodesk/XLB/HEAD/xlb/velocity_set/velocity_set.py --------------------------------------------------------------------------------