solvr 0.2.0-beta.2

Advanced computing library for real-world problem solving - optimization, differential equations, interpolation, statistics, and more
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

//! WebGPU implementation of finite element PDE solvers.

use numr::runtime::wgpu::{WgpuClient, WgpuRuntime};
use numr::tensor::Tensor;

use crate::pde::error::PdeResult;
use crate::pde::impl_generic::{fem_1d_impl, fem_triangular_impl};
use crate::pde::traits::FiniteElementAlgorithms;
use crate::pde::types::{BoundarySpec, FdmOptions, FemResult};

impl FiniteElementAlgorithms<WgpuRuntime> for WgpuClient {
    fn fem_1d(
        &self,
        f_rhs: &Tensor<WgpuRuntime>,
        x_nodes: &Tensor<WgpuRuntime>,
        boundary: &[BoundarySpec<WgpuRuntime>],
        options: &FdmOptions,
    ) -> PdeResult<FemResult<WgpuRuntime>> {
        fem_1d_impl(self, f_rhs, x_nodes, boundary, options)
    }

    fn fem_triangular(
        &self,
        f_rhs: &Tensor<WgpuRuntime>,
        nodes: &Tensor<WgpuRuntime>,
        elements: &Tensor<WgpuRuntime>,
        boundary_nodes: &Tensor<WgpuRuntime>,
        boundary_values: &Tensor<WgpuRuntime>,
        options: &FdmOptions,
    ) -> PdeResult<FemResult<WgpuRuntime>> {
        fem_triangular_impl(
            self,
            f_rhs,
            nodes,
            elements,
            boundary_nodes,
            boundary_values,
            options,
        )
    }
}