Skip to content

Commit

Permalink
Use fork of hstu
Browse files Browse the repository at this point in the history
  • Loading branch information
@xuzhao9
xuzhao9 committed Nov 14, 2024
1 parent d2f37a4 commit b140dcc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion submodules/generative-recommenders
Submodule generative-recommenders updated 23 files
+16 −1 README.md
+3 −3 generative_recommenders/data/item_features.py
+1 −3 generative_recommenders/modeling/sequential/input_features_preprocessors.py
+50 −0 generative_recommenders/ops/cpp/batched_complete_cumsum.cpp
+136 −0 generative_recommenders/ops/cpp/batched_complete_cumsum.cu
+52 −0 generative_recommenders/ops/cpp/common.h
+111 −0 generative_recommenders/ops/cpp/concat_1d_jagged_jagged.cpp
+127 −0 generative_recommenders/ops/cpp/concat_1d_jagged_jagged.cu
+99 −0 generative_recommenders/ops/cpp/cpp_ops.cpp
+97 −0 generative_recommenders/ops/cpp/expand_1d_jagged_to_dense.cpp
+103 −0 generative_recommenders/ops/cpp/expand_1d_jagged_to_dense.cu
+6 −3 generative_recommenders/ops/triton/triton_addmm.py
+1,693 −0 generative_recommenders/ops/triton/triton_hstu_linear.py
+2,389 −0 generative_recommenders/ops/triton/triton_jagged.py
+965 −0 generative_recommenders/ops/triton/triton_layer_norm.py
+800 −0 generative_recommenders/ops/triton/triton_position.py
+3,670 −0 generative_recommenders/ops/triton/triton_ragged_hstu_attention.py
+143 −0 generative_recommenders/ops/triton/utils.py
+0 −7 generative_recommenders/trainer/train.py
+0 −1 main.py
+0 −1,147 ops/triton/triton_ragged_hstu_attention.py
+2 −0 run_fractal_expansion.py
+0 −178 tests/ragged_hstu_attention_test.py

0 comments on commit b140dcc

Please sign in to comment.