Skip to content

Commit

Permalink
Fix hstu submodule
Browse files Browse the repository at this point in the history
  • Loading branch information
@xuzhao9
xuzhao9 committed Dec 10, 2024
1 parent ba55033 commit f057ace
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion submodules/generative-recommenders
Submodule generative-recommenders updated 27 files
+13 −10 README.md
+4 −0 generative_recommenders/data/eval.py
+2 −58 generative_recommenders/indexing/candidate_index.py
+8 −2 generative_recommenders/indexing/utils.py
+0 −569 generative_recommenders/modeling/similarity/mol.py
+59 −89 generative_recommenders/modeling/similarity_utils.py
+8 −0 generative_recommenders/ops/cpp/common.h
+44 −0 generative_recommenders/ops/cpp/complete_cumsum.cpp
+49 −0 generative_recommenders/ops/cpp/complete_cumsum.cu
+49 −0 generative_recommenders/ops/cpp/cpp_ops.cpp
+40 −0 generative_recommenders/ops/cpp/sort_kv_pairs_cuda.cpp
+82 −0 generative_recommenders/ops/cpp/sort_kv_pairs_cuda_kernels_template.cu
+12 −0 generative_recommenders/ops/cpp/sort_kv_pairs_cuda_kernels_template.h
+196 −0 generative_recommenders/ops/pytorch/jagged.py
+152 −52 generative_recommenders/ops/triton/triton_jagged.py
+25 −0 generative_recommenders/ops/triton/triton_layer_norm.py
+41 −0 generative_recommenders/ops/triton/triton_position.py
+97 −28 generative_recommenders/ops/triton/triton_ragged_hstu_attention.py
+41 −0 generative_recommenders/rails/indexing/candidate_index.py
+2 −3 generative_recommenders/rails/indexing/mips_top_k.py
+133 −0 generative_recommenders/rails/indexing/mol_top_k.py
+80 −0 generative_recommenders/rails/similarities/layers.py
+50 −0 generative_recommenders/rails/similarities/mol/embeddings_fn.py
+96 −0 generative_recommenders/rails/similarities/mol/item_embeddings_fn.py
+161 −0 generative_recommenders/rails/similarities/mol/query_embeddings_fn.py
+383 −0 generative_recommenders/rails/similarities/mol/similarity_fn.py
+20 −4 generative_recommenders/trainer/train.py

0 comments on commit f057ace

Please sign in to comment.