dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix comments

This commit is contained in:
nathaniel 2024-09-16 09:10:34 -04:00
parent a1d2b13e3e
commit 395d84ce71
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
crates/burn-fusion/src/stream/base.rs
View File

@ -5,13 +5,13 @@ use burn_tensor::repr::OperationDescription;
/// A growing list of [tensor operation descriptions](OperationDescription). /// A growing list of [tensor operation descriptions](OperationDescription).
pub struct OperationQueue<R: FusionRuntime> { pub struct OperationQueue<R: FusionRuntime> {
/// List of operation descriptions. These contain the exact tensor IDs /// List of operation descriptions. These contain the exact tensor IDs
/// and shapes so that kernels can be optimized correctly. /// and shapes so that kernels can be run correctly.
/// ///
/// The length of this list is the same as the length of the `operations` list. /// The length of this list is the same as the length of the `operations` list.
pub(crate) global: Vec<OperationDescription>, pub(crate) global: Vec<OperationDescription>,
/// List of operation descriptions. The tensor IDs and shapes are relative /// List of operation descriptions. The tensor IDs and shapes are relative
/// because we don't need to know the exact values to determine which operations /// because we don't need to know the exact values, but they are sufficient to
/// can be fused. /// determine which operations can be fused.
pub(crate) relative: Vec<OperationDescription>, pub(crate) relative: Vec<OperationDescription>,
pub(crate) converter: OperationConverter, pub(crate) converter: OperationConverter,
pub(crate) operations: Vec<Box<dyn Operation<R>>>, pub(crate) operations: Vec<Box<dyn Operation<R>>>,