vyre-driver-cuda 0.6.1

CUDA/PTX backend for vyre through the CUDA driver API.
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

//! CUDA launch-wave planning for resident e-graph device images.
//!
//! Equality-saturation kernels need deterministic row, child-edge, and
//! e-class-group work partitions. This module converts the checked resident
//! image view into bounded launch waves without rebuilding graph metadata or
//! depending on e-graph semantics in the CUDA backend.

#[path = "egraph_kernel_plan/constants.rs"]
mod constants;
#[path = "egraph_kernel_plan/types_launch.rs"]
mod types_launch;
#[path = "egraph_kernel_plan/types_snapshot.rs"]
mod types_snapshot;
#[path = "egraph_kernel_plan/types_signature.rs"]
mod types_signature;
#[path = "egraph_kernel_plan/types_union.rs"]
mod types_union;
#[path = "egraph_kernel_plan/types_canonicalization.rs"]
mod types_canonicalization;
#[path = "egraph_kernel_plan/error.rs"]
mod error;
#[path = "egraph_kernel_plan/helpers.rs"]
mod helpers;
#[path = "egraph_kernel_plan/plan_kernel_work.rs"]
mod plan_kernel_work;
#[path = "egraph_kernel_plan/plan_signature.rs"]
mod plan_signature;
#[path = "egraph_kernel_plan/plan_equivalence.rs"]
mod plan_equivalence;
#[path = "egraph_kernel_plan/plan_union.rs"]
mod plan_union;
#[path = "egraph_kernel_plan/backend_structural.rs"]
mod backend_structural;
#[path = "egraph_kernel_plan/backend_rewrite.rs"]
mod backend_rewrite;
#[path = "egraph_kernel_plan/backend_canonicalization.rs"]
mod backend_canonicalization;

#[path = "egraph_kernel_plan/args.rs"]
mod args;
#[path = "egraph_kernel_plan/ptx.rs"]
mod ptx;

#[cfg(test)]
#[path = "egraph_kernel_plan/tests.rs"]
mod tests;

pub use constants::*;
pub use error::CudaEGraphKernelPlanError;
pub use helpers::cuda_egraph_signature_pair_rows;
pub use plan_equivalence::{
    collect_cuda_egraph_structural_equivalences, pack_cuda_egraph_signature_bucket_device_image,
    plan_cuda_egraph_structural_equivalence_launch_artifact,
    plan_cuda_egraph_structural_equivalence_launch_artifact_from_plan,
    plan_cuda_egraph_structural_equivalence_output, plan_cuda_egraph_structural_equivalences,
};
pub use plan_kernel_work::plan_cuda_egraph_kernel_work;
pub use plan_signature::{
    plan_cuda_egraph_signature_buckets, plan_cuda_egraph_signature_buckets_from_resident_snapshot,
    plan_cuda_egraph_signature_buckets_from_signature_snapshot,
};
pub use plan_union::{
    pack_cuda_egraph_canonical_rewrite_device_image, plan_cuda_egraph_union_compaction,
};
pub use ptx::{
    cuda_egraph_canonical_rewrite_kernel_ptx, cuda_egraph_signature_refresh_kernel_ptx,
    cuda_egraph_structural_equivalence_kernel_ptx,
};
pub use types_canonicalization::*;
pub use types_launch::*;
pub use types_signature::*;
pub use types_snapshot::*;
pub use types_union::*;