Skip to content

Commit

Permalink
Update to generative-recommenders@7906fe79 (#92)
Browse files Browse the repository at this point in the history 
Summary: Pull Request resolved: #92

Reviewed By: xuzhao9

Differential Revision: D66705018

Pulled By: bertmaher

fbshipit-source-id: 01b019eb13ef322d3b2c2fcf9fcd2816ba066c5e
  • Loading branch information
@bertmaher @facebook-github-bot
bertmaher authored and facebook-github-bot committed Dec 3, 2024
1 parent ce74f5e commit f92c38e
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 f92c38e

Please sign in to comment.