Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Jokeren committed Apr 22, 2024
1 parent 286eb60 commit 36bd87a
Showing 1 changed file with 13 additions and 4 deletions.
17 changes: 13 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,9 +1,18 @@
# FASTEN: A Library of *Fast Segment* Operators
# Fasten: A Library of *Fast Segment* Operators

## Introduction
FASTEN is a library aimed at speeding up Heterogeneous Graph Neural Network (HGNN) workloads.
The current version of FASTEN focuses on improving segmented matrix multiplication, a critical operator in HGNNs.
Fasten is a library aimed at speeding up Heterogeneous Graph Neural Network (HGNN) workloads.
The current version of Fasten focuses on improving segmented matrix multiplication, a critical operator in HGNNs.
Fasten implements a simple interface, making it easy to integrate with existing graph library PyG with minimal changes.
Fasten achieved an average speedup of 5.71$\times$ and 2.46$\times$ in operator-wise benchmarks compared to CUTLASS and cuBLAS, respectively

### Fasten vs CUTLASS (no autotune)

![figure9_fasten_vs_cutlass(pyg)](https://github.com/Deep-Learning-Profiling-Tools/fasten/assets/2306281/d88fab7c-a331-4978-9157-08e448afcce5)

### Fasten vs cuBLAS

![figure9_fasten_vs_cublas(torch)](https://github.com/Deep-Learning-Profiling-Tools/fasten/assets/2306281/4f8fbe5f-f8d4-45b2-9f92-ac3f7cb97c28)

## Installation

Expand All @@ -24,7 +33,7 @@ pip install .
```

## Examples
FASTEN's segment matrix multiplication operator has been integrated with various HGNN architecture such as *RGCN*, *HGT*, *RGAT* in PyG.
Fasten's segment matrix multiplication operator has been integrated with various HGNN architecture such as *RGCN*, *HGT*, *RGAT* in PyG.
Examples on how to run the examples can be found below:

### GNN Examples
Expand Down

0 comments on commit 36bd87a

Please sign in to comment.