Init

rileysu · Jun 24, 2023 · 01c6366 · 01c6366
commit 01c6366
Show file tree

Hide file tree

Showing 13 changed files with 225 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0  1,2 @@
 /target
 /old_src
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -0,0  1,3 @@
 {
     "rust-analyzer.showUnlinkedFileNotification": false
 }
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -0,0  1,8 @@
 [package]
 name = "therml"
 version = "0.1.0"
 edition = "2021"
 
 [dependencies]
 ord_subset = "3.1.1"
 wide = "0.7.8"
diff --git a/ideas.md b/ideas.md
@@ -0,0  1,8 @@
 # Ideas
 
 ## Context
 
 - Context creates tensors and handles data storage
 - Internally will create graphs for operations
 - Will handle gradient calc
 
diff --git a/src/engine/mod.rs b/src/engine/mod.rs
@@ -0,0  1,5 @@
 mod tensor;
 
 pub trait Engine {
     type Data: 
 }
diff --git a/src/engine/tensor/array.rs b/src/engine/tensor/array.rs
@@ -0,0  1,45 @@
 use std::sync::Arc;
 use crate::helper::{Shape, Stride};
 use super::EngineTensor;
 
 struct ArrayEngineTensor<T: Sized> {
     data: Arc<[T]>,
     shape: Shape,
     stride: Stride,
 }
 
 
 
 impl<T: Sized> EngineTensor<T> for ArrayEngineTensor<T> {
     type EngineTensorIterator = ArrayEngineTensorIterator<T>;
 
     fn shape(&self) -> Shape {
         todo!()
     }
 
     fn stride(&self) -> Stride {
         todo!()
     }
 
     fn view(&self, shape: Shape) -> Result<Self, ()> {
         todo!()
     }
 
     fn iter(&self, comp_range: &[std::ops::Range<usize>]) -> Self::EngineTensorIterator {
         todo!()
     }
 }
 
 struct ArrayEngineTensorIterator<T: Sized> {
     base: ArrayEngineTensor<T>,
     pos: Vec<usize>,
     end: Vec<usize>,
 }
 
 impl<T: Sized> Iterator for ArrayEngineTensorIterator<T> {
     type Item = T;
 
     fn next(&mut self) -> Option<Self::Item> {
 
     }
 }
diff --git a/src/engine/tensor/mod.rs b/src/engine/tensor/mod.rs
@@ -0,0  1,27 @@
 pub mod array;
 
 use std::ops::Range;
 
 use crate::helper::{Shape, Stride};
 
 // No mutation should be done through this trait
 // It exists purely as a way to get data for the engine
 trait EngineTensor<T: Sized>
     where Self: Sized 
 {
     type EngineTensorIterator: Iterator<Item = T>;
 
     fn shape(&self) -> Shape;
     fn stride(&self) -> Stride;
 
     fn is_contiguous(&self) -> bool {
         Stride::from(self.shape()) == self.stride()
     }
 
     // Create a view into the same memory
     fn view(&self, shape: Shape) -> Result<Self, ()>;
 
     // Iterator from element to element in order
     fn iter(&self, comp_range: &[Range<usize>]) -> Self::EngineTensorIterator;
     //fn (comp_range: &[Range<usize>]) -> Self::DataSlice;
 }
diff --git a/src/helper/mod.rs b/src/helper/mod.rs
@@ -0,0  1,5 @@
 mod shape;
 mod stride;
 
 pub use shape::*;
 pub use stride::*;
diff --git a/src/helper/position.rs b/src/helper/position.rs
@@ -0,0  1,7 @@
 #[derive(Clone, PartialEq, Eq, Debug)]
 struct Position(Box<[usize]>);
 
 impl Position {
 
 }
 
diff --git a/src/helper/shape.rs b/src/helper/shape.rs
@@ -0,0  1,18 @@
 #[derive(Clone, PartialEq, Eq, Debug)]
 pub struct Shape(Box<[usize]>);
 
 impl Shape {
     pub fn as_boxed_slice(&self) -> &Box<[usize]> {
         &self.0
     }
 
     pub fn total_elements(&self) -> usize {
         self.0.iter().product()
     }
 }
 
 impl From<&[usize]> for Shape {
     fn from(value: &[usize]) -> Self {
         Self(Box::from(value))
     }
 }
diff --git a/src/helper/stride.rs b/src/helper/stride.rs
@@ -0,0  1,49 @@
 use crate::helper::Shape;
 
 #[derive(Clone, PartialEq, Eq, Debug)]
 pub struct Stride(Box<[usize]>);
 
 impl Stride {
     pub fn as_boxed_slice(&self) -> &Box<[usize]> {
         &self.0
     }
 }
 
 impl From<&[usize]> for Stride {
     fn from(value: &[usize]) -> Self {
         Self(Box::from(value))
     }
 }
 
 impl From<Shape> for Stride {
     fn from(value: Shape) -> Self {
         let mut stride: Vec<usize> = Vec::with_capacity(value.as_boxed_slice().len());
 
         let mut next = 1usize;
         for dim in value.as_boxed_slice().iter().rev() {
             stride.push(next);
             next = dim * next;
         }
 
         stride.reverse();
 
         Stride(stride.into_boxed_slice())
     }
 }
 
 #[cfg(test)]
 mod test {
     use super::*;
 
     #[test]
     fn from_shape_examples() {
         let examples = [
             (Shape::from([1].as_slice()), Stride::from([1].as_slice())),
             (Shape::from([20, 50, 4].as_slice()), Stride::from([200, 4, 1].as_slice()))
         ];
 
         for (shape, stride) in examples {
             assert_eq!(Stride::from(shape), stride);
         }
     }
 }
diff --git a/src/main.rs b/src/main.rs
@@ -0,0  1,6 @@
 mod engine;
 mod helper;
 
 fn main() {
 
 }