fbgemm-gpu-nightly-cpu 2025.3.27__cp311-cp311-manylinux_2_28_aarch64.whl → 2026.1.29__cp311-cp311-manylinux_2_28_aarch64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (106) hide show
  1. fbgemm_gpu/__init__.py +118 -23
  2. fbgemm_gpu/asmjit.so +0 -0
  3. fbgemm_gpu/batched_unary_embeddings_ops.py +3 -3
  4. fbgemm_gpu/config/feature_list.py +7 -1
  5. fbgemm_gpu/docs/jagged_tensor_ops.py +0 -1
  6. fbgemm_gpu/docs/sparse_ops.py +142 -1
  7. fbgemm_gpu/docs/target.default.json.py +6 -0
  8. fbgemm_gpu/enums.py +3 -4
  9. fbgemm_gpu/fbgemm.so +0 -0
  10. fbgemm_gpu/fbgemm_gpu_config.so +0 -0
  11. fbgemm_gpu/fbgemm_gpu_embedding_inplace_ops.so +0 -0
  12. fbgemm_gpu/fbgemm_gpu_py.so +0 -0
  13. fbgemm_gpu/fbgemm_gpu_sparse_async_cumsum.so +0 -0
  14. fbgemm_gpu/fbgemm_gpu_tbe_cache.so +0 -0
  15. fbgemm_gpu/fbgemm_gpu_tbe_common.so +0 -0
  16. fbgemm_gpu/fbgemm_gpu_tbe_index_select.so +0 -0
  17. fbgemm_gpu/fbgemm_gpu_tbe_inference.so +0 -0
  18. fbgemm_gpu/fbgemm_gpu_tbe_optimizers.so +0 -0
  19. fbgemm_gpu/fbgemm_gpu_tbe_training_backward.so +0 -0
  20. fbgemm_gpu/fbgemm_gpu_tbe_training_backward_dense.so +0 -0
  21. fbgemm_gpu/fbgemm_gpu_tbe_training_backward_gwd.so +0 -0
  22. fbgemm_gpu/fbgemm_gpu_tbe_training_backward_pt2.so +0 -0
  23. fbgemm_gpu/fbgemm_gpu_tbe_training_backward_split_host.so +0 -0
  24. fbgemm_gpu/fbgemm_gpu_tbe_training_backward_vbe.so +0 -0
  25. fbgemm_gpu/fbgemm_gpu_tbe_training_forward.so +0 -0
  26. fbgemm_gpu/fbgemm_gpu_tbe_utils.so +0 -0
  27. fbgemm_gpu/permute_pooled_embedding_modules.py +5 -4
  28. fbgemm_gpu/permute_pooled_embedding_modules_split.py +4 -4
  29. fbgemm_gpu/quantize/__init__.py +2 -0
  30. fbgemm_gpu/quantize/quantize_ops.py +1 -0
  31. fbgemm_gpu/quantize_comm.py +29 -12
  32. fbgemm_gpu/quantize_utils.py +88 -8
  33. fbgemm_gpu/runtime_monitor.py +9 -5
  34. fbgemm_gpu/sll/__init__.py +3 -0
  35. fbgemm_gpu/sll/cpu/cpu_sll.py +8 -8
  36. fbgemm_gpu/sll/triton/__init__.py +0 -10
  37. fbgemm_gpu/sll/triton/triton_jagged2_to_padded_dense.py +2 -3
  38. fbgemm_gpu/sll/triton/triton_jagged_bmm.py +2 -2
  39. fbgemm_gpu/sll/triton/triton_jagged_dense_elementwise_add.py +1 -0
  40. fbgemm_gpu/sll/triton/triton_jagged_dense_flash_attention.py +5 -6
  41. fbgemm_gpu/sll/triton/triton_jagged_flash_attention_basic.py +1 -2
  42. fbgemm_gpu/sll/triton/triton_multi_head_jagged_flash_attention.py +1 -2
  43. fbgemm_gpu/sparse_ops.py +244 -76
  44. fbgemm_gpu/split_embedding_codegen_lookup_invokers/__init__.py +26 -0
  45. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_adagrad.py +208 -105
  46. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_adam.py +261 -53
  47. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_args.py +9 -58
  48. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_args_ssd.py +10 -59
  49. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_lamb.py +225 -41
  50. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_lars_sgd.py +211 -36
  51. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_none.py +195 -26
  52. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_partial_rowwise_adam.py +225 -41
  53. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_partial_rowwise_lamb.py +225 -41
  54. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad.py +216 -111
  55. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad_ssd.py +221 -37
  56. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad_with_counter.py +259 -53
  57. fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_sgd.py +192 -96
  58. fbgemm_gpu/split_embedding_configs.py +287 -3
  59. fbgemm_gpu/split_embedding_inference_converter.py +7 -6
  60. fbgemm_gpu/split_embedding_optimizer_codegen/optimizer_args.py +2 -0
  61. fbgemm_gpu/split_embedding_optimizer_codegen/split_embedding_optimizer_rowwise_adagrad.py +2 -0
  62. fbgemm_gpu/split_table_batched_embeddings_ops_common.py +275 -9
  63. fbgemm_gpu/split_table_batched_embeddings_ops_inference.py +44 -37
  64. fbgemm_gpu/split_table_batched_embeddings_ops_training.py +900 -126
  65. fbgemm_gpu/split_table_batched_embeddings_ops_training_common.py +44 -1
  66. fbgemm_gpu/ssd_split_table_batched_embeddings_ops.py +0 -1
  67. fbgemm_gpu/tbe/bench/__init__.py +13 -2
  68. fbgemm_gpu/tbe/bench/bench_config.py +37 -9
  69. fbgemm_gpu/tbe/bench/bench_runs.py +301 -12
  70. fbgemm_gpu/tbe/bench/benchmark_click_interface.py +189 -0
  71. fbgemm_gpu/tbe/bench/eeg_cli.py +138 -0
  72. fbgemm_gpu/tbe/bench/embedding_ops_common_config.py +4 -5
  73. fbgemm_gpu/tbe/bench/eval_compression.py +3 -3
  74. fbgemm_gpu/tbe/bench/tbe_data_config.py +116 -198
  75. fbgemm_gpu/tbe/bench/tbe_data_config_bench_helper.py +332 -0
  76. fbgemm_gpu/tbe/bench/tbe_data_config_loader.py +158 -32
  77. fbgemm_gpu/tbe/bench/tbe_data_config_param_models.py +16 -8
  78. fbgemm_gpu/tbe/bench/utils.py +129 -5
  79. fbgemm_gpu/tbe/cache/__init__.py +1 -0
  80. fbgemm_gpu/tbe/cache/kv_embedding_ops_inference.py +385 -0
  81. fbgemm_gpu/tbe/cache/split_embeddings_cache_ops.py +4 -5
  82. fbgemm_gpu/tbe/ssd/common.py +27 -0
  83. fbgemm_gpu/tbe/ssd/inference.py +15 -15
  84. fbgemm_gpu/tbe/ssd/training.py +2930 -195
  85. fbgemm_gpu/tbe/ssd/utils/partially_materialized_tensor.py +34 -3
  86. fbgemm_gpu/tbe/stats/__init__.py +10 -0
  87. fbgemm_gpu/tbe/stats/bench_params_reporter.py +349 -0
  88. fbgemm_gpu/tbe/utils/offsets.py +6 -6
  89. fbgemm_gpu/tbe/utils/quantize.py +8 -8
  90. fbgemm_gpu/tbe/utils/requests.py +53 -28
  91. fbgemm_gpu/tbe_input_multiplexer.py +16 -7
  92. fbgemm_gpu/triton/common.py +0 -1
  93. fbgemm_gpu/triton/jagged/triton_jagged_tensor_ops.py +11 -11
  94. fbgemm_gpu/triton/quantize.py +14 -9
  95. fbgemm_gpu/utils/filestore.py +56 -5
  96. fbgemm_gpu/utils/torch_library.py +2 -2
  97. fbgemm_gpu/utils/writeback_util.py +124 -0
  98. fbgemm_gpu/uvm.py +3 -0
  99. {fbgemm_gpu_nightly_cpu-2025.3.27.dist-info → fbgemm_gpu_nightly_cpu-2026.1.29.dist-info}/METADATA +3 -6
  100. fbgemm_gpu_nightly_cpu-2026.1.29.dist-info/RECORD +135 -0
  101. fbgemm_gpu_nightly_cpu-2026.1.29.dist-info/top_level.txt +2 -0
  102. fbgemm_gpu/docs/version.py → list_versions/__init__.py +5 -3
  103. list_versions/cli_run.py +161 -0
  104. fbgemm_gpu_nightly_cpu-2025.3.27.dist-info/RECORD +0 -126
  105. fbgemm_gpu_nightly_cpu-2025.3.27.dist-info/top_level.txt +0 -1
  106. {fbgemm_gpu_nightly_cpu-2025.3.27.dist-info → fbgemm_gpu_nightly_cpu-2026.1.29.dist-info}/WHEEL +0 -0
@@ -0,0 +1,135 @@
1
+ fbgemm_gpu/__init__.py,sha256=JrSxUgY_diRl9kXapbyq3iteiB32D02CPan3stEFiAM,6434
2
+ fbgemm_gpu/asmjit.so,sha256=j3yeBSR2egw60Od2aIs6-mcVEhCaL4OFXQUsU2h3oyk,526272
3
+ fbgemm_gpu/batched_unary_embeddings_ops.py,sha256=pZqqUfvPIsaIo1CWX-_W087WQg-YEZuS0GNGoKFO_9c,2915
4
+ fbgemm_gpu/enums.py,sha256=37ewGSfO1x7sO31ZkRiqV1yKuklfHXT5qZIxzeeGogo,755
5
+ fbgemm_gpu/fbgemm.so,sha256=HibkS9eOXd0H4HnVQL8_sjCJhwFSwfXE8UnkXiFMMas,1378600
6
+ fbgemm_gpu/fbgemm_gpu_config.so,sha256=m9ScFJOCn8P4YQT7evcFq1g75IO9X6VB3h1Eojm1A-k,67528
7
+ fbgemm_gpu/fbgemm_gpu_embedding_inplace_ops.so,sha256=nKTMQeQptvT8uKuhdwCVFunWLjVFqQ8dKwawRIyaEbY,133400
8
+ fbgemm_gpu/fbgemm_gpu_py.so,sha256=ha2K03pws-2S001OH7HyBOfpHsNlq3ekh_XasmHdEQg,4667832
9
+ fbgemm_gpu/fbgemm_gpu_sparse_async_cumsum.so,sha256=MDL4JvJB-r72fFz_QDyqXhHn26eMff8XJh14cURL3Og,133200
10
+ fbgemm_gpu/fbgemm_gpu_tbe_cache.so,sha256=Qy2xM7LWHfn3Xh78AfNbrMJyHmyvfA2-z0fH25Cw26M,331352
11
+ fbgemm_gpu/fbgemm_gpu_tbe_common.so,sha256=QG3Y-63J_c62MIZQOESFYKZyiTwZpIIliNCY4UY47rk,463944
12
+ fbgemm_gpu/fbgemm_gpu_tbe_index_select.so,sha256=2QEr7JKp-rGfkDpT4AwjSi18RqtlmumOymZJv_Cu21w,330064
13
+ fbgemm_gpu/fbgemm_gpu_tbe_inference.so,sha256=FjQTkVJ_-E2zv7NiGLG43A-2MzIkmZku1LanCdwToGE,593608
14
+ fbgemm_gpu/fbgemm_gpu_tbe_optimizers.so,sha256=PvFTSTJPOc4c0Znvub9EFh2C534elsn6Ryb7BYPic0U,67128
15
+ fbgemm_gpu/fbgemm_gpu_tbe_training_backward.so,sha256=iOi0axu_FJL-IWEeHgIqp7dkEA1wIUU-X5sKtxlt3uk,1127800
16
+ fbgemm_gpu/fbgemm_gpu_tbe_training_backward_dense.so,sha256=tglT4kIO_7ii8f9j4wMo9D80iue8lrdMrYqTxQPWZSk,67128
17
+ fbgemm_gpu/fbgemm_gpu_tbe_training_backward_gwd.so,sha256=RoWy1q3QHApZw9SIoTq-S_KrfDAmaOwGOABQfVs0weo,67128
18
+ fbgemm_gpu/fbgemm_gpu_tbe_training_backward_pt2.so,sha256=1uU9-vaWm5jRTmCSVZ5EHxmRQ3qMohCEjSkXXP2tumc,3291352
19
+ fbgemm_gpu/fbgemm_gpu_tbe_training_backward_split_host.so,sha256=ghqy1C0cu6s_Rp2SOP0egSgHa2y-zvZXPY9CORvoT8Q,67128
20
+ fbgemm_gpu/fbgemm_gpu_tbe_training_backward_vbe.so,sha256=SG48X3jqnUV_H4XhmzQbHpcIpXTytALZ_oh-8UAeY_s,67128
21
+ fbgemm_gpu/fbgemm_gpu_tbe_training_forward.so,sha256=hl-8Df6FJO86GsOCNBLMDfWLLq7VzuQLCbhYK6lHxkU,200160
22
+ fbgemm_gpu/fbgemm_gpu_tbe_utils.so,sha256=qGGsojKO4pkIS4nwfoMI6XjFOwPGNwhuZlmGzJ0zJG4,133600
23
+ fbgemm_gpu/metrics.py,sha256=TsurFLJf0nJvPDN7urWb4LMQlf5RgdWPTTTDO7S4wtI,5663
24
+ fbgemm_gpu/permute_pooled_embedding_modules.py,sha256=B4_-TufcYZq__8ek92cKGjOIkqkZO47pQMkQSDKJIWo,5141
25
+ fbgemm_gpu/permute_pooled_embedding_modules_split.py,sha256=f3VJvH_kw9Ltd_DXtaf_PJPHmlmEWrQgzQ7MDkhh5Nw,2746
26
+ fbgemm_gpu/quantize_comm.py,sha256=gtp0zWYdobAnG6Xe8vZuTu0ZWKDu2hWmsmvV1zA09UQ,11992
27
+ fbgemm_gpu/quantize_utils.py,sha256=sROgIdOrAjQT5_CmFafg40GMo0-pe4d56bAZTI57548,10243
28
+ fbgemm_gpu/runtime_monitor.py,sha256=YXRUv6nXCsoTgh5_RzailTGvCYzwoYDb-eR4rlGwtaw,7619
29
+ fbgemm_gpu/sparse_ops.py,sha256=lJ55cgpP7MoNKo6l6QTDgvfEx8ftkJQrj8kUiIHWBvY,52183
30
+ fbgemm_gpu/split_embedding_configs.py,sha256=bEFnWzCGoHFfJIfzyusmSnSSl9tTd5C8z_j176SS0w0,16584
31
+ fbgemm_gpu/split_embedding_inference_converter.py,sha256=TpGZUXLA0rYemPT37Y0zmZnMIzjHogkRcL0gIhggbM8,7063
32
+ fbgemm_gpu/split_embedding_optimizer_ops.py,sha256=wXuGazClBMk62yL_r9udUIKaPgQP7SlkSb5ugB75wrQ,711
33
+ fbgemm_gpu/split_embedding_utils.py,sha256=Gb40ZKeATxIKEKI3aVQMgDDBanNpKMc53Z43mnzdR_I,851
34
+ fbgemm_gpu/split_table_batched_embeddings_ops.py,sha256=_MIp6uHYHLn4GxGdrGsfddfSsZ2Z9mjsYIrih3ncI1I,2339
35
+ fbgemm_gpu/split_table_batched_embeddings_ops_common.py,sha256=_uUplpcyQOQuxqv8-HV94VUM5lG8e3aGWltXhOgICQc,19294
36
+ fbgemm_gpu/split_table_batched_embeddings_ops_inference.py,sha256=dGC85xjQiRUrequBibSf9oMAVHT5Q49zsVo2zW4n_88,81679
37
+ fbgemm_gpu/split_table_batched_embeddings_ops_training.py,sha256=uCPngWxxC5OQhJv7o6aGs8xf3WlRSrdRHbpCBlPbIuE,191511
38
+ fbgemm_gpu/split_table_batched_embeddings_ops_training_common.py,sha256=jofAN2UB_iSk53Id6MBvn9Bi3Qxw67IL0_VE_EHlw_Q,7593
39
+ fbgemm_gpu/ssd_split_table_batched_embeddings_ops.py,sha256=2TTKsF5yaROTaI69YdCIt8hr_v2TDEo8EraZ0QXNBxc,717
40
+ fbgemm_gpu/tbe_input_multiplexer.py,sha256=MbZF8aZdm_kV-JRMaooeZrqlh6Pn5IuNkSXBXODp-LE,3062
41
+ fbgemm_gpu/uvm.py,sha256=5kOlOauKhOmj-B8AUqpal7riMwTfmsL0HGrh1Wweb80,1058
42
+ fbgemm_gpu/config/__init__.py,sha256=yN0KAneCICgF2BTfOYGsd0qU1PvZX_6msC6YHHZKLMg,292
43
+ fbgemm_gpu/config/feature_list.py,sha256=hhDNkkafd-Oetvuqv9ylBVTNM-lKPi029mpRqq-JZCA,2467
44
+ fbgemm_gpu/docs/__init__.py,sha256=DR6hMSQrsZALfH2AnuJQ4Zq2CfBUUhMN8YjD6APjiAE,523
45
+ fbgemm_gpu/docs/common.py,sha256=8ipXTwVb222X-aZ71O6n8fhxHCHPNhJEHMFiO7epcIs,273
46
+ fbgemm_gpu/docs/examples.py,sha256=ZMN_6sL74LH_hrp2bF_hmg8gi29GhcgvwV3kCMjxkoE,2377
47
+ fbgemm_gpu/docs/jagged_tensor_ops.py,sha256=g8MA8ezTXiqingvk1DlTZJDQcmcCZPXpshuiWxS34F0,7380
48
+ fbgemm_gpu/docs/merge_pooled_embedding_ops.py,sha256=oJLgSgZQmhsyGLbTmZTxNgQrk65_3E8xSJaWSj_Jbo8,1102
49
+ fbgemm_gpu/docs/permute_pooled_embedding_ops.py,sha256=tZUqLVXlk5O6VAKKDA-OEMx2fCu5QPOOeoAPZA9_nLY,4454
50
+ fbgemm_gpu/docs/quantize_ops.py,sha256=xTtOaVK1P02ymreE_i21YiyYDZCqhoZY9eWp_mEIRlo,1297
51
+ fbgemm_gpu/docs/sparse_ops.py,sha256=gSLUFdnu8lle_6gLewFkM20wL3ek2jKLvDGMKR6POaY,27292
52
+ fbgemm_gpu/docs/target.default.json.py,sha256=_BcuMA1hCJ_Jtf08E7O8t-R8A5HiRXHH3Z9rpgCq66U,79
53
+ fbgemm_gpu/quantize/__init__.py,sha256=yPUCmLhNdahHFireHPQMmmiRp3g6W2dkIl5MB51M6SU,942
54
+ fbgemm_gpu/quantize/quantize_ops.py,sha256=C3SN79GcL7fczzoFkxUojm6cGkvvI4iWttkGN4LFQcM,2239
55
+ fbgemm_gpu/sll/__init__.py,sha256=nLFeTiRed6A5STRi_EgHCyNoik0zhXUk2db5kTmMUNU,4221
56
+ fbgemm_gpu/sll/cpu/__init__.py,sha256=glsukNpXtf47VRIdBktILD-4CmVcf4621SGB55lT_ho,2692
57
+ fbgemm_gpu/sll/cpu/cpu_sll.py,sha256=2XyvpZ_UgSThCzUmFDQbjUdLFbz0AvhvqPR_suUcyd8,27866
58
+ fbgemm_gpu/sll/meta/__init__.py,sha256=2sMcD67XGsweBZ-UV2AEJmM4ELPsHeRAYED6kqfgAd4,1077
59
+ fbgemm_gpu/sll/meta/meta_sll.py,sha256=Jk14EOW9VPFwawD7Bwky0R0A5rmbcLWMo52oH8J6Koc,8305
60
+ fbgemm_gpu/sll/triton/__init__.py,sha256=ndvZ5OO81KP65HopJql91R9y_5fC88WnNIGYxCAVKwM,4099
61
+ fbgemm_gpu/sll/triton/common.py,sha256=hISlX4Y-7FtGof-Xx4_B8-2vlF27F9t4p2qyLMUnJ8A,798
62
+ fbgemm_gpu/sll/triton/triton_dense_jagged_cat_jagged_out.py,sha256=J9qOqjNJ72LUBqs-pGI9wrFzzzBpsZ5fzYjgfKc2YhY,1885
63
+ fbgemm_gpu/sll/triton/triton_jagged2_to_padded_dense.py,sha256=lxIYe2MUde2qxLVO_aeTm34fDsMIz8ZkIjyx9Xk-YkE,5923
64
+ fbgemm_gpu/sll/triton/triton_jagged_bmm.py,sha256=bZIgk-GBdP8lPOoAOiIvO-9IE86B5Ejljmnh6-IuQeA,11785
65
+ fbgemm_gpu/sll/triton/triton_jagged_bmm_jagged_out.py,sha256=hccLxsKoSZKiWid5P_yl-IVdBSXw1Rt0WeiRsjLD2Iw,13864
66
+ fbgemm_gpu/sll/triton/triton_jagged_dense_elementwise_add.py,sha256=FRZ7vqaTIxVWkztr50q94Uic209e2KriLgF-3PQD6QM,1603
67
+ fbgemm_gpu/sll/triton/triton_jagged_dense_elementwise_mul_jagged_out.py,sha256=9R7BOOe8SJiko1PgbiuHlFyPKtGaaCFSlZ1RaEQyICE,4198
68
+ fbgemm_gpu/sll/triton/triton_jagged_dense_flash_attention.py,sha256=qJvMCRUqMOwL_kxYs1fd5QvYdbjaGeoBy9ovNGpjMws,22779
69
+ fbgemm_gpu/sll/triton/triton_jagged_flash_attention_basic.py,sha256=po9Nx4uAGVu_YIZ9CWvrmzSwxDsnDuNAtnk9VR7-Ems,17750
70
+ fbgemm_gpu/sll/triton/triton_jagged_self_substraction_jagged_out.py,sha256=VaOIxQn4Obvna2Co5VNDGILCDfKuYwkhVxK2oCi5mPI,1754
71
+ fbgemm_gpu/sll/triton/triton_jagged_softmax.py,sha256=odN66XGPc5VWmMZ34FRBsodpUtbpEILDpOgPtpCNrY4,14225
72
+ fbgemm_gpu/sll/triton/triton_multi_head_jagged_flash_attention.py,sha256=nEo5I-bba1XlG59qoACGB18OrA1LISs-e7Lasgys1s8,19572
73
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/__init__.py,sha256=kHLPaFr6UcvXDtdDQuF9CP-fvRNdniOORrG5B8O8SmU,6917
74
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_adagrad.py,sha256=V8CXfcyi5cXSP4-EbXGAq8NRXZdibZQSbPoFLHEcmo8,9733
75
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_adam.py,sha256=Bm0R9a6zL6LTEavWsRgkQilPW7aWg3SBOyE-S5AV8B8,12735
76
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_args.py,sha256=k7ZvSHi8fEsZP2GjofNIEFO9mdaQbQxINIhDbPdol0U,2830
77
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_args_ssd.py,sha256=pOhpRdDutSGpOZW5CylR4IIxljYpodizlLUbrO7PoF8,2909
78
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_lamb.py,sha256=GyRwkpONuthj_MG0PEbDpkiTMUpQ6ffg1xo5NgGbpGU,10720
79
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_lars_sgd.py,sha256=xHxf8neshHuE_6ybtAOrVFFMnwxfPQG11iuF0QIItVs,10285
80
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_none.py,sha256=7xOrQArbnUvGR2xMMRZ5gEsxoKRbDXi5ufxd-55b24g,9414
81
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_partial_rowwise_adam.py,sha256=7V7BdQCUZPOp8gmxrQvTfeinULf_uQppdFe7t9POBZ8,10425
82
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_partial_rowwise_lamb.py,sha256=oRjJ302FMr1O9ibFvNtXqn3i-lpmNDh-3JslMstBAxY,10425
83
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad.py,sha256=wQyjKE5xjZNDyNwQmiwuviMrgtEv2QX-MQMDZ8St2_A,10182
84
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad_ssd.py,sha256=YE6RgV8By8FGUxnzduUrjJdNI8j2JOmdEuWcCAikLMk,11523
85
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad_with_counter.py,sha256=LZLuTucNYd2wlzfC4pU6339SmRZJiKIWlYwFDU1VFt4,12172
86
+ fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_sgd.py,sha256=N32H1lUbbWModjDk_Ci8iF8P0hmlSmemmJynGQhuBGI,9195
87
+ fbgemm_gpu/split_embedding_optimizer_codegen/optimizer_args.py,sha256=xWSmk56JgoYfO8eiiK4BP9Brbhixs4tUAMeWp5TPZ30,956
88
+ fbgemm_gpu/split_embedding_optimizer_codegen/split_embedding_optimizer_rowwise_adagrad.py,sha256=bjrbKAypa-FnOIVKH-IUnWP1Jhlu0lk1SopZ0KLFVdo,6623
89
+ fbgemm_gpu/tbe/__init__.py,sha256=fE0IHi1JJpxsNVBNzWNee2thrNXFFRhY94c80RxNSIE,231
90
+ fbgemm_gpu/tbe/bench/__init__.py,sha256=TyUVsIH4p-RtFaXAKppYoaWbf9UTjCTUpnIV7RD_O5E,1653
91
+ fbgemm_gpu/tbe/bench/bench_config.py,sha256=xgtlGLCeZVW6jBYwkKsiQeCslCrWDgJbV2NLLwCRSn4,5452
92
+ fbgemm_gpu/tbe/bench/bench_runs.py,sha256=K4HRUcsX4BWqtrYwinZSXjnjNDFkvpoEdQmv-6rz7Tk,23518
93
+ fbgemm_gpu/tbe/bench/benchmark_click_interface.py,sha256=ofcGsiTUj3_Ml7JSsqg_LcMw3CV-6ypmlRWAUmT_cjc,6941
94
+ fbgemm_gpu/tbe/bench/eeg_cli.py,sha256=B3QOZhtycMDwHMG3dFKnlFuWOqYRCF3RCozEQfrqv8o,3580
95
+ fbgemm_gpu/tbe/bench/embedding_ops_common_config.py,sha256=zdL_ve1Ga6ziU5LjfnzJXOBOIqtCjLlhSrlGfa42H9w,4978
96
+ fbgemm_gpu/tbe/bench/eval_compression.py,sha256=ulFMaNZF2g_vfkXLWZSh02ibotg1zpTz3swVU484mzU,3486
97
+ fbgemm_gpu/tbe/bench/reporter.py,sha256=ZK5RFolUmZEcsEaife270_iOdXAQD5EjTUkuxctnAbY,804
98
+ fbgemm_gpu/tbe/bench/tbe_data_config.py,sha256=d724L4Is3Bo2D5reglgsBs7H6ezLFDrQUbTP5tsnPEQ,8509
99
+ fbgemm_gpu/tbe/bench/tbe_data_config_bench_helper.py,sha256=c-IwLbx04Qbqxzfcn9N4U9Eo9QnmgbBN6HxJYAJwvMw,11311
100
+ fbgemm_gpu/tbe/bench/tbe_data_config_loader.py,sha256=fSdtEAnKu6r56mHMtMJIHo-S6m3vC4cPRyXJKKUevzc,11996
101
+ fbgemm_gpu/tbe/bench/tbe_data_config_param_models.py,sha256=I9dozlJAW_XzuopyJapJ4gmDkLU0YSUz2znugiLZRMg,6203
102
+ fbgemm_gpu/tbe/bench/utils.py,sha256=C0GTTomJO3r9LVfbpzlkudxoA_3QyeMdM-7zM-YOAHA,6716
103
+ fbgemm_gpu/tbe/cache/__init__.py,sha256=lrYwhvqX2eWN0vAPe89HYgMW_O1vccoOcoFHJ9cyM-s,398
104
+ fbgemm_gpu/tbe/cache/kv_embedding_ops_inference.py,sha256=VmG9EennGcq2By8Tj8VkFsJG0oOCGw8EhlPo8-t--Fk,14604
105
+ fbgemm_gpu/tbe/cache/split_embeddings_cache_ops.py,sha256=vZHj7KIe1DoJDy5eft29XtGg6I-tRx60tjKOcTHRAYI,1321
106
+ fbgemm_gpu/tbe/ssd/__init__.py,sha256=wzfMT10cp_dqK2lrebC449hOdexBnizcf_98lA1NyHs,483
107
+ fbgemm_gpu/tbe/ssd/common.py,sha256=zecFfJCcQIwNYbaGoI44Q8rGCskvtmOmc1zxqYHS7Tg,1055
108
+ fbgemm_gpu/tbe/ssd/inference.py,sha256=MwSXP4l2fJUSQJRPu9-bqU08Kg9-0ux8uA5UPSabW3M,22812
109
+ fbgemm_gpu/tbe/ssd/training.py,sha256=2CFA4KmA9IfcpX14K4MlzBuSRPD9h5NM1M7TqepH6vA,212168
110
+ fbgemm_gpu/tbe/ssd/utils/__init__.py,sha256=5DgmR2HA6NtmYh2ddkUgpDsZ6a7hF0DPedA1gMpdh18,250
111
+ fbgemm_gpu/tbe/ssd/utils/partially_materialized_tensor.py,sha256=SFg2-29b-i49LWm-FlaWUkTz2XzXbicYi_AzVj4jKNE,7601
112
+ fbgemm_gpu/tbe/stats/__init__.py,sha256=on29iDtq7cVNh90JR9aeFNG-K9DDoYq0JryzoplL49I,322
113
+ fbgemm_gpu/tbe/stats/bench_params_reporter.py,sha256=_lA4peKXI0GCWsZHJ7IUKlUHU98CA-gVoOc-uhRfcoY,13233
114
+ fbgemm_gpu/tbe/utils/__init__.py,sha256=rlXFm-kTByFZO4SS5C5zMzANRiQmM1NT__eWBayncYg,549
115
+ fbgemm_gpu/tbe/utils/common.py,sha256=KBCyBT-7ShhTRRd1Rs5sEU4g8JggEM7Es6wQ0qhWY-o,1313
116
+ fbgemm_gpu/tbe/utils/offsets.py,sha256=DDWwGaQsVZbhaEZ_fRxxeY8ndLc7IORPZrx61eOqwJc,1904
117
+ fbgemm_gpu/tbe/utils/quantize.py,sha256=EdYh9FS_kMsvCWPuvNms4uSE9de_3cQNo_DCScGG3zI,9166
118
+ fbgemm_gpu/tbe/utils/requests.py,sha256=_lxGVt2J0tEmG7aXv24BMrvfdK6HuvMPlPZHWsF_EDI,18038
119
+ fbgemm_gpu/triton/__init__.py,sha256=kPn_Ye6J9DAzWtqi76KYGwfKSqw0IhqG3Bir5aUpkWM,658
120
+ fbgemm_gpu/triton/common.py,sha256=tsK56Dom_XSb5kXuoN0KnGAWlC5HWV7Ook--a59UHdI,2130
121
+ fbgemm_gpu/triton/quantize.py,sha256=I0pxyfIx04zyq55x4Pvj-28Cb2ZeF-SGtFhAymFagkg,27073
122
+ fbgemm_gpu/triton/quantize_ref.py,sha256=q4RBmFaqPVPELU52lbSgB0n26Aun7apeK7bRF2MWS80,11553
123
+ fbgemm_gpu/triton/jagged/__init__.py,sha256=om0yhjuzKuE1UQakFMWHsXN4WNb8mvNkZtYofQ8hdn4,246
124
+ fbgemm_gpu/triton/jagged/triton_jagged_tensor_ops.py,sha256=F2eQWjkWMR5RWQ48oIr-8OU_CRZyLazDpT7DFrDWS6g,29871
125
+ fbgemm_gpu/utils/__init__.py,sha256=JQQNdcTTaEU6ptK-OW-ZQBwTFxEZZpWOtBXWwEZm39o,354
126
+ fbgemm_gpu/utils/filestore.py,sha256=oVtbKGaPQki1JgbJCkrkElukOFVyxntQpSC0lYBKgho,6455
127
+ fbgemm_gpu/utils/loader.py,sha256=1hCEhNvkflniH46fGcrguLeP1z-6uyOu2QFwqKU5CIM,990
128
+ fbgemm_gpu/utils/torch_library.py,sha256=ywsAHjbuwesj50LjEu99WkAH17FlaVgePZ9OmFg6YE4,4193
129
+ fbgemm_gpu/utils/writeback_util.py,sha256=PyVbHp1EuF-GKrJv_CTP6B50Z0oBblXKucf7Rhd6KKY,4614
130
+ list_versions/__init__.py,sha256=UmTeqCk-UJWFtlZQWvZao3xvui2w9E3X_JdOXVjRaNw,315
131
+ list_versions/cli_run.py,sha256=BCRaJvjVFBFmD5WPdjC_yJwlLv1w_TYOe3eYlf_9ZMo,4506
132
+ fbgemm_gpu_nightly_cpu-2026.1.29.dist-info/METADATA,sha256=sqUYIVBwodRVxysq3jEToUNFX12vtC4tZenZnKnynjo,2654
133
+ fbgemm_gpu_nightly_cpu-2026.1.29.dist-info/WHEEL,sha256=IaW-egZU3n4QvS-XsoO31KhIl6_BixcQGgBPEoTC6GI,109
134
+ fbgemm_gpu_nightly_cpu-2026.1.29.dist-info/top_level.txt,sha256=_2s1Aa08r_eDn0JP4FjOhzK09Q8bVlEI7q8pMep51UY,25
135
+ fbgemm_gpu_nightly_cpu-2026.1.29.dist-info/RECORD,,
@@ -0,0 +1,2 @@
1
+ fbgemm_gpu
2
+ list_versions
@@ -1,4 +1,3 @@
1
-
2
1
  #!/usr/bin/env python3
3
2
  # Copyright (c) Meta Platforms, Inc. and affiliates.
4
3
  # All rights reserved.
@@ -6,5 +5,8 @@
6
5
  # This source code is licensed under the BSD-style license found in the
7
6
  # LICENSE file in the root directory of this source tree.
8
7
 
9
- __version__: str = "2025.3.27"
10
- __variant__: str = "cpu"
8
+ # pyre-strict
9
+
10
+ from .cli_run import CLI, CLIOutput
11
+
12
+ __all__ = ["CLI", "CLIOutput"]
@@ -0,0 +1,161 @@
1
+ #!/usr/bin/env python3
2
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
3
+ # All rights reserved.
4
+ #
5
+ # This source code is licensed under the BSD-style license found in the
6
+ # LICENSE file in the root directory of this source tree.
7
+
8
+ # pyre-strict
9
+
10
+ import logging
11
+ import subprocess
12
+ from datetime import datetime
13
+ from typing import Union
14
+
15
+ import click
16
+ import pandas as pd
17
+ import torch
18
+
19
+
20
+ class CLIOutput:
21
+ def __init__(
22
+ self,
23
+ cli: str = "",
24
+ stdout: str = "",
25
+ stderr: str = "",
26
+ returncode: int = 0,
27
+ timestamp: str = "2025-01-01T20:00:00.00000",
28
+ visible: bool = True,
29
+ ) -> None:
30
+ self._cli = cli
31
+ self._stdout = stdout
32
+ self._stderr = stderr
33
+ self._returncode = returncode
34
+ self._timestamp = timestamp
35
+ self._visible = visible
36
+
37
+ def to_dict(self) -> dict[str, Union[int, str]]:
38
+ return {
39
+ "cli": self._cli,
40
+ "stdout": self._stdout,
41
+ "stderr": self._stderr,
42
+ "returncode": self._returncode,
43
+ "timestamp": self._timestamp,
44
+ "visible": self._visible,
45
+ }
46
+
47
+
48
+ class CLI:
49
+ def __init__(self) -> None:
50
+ pd.options.display.max_rows
51
+ pd.set_option("display.max_colwidth", None)
52
+ self._cli_outputs: list[CLIOutput] = [
53
+ CLIOutput(
54
+ cli="python –c “import torch; print(torch.__version__)”",
55
+ stdout="{}".format(torch.__version__),
56
+ stderr="",
57
+ returncode=0,
58
+ timestamp=datetime.now().isoformat(),
59
+ visible=True,
60
+ )
61
+ ]
62
+
63
+ def run(
64
+ self,
65
+ cli: Union[str, list[str]],
66
+ visible: bool = True,
67
+ input: str = "",
68
+ capture_output: bool = True,
69
+ ) -> CLIOutput:
70
+ if isinstance(cli, str):
71
+ cli = cli.split()
72
+ result = CLIOutput()
73
+ try:
74
+ completed = subprocess.run(
75
+ cli, text=True, check=False, capture_output=capture_output, input=input
76
+ )
77
+ result = CLIOutput(
78
+ cli=" ".join(cli),
79
+ stdout=completed.stdout,
80
+ stderr=completed.stderr,
81
+ returncode=completed.returncode,
82
+ timestamp=datetime.now().isoformat(),
83
+ visible=visible,
84
+ )
85
+ if visible:
86
+ self._cli_outputs.append(result)
87
+ except Exception as e:
88
+ logging.error(f'For cli {" ".join(cli)} we got exception {e}')
89
+ result = CLIOutput(
90
+ cli=" ".join(cli),
91
+ stdout="",
92
+ stderr=str(e),
93
+ returncode=-1,
94
+ visible=visible,
95
+ timestamp=datetime.now().isoformat(),
96
+ )
97
+ if visible:
98
+ self._cli_outputs.append(result)
99
+ return result
100
+
101
+ def run_piped(self, clis: list[str]) -> None:
102
+ the_input = ""
103
+ for cli in clis[:-1]:
104
+ result = self.run(
105
+ cli=cli, visible=False, input=the_input, capture_output=True
106
+ )
107
+ the_input = result._stdout
108
+ self.run(cli=clis[-1], visible=True, input=the_input, capture_output=True)
109
+
110
+ def to_dataframe(self) -> pd.DataFrame:
111
+ return pd.DataFrame([output.to_dict() for output in self._cli_outputs])
112
+
113
+ def save(self, filename: str, format: str = "csv") -> None:
114
+ df = self.to_dataframe()
115
+ if format == "csv":
116
+ df.to_csv(filename, index=False)
117
+ elif format == "json":
118
+ df.to_json(filename, orient="records", lines=True)
119
+ else:
120
+ raise ValueError(f"Invalid format {format} : must be one of 'csv', 'json'")
121
+
122
+
123
+ @click.command()
124
+ @click.option("--json", default="")
125
+ @click.option("--csv", default="")
126
+ def cli_run(
127
+ json: str,
128
+ csv: str,
129
+ ) -> None:
130
+ cli = CLI()
131
+
132
+ the_rpm = "rpm -qa"
133
+ the_grep1 = "grep -E ^amdgpu-(dkms|kmod)"
134
+ the_grep2 = "grep -v firmware"
135
+ the_sed1 = "sed -E s/^[^-]-[^-]-//"
136
+ the_sed2 = "sed -E s/.[^.].[^.]$//"
137
+ cli.run_piped([the_rpm, the_grep1, the_grep2, the_sed1, the_sed2])
138
+
139
+ cli.run("uname -r")
140
+
141
+ cli.run("fw-util all --version")
142
+
143
+ cli.run("amd-smi firmware")
144
+ cli.run("amd-smi version")
145
+ cli.run("amd-smi static")
146
+
147
+ if len(csv):
148
+ cli.save(csv)
149
+
150
+ if len(json):
151
+ cli.save(json, format="json")
152
+
153
+ print(cli.to_dataframe())
154
+
155
+
156
+ def main() -> None:
157
+ cli_run()
158
+
159
+
160
+ if __name__ == "__main__":
161
+ main()
@@ -1,126 +0,0 @@
1
- fbgemm_gpu/__init__.py,sha256=GCtlMprObrPh1r0Ti6rZpgTRAzvcKGe1P7SVEI9turg,3470
2
- fbgemm_gpu/asmjit.so,sha256=sNvOBVJgObazT6YIDn25fCExzL8kqneLXI64og9l59E,526272
3
- fbgemm_gpu/batched_unary_embeddings_ops.py,sha256=u7LfseNeM5gGFQGLAMVO7h2QkFWEOL3ezV5RuhbZn4M,2928
4
- fbgemm_gpu/enums.py,sha256=GVuzF5cFTLzttkvlH1SdcGrxrppMhDSbQj_Vm_4zmEo,789
5
- fbgemm_gpu/fbgemm.so,sha256=xau1bMRrCscjMtn6TUG3iXpn3qNbmCz1Nkrwbedji9I,1312776
6
- fbgemm_gpu/fbgemm_gpu_config.so,sha256=BjNblyddeIRqItZ4_Vc_qxwfpCg2XMQhqsuLVmXWvSg,67528
7
- fbgemm_gpu/fbgemm_gpu_embedding_inplace_ops.so,sha256=m_2AtvPopBaD2Kt0kLbxQdby8S0QQoMg7dh3mrnyIlo,67792
8
- fbgemm_gpu/fbgemm_gpu_py.so,sha256=FIzW9kcLbKgdI8c7f1EIpUJy7jnJfcW8TXbORPFSxMg,3876448
9
- fbgemm_gpu/fbgemm_gpu_sparse_async_cumsum.so,sha256=SwYj8eSNQ3JFYNuhCvuhKQAwkPyJ0acn5-0mCPn3aLM,67592
10
- fbgemm_gpu/fbgemm_gpu_tbe_cache.so,sha256=OkQ1XiEOse5q_FDZqduNF_m774uitf3jbawhpr7ArZ8,134080
11
- fbgemm_gpu/fbgemm_gpu_tbe_common.so,sha256=79Qm9awbj5N3EeScbtexPGuIChCdGTlHnNP3elFh6Ck,330224
12
- fbgemm_gpu/fbgemm_gpu_tbe_index_select.so,sha256=YjUX-Qw4MKDaMy6p-TZy9TmrgQvHOEKW3X2M5ZIodqE,330016
13
- fbgemm_gpu/fbgemm_gpu_tbe_inference.so,sha256=zwGkIp8bBVVKejKUV5R-M42cDT01m0Qnmoh_vUMF_s4,593248
14
- fbgemm_gpu/fbgemm_gpu_tbe_optimizers.so,sha256=nhA1KSJ8_uO8kp36D2agrrhkUKiLmCxUJnv856OKS3g,67128
15
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward.so,sha256=u2GEFoNb4X9BZuFskN3sIqMdwFLLBd1UcRnX-alSW4o,862352
16
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward_dense.so,sha256=6r6kBgfZyasFAwABIDe5kuty8jQjUiIFkRY26GVPUxw,67128
17
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward_gwd.so,sha256=LmQYTbhWU8x-4CUvDd_mEQA3dc_cyl8rDMMdK6un1gs,67128
18
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward_pt2.so,sha256=zh-hRUJAS2NSEXP5geFn0er_mtIUCGJ5sr7UOEFeZk8,3089264
19
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward_split_host.so,sha256=hOgYJNo98a-4pqUutxL5GD6PtFOomM2x55-Z5DvFrgI,67128
20
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward_vbe.so,sha256=8QNJQ9X67gnTlVDhc1jbZPlggQvwKU6aO-n5TfnXVOk,67128
21
- fbgemm_gpu/fbgemm_gpu_tbe_training_forward.so,sha256=y7fGiGiRhYG0z4_YKEK4m-pYmi2xU9c1iPs74W6AerY,199832
22
- fbgemm_gpu/fbgemm_gpu_tbe_utils.so,sha256=YYY1k8J-QI0LCfkLtbXn8k_iLAXi9rtRXmQKfeDDfa8,133312
23
- fbgemm_gpu/metrics.py,sha256=TsurFLJf0nJvPDN7urWb4LMQlf5RgdWPTTTDO7S4wtI,5663
24
- fbgemm_gpu/permute_pooled_embedding_modules.py,sha256=kjWuWmQY8e2kMRwIPTzjGjyjV4syKPrphtHdsQTAjWM,5136
25
- fbgemm_gpu/permute_pooled_embedding_modules_split.py,sha256=cUrEbRIvLFW_3Zmh07QkN4S1Cfvvge6TYO1VXBFCpz8,2752
26
- fbgemm_gpu/quantize_comm.py,sha256=YUzk8F1MZckbkseEoDJ4rOKiglGCGtrPdR1IKGD6Mk4,11177
27
- fbgemm_gpu/quantize_utils.py,sha256=hb8G_1xzRwYmwFp8VJrmoaolNxCwqcFwkwnyStk1C0w,7394
28
- fbgemm_gpu/runtime_monitor.py,sha256=HM_0cxMO7uuAq8sCiv2lmGgp1jKGzba2qhuUcGyRMog,7425
29
- fbgemm_gpu/sparse_ops.py,sha256=jYKnz-pwWFtFUxMxqwxI83EsQ_-yKSbY5m6ZuzyYuIU,46186
30
- fbgemm_gpu/split_embedding_configs.py,sha256=s177mJqR816qvsr_9dTSCDrwSMQ85ub0Yh60QyiVmaM,6392
31
- fbgemm_gpu/split_embedding_inference_converter.py,sha256=ilVVowkTiY0WDpOYorj917Tqsez4KWNBdTXuz2bWbp8,7063
32
- fbgemm_gpu/split_embedding_optimizer_ops.py,sha256=wXuGazClBMk62yL_r9udUIKaPgQP7SlkSb5ugB75wrQ,711
33
- fbgemm_gpu/split_embedding_utils.py,sha256=Gb40ZKeATxIKEKI3aVQMgDDBanNpKMc53Z43mnzdR_I,851
34
- fbgemm_gpu/split_table_batched_embeddings_ops.py,sha256=_MIp6uHYHLn4GxGdrGsfddfSsZ2Z9mjsYIrih3ncI1I,2339
35
- fbgemm_gpu/split_table_batched_embeddings_ops_common.py,sha256=wscggjbhy3IOQ5WuXXnQtml8V2FrI33ZDqD1qoTMojM,6324
36
- fbgemm_gpu/split_table_batched_embeddings_ops_inference.py,sha256=4eokBcwOH_UU4JVbJaZdoaJr1qEcPwvVVL6_LVZ7bjI,81246
37
- fbgemm_gpu/split_table_batched_embeddings_ops_training.py,sha256=HWrbcgEYf3jxMb6-PbNCU4KyeQ36MJHCQuYpPbIVf7A,158298
38
- fbgemm_gpu/split_table_batched_embeddings_ops_training_common.py,sha256=ktC10-nakOBpcmJNCOGQsxuBCP8XTwXJ2WeEgIg91tc,5455
39
- fbgemm_gpu/ssd_split_table_batched_embeddings_ops.py,sha256=7qGkO8FARku38mFYl4Bc4qL8dS1wrfyorS9l1m5ZAVA,718
40
- fbgemm_gpu/tbe_input_multiplexer.py,sha256=Y7NjF4JpQHpAu6aKWgy2UX0QmtIt4fWHWZwnds5PiQg,2611
41
- fbgemm_gpu/uvm.py,sha256=-cZunsuvnAKUEQptIwdYVar_3hUE99FbQUsyfBVeXPE,925
42
- fbgemm_gpu/config/__init__.py,sha256=yN0KAneCICgF2BTfOYGsd0qU1PvZX_6msC6YHHZKLMg,292
43
- fbgemm_gpu/config/feature_list.py,sha256=kBXRV_3Hc-eC2zy5YGo_viS0t7awojQzmkHE7AgATuY,2267
44
- fbgemm_gpu/docs/__init__.py,sha256=DR6hMSQrsZALfH2AnuJQ4Zq2CfBUUhMN8YjD6APjiAE,523
45
- fbgemm_gpu/docs/common.py,sha256=8ipXTwVb222X-aZ71O6n8fhxHCHPNhJEHMFiO7epcIs,273
46
- fbgemm_gpu/docs/examples.py,sha256=ZMN_6sL74LH_hrp2bF_hmg8gi29GhcgvwV3kCMjxkoE,2377
47
- fbgemm_gpu/docs/jagged_tensor_ops.py,sha256=Bsx-ZxvvdMv5CaldSvuw9GPR-HRcLbRR2IEXCOCm9r0,7381
48
- fbgemm_gpu/docs/merge_pooled_embedding_ops.py,sha256=oJLgSgZQmhsyGLbTmZTxNgQrk65_3E8xSJaWSj_Jbo8,1102
49
- fbgemm_gpu/docs/permute_pooled_embedding_ops.py,sha256=tZUqLVXlk5O6VAKKDA-OEMx2fCu5QPOOeoAPZA9_nLY,4454
50
- fbgemm_gpu/docs/quantize_ops.py,sha256=xTtOaVK1P02ymreE_i21YiyYDZCqhoZY9eWp_mEIRlo,1297
51
- fbgemm_gpu/docs/sparse_ops.py,sha256=cw-ioy9wg08yL4wYtbRSx4Ywez6y-SurQeBKGZ2zVBk,20272
52
- fbgemm_gpu/docs/version.py,sha256=PhV5xGZPXwtJzGL3r2hAGl9qlyGEO9kp2CjCQVGzXPw,289
53
- fbgemm_gpu/quantize/__init__.py,sha256=pftciXHE7csekDFkl7Ui1AWglVMMnSrOO04mREnUdb0,921
54
- fbgemm_gpu/quantize/quantize_ops.py,sha256=25AIOv9n2UoxamMUaI6EK1Ur4gSHxbZIReHBtgOjjCs,2228
55
- fbgemm_gpu/sll/__init__.py,sha256=rgXh35-OFUE54E9gGBq3NGxouGvgMv2ccY2bWUTxONY,4191
56
- fbgemm_gpu/sll/cpu/__init__.py,sha256=glsukNpXtf47VRIdBktILD-4CmVcf4621SGB55lT_ho,2692
57
- fbgemm_gpu/sll/cpu/cpu_sll.py,sha256=hqUDEIv_cBWZDRbOosXUuWH07GZA_MrihnENcfwMtw4,27877
58
- fbgemm_gpu/sll/meta/__init__.py,sha256=2sMcD67XGsweBZ-UV2AEJmM4ELPsHeRAYED6kqfgAd4,1077
59
- fbgemm_gpu/sll/meta/meta_sll.py,sha256=Jk14EOW9VPFwawD7Bwky0R0A5rmbcLWMo52oH8J6Koc,8305
60
- fbgemm_gpu/sll/triton/__init__.py,sha256=dW_cEW0R8635sKLozsL88SP0Cch5QnBGvfnAmoqWMic,4109
61
- fbgemm_gpu/sll/triton/common.py,sha256=hISlX4Y-7FtGof-Xx4_B8-2vlF27F9t4p2qyLMUnJ8A,798
62
- fbgemm_gpu/sll/triton/triton_dense_jagged_cat_jagged_out.py,sha256=J9qOqjNJ72LUBqs-pGI9wrFzzzBpsZ5fzYjgfKc2YhY,1885
63
- fbgemm_gpu/sll/triton/triton_jagged2_to_padded_dense.py,sha256=UY2l3ONwAl1tGuUqD0eQ_kKfDrnrz3A1oj7hdAoVcCI,5950
64
- fbgemm_gpu/sll/triton/triton_jagged_bmm.py,sha256=QFhaIQc8g-TRHr7wjm-Wd-atNJS1fDDkImHXXB3v-gU,11789
65
- fbgemm_gpu/sll/triton/triton_jagged_bmm_jagged_out.py,sha256=hccLxsKoSZKiWid5P_yl-IVdBSXw1Rt0WeiRsjLD2Iw,13864
66
- fbgemm_gpu/sll/triton/triton_jagged_dense_elementwise_add.py,sha256=_0hke_aaAdKQJpGUYX20NLss1_cXDIKxqblX4QQb7Io,1592
67
- fbgemm_gpu/sll/triton/triton_jagged_dense_elementwise_mul_jagged_out.py,sha256=9R7BOOe8SJiko1PgbiuHlFyPKtGaaCFSlZ1RaEQyICE,4198
68
- fbgemm_gpu/sll/triton/triton_jagged_dense_flash_attention.py,sha256=2mxCdd2vY3CZvDCLPveuv0QstIiHKSA1-5JF1DQPTh0,22808
69
- fbgemm_gpu/sll/triton/triton_jagged_flash_attention_basic.py,sha256=6zmeA68G3gygt4uswVSqHDylSDADdLKKI9Deifxd3Ok,17775
70
- fbgemm_gpu/sll/triton/triton_jagged_self_substraction_jagged_out.py,sha256=VaOIxQn4Obvna2Co5VNDGILCDfKuYwkhVxK2oCi5mPI,1754
71
- fbgemm_gpu/sll/triton/triton_jagged_softmax.py,sha256=odN66XGPc5VWmMZ34FRBsodpUtbpEILDpOgPtpCNrY4,14225
72
- fbgemm_gpu/sll/triton/triton_multi_head_jagged_flash_attention.py,sha256=D9XVxiBYAauwuAyRnfsJBqdjaPCAhOM_z7IzzeIsx4I,19597
73
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/__init__.py,sha256=HHj5YVaerTi6a2UTNabmTYtFTJUHNHYYuQy0FCJ1RXM,5941
74
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_adagrad.py,sha256=zObEoIXvn2U7JoFz3I4XR7Vuj5tIWVBxQBJNiCjdNyQ,7075
75
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_adam.py,sha256=RbWplF88Omjen-MulJtBRfytlVp1KAD29MGc3VHJvPY,4965
76
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_args.py,sha256=IoGqSXM-wYHUbnMx0ZdmIpULd3r4WbzPoytwSqZtnZo,4089
77
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_args_ssd.py,sha256=N56-cQ_Vglm9vzFod11QdhD0foV6u_9yrKqKD6BDC_0,4171
78
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_lamb.py,sha256=rF8CE4twaPpLXtuBmotagr2Vlh8lKwSYoiFttGTwZ14,4075
79
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_lars_sgd.py,sha256=CxnR5bZx6gEsrLzWOlSzolh6E0e36EVnOyAjjgTsh3s,3862
80
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_none.py,sha256=ipTTah_vldJATL-XyY8380TKH45NYwXKQzDdknkId4Q,3078
81
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_partial_rowwise_adam.py,sha256=P4zPCV5u7j6mFVgOgaLreZsoC3yk2N-G2RKN5CAjNXY,3780
82
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_partial_rowwise_lamb.py,sha256=-1TeiX1_RhrWl3L0my_rdiAx_yUcORIoEBKU52m5AZA,3780
83
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad.py,sha256=77IFekylnxJSlPd9y6awMzT30mA2NLU6o-1CL80fLa8,7407
84
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad_ssd.py,sha256=2h-qxSeeh3xlLP2uXWmvoojGsJpFz7xsOKBBZeQYw74,4619
85
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad_with_counter.py,sha256=shEOjpeVJGjqCQ2vNaj9WbpyABqCKXm-UKB0gtSSMg4,4478
86
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_sgd.py,sha256=ywKC5OmsvfsoG7S_GUWuHPxM4_tD9_wIeKm6MozG1bk,6657
87
- fbgemm_gpu/split_embedding_optimizer_codegen/optimizer_args.py,sha256=l85_PY8cyf4cF1dRe3ypXzW1c6QZxmp8tNrfa4f7ZkM,892
88
- fbgemm_gpu/split_embedding_optimizer_codegen/split_embedding_optimizer_rowwise_adagrad.py,sha256=DXOXGXcy2ObtF2jWFHL8m0EvQCWrXQhyllqByWjMUVA,6534
89
- fbgemm_gpu/tbe/__init__.py,sha256=fE0IHi1JJpxsNVBNzWNee2thrNXFFRhY94c80RxNSIE,231
90
- fbgemm_gpu/tbe/bench/__init__.py,sha256=tNKHGiUjqwAznB5n5H0zHLGWJCnVmTqSd8kvtrI7Y1Y,1380
91
- fbgemm_gpu/tbe/bench/bench_config.py,sha256=GYxVotMi-JK7AEm96sOfB2uOlmRtXpOTr-tkPEB1Q48,4241
92
- fbgemm_gpu/tbe/bench/bench_runs.py,sha256=rA6cbzziH7vV6JzynqExhmhk-gdESCa0inGeSQGAonc,13697
93
- fbgemm_gpu/tbe/bench/embedding_ops_common_config.py,sha256=yBk0zaZJ5b3ffPG91SoqD088shrLYdyCFGfn6DsXK9A,5021
94
- fbgemm_gpu/tbe/bench/eval_compression.py,sha256=bINVERk42VJDSdenQHKWApmRMrW8rhkevOgE0hDR-S8,3499
95
- fbgemm_gpu/tbe/bench/reporter.py,sha256=ZK5RFolUmZEcsEaife270_iOdXAQD5EjTUkuxctnAbY,804
96
- fbgemm_gpu/tbe/bench/tbe_data_config.py,sha256=nynSj2YttH9LMaGPkaaVMTgoMWNBHzlCUhedCPbPXT0,9406
97
- fbgemm_gpu/tbe/bench/tbe_data_config_loader.py,sha256=eSzD7JP4oIo3LP6nVhqwfxfeDARSy-TS48ue-5duodE,7519
98
- fbgemm_gpu/tbe/bench/tbe_data_config_param_models.py,sha256=Ab9SbQG_hjZphi-q-HLne9-8SVi27tPU_Qf_b6QsNfE,5796
99
- fbgemm_gpu/tbe/bench/utils.py,sha256=cq_6FJHlgZ5femAK6XKpj7nJ9jc03qXI16N1ht1CcLg,1721
100
- fbgemm_gpu/tbe/cache/__init__.py,sha256=oM-g5nq0EXZgO79C6DhAl_Om9FTPC-WiaqclQCG3HTk,323
101
- fbgemm_gpu/tbe/cache/split_embeddings_cache_ops.py,sha256=8pfRDv-CPS_weM7EVYK7iIHejSQlTqpTRSo5jNrka1w,1406
102
- fbgemm_gpu/tbe/ssd/__init__.py,sha256=wzfMT10cp_dqK2lrebC449hOdexBnizcf_98lA1NyHs,483
103
- fbgemm_gpu/tbe/ssd/common.py,sha256=qDKTU9TtuAMplMFvR9c6iUhdYW6nUBcg4MITgkpGJEw,482
104
- fbgemm_gpu/tbe/ssd/inference.py,sha256=DTjwj3f6JaUMcecWoRNkZpRgXDJ-eE3grtixYwKb5DI,22829
105
- fbgemm_gpu/tbe/ssd/training.py,sha256=0ACGeLH8HNOiryQ2mj9i7zGO1jHUv68aLyfS5a0UrJk,87525
106
- fbgemm_gpu/tbe/ssd/utils/__init__.py,sha256=5DgmR2HA6NtmYh2ddkUgpDsZ6a7hF0DPedA1gMpdh18,250
107
- fbgemm_gpu/tbe/ssd/utils/partially_materialized_tensor.py,sha256=Uy59mpnH5IGbeFvQawL1CQ_ClFS_HkA5HseT1fAZcZ0,6377
108
- fbgemm_gpu/tbe/utils/__init__.py,sha256=rlXFm-kTByFZO4SS5C5zMzANRiQmM1NT__eWBayncYg,549
109
- fbgemm_gpu/tbe/utils/common.py,sha256=KBCyBT-7ShhTRRd1Rs5sEU4g8JggEM7Es6wQ0qhWY-o,1313
110
- fbgemm_gpu/tbe/utils/offsets.py,sha256=bs08kDiQ54oucZl6rmPLDs-bN6m1EMa1Wju06mCKZBY,1917
111
- fbgemm_gpu/tbe/utils/quantize.py,sha256=byjmzGpUjXD_UVAiBKyszmWlzYLkQxq5HBs6hBOuHZo,9185
112
- fbgemm_gpu/tbe/utils/requests.py,sha256=nn5G1LGcmSMkLSl0BH-48-UOjY-qNo1v8XPviDX9nCY,16973
113
- fbgemm_gpu/triton/__init__.py,sha256=kPn_Ye6J9DAzWtqi76KYGwfKSqw0IhqG3Bir5aUpkWM,658
114
- fbgemm_gpu/triton/common.py,sha256=wnkLd2a8fKpefymLL-LjNKEL4hDVSxFiF5g3aF8mzsw,2131
115
- fbgemm_gpu/triton/quantize.py,sha256=K5pqBQqs4YsD5m5TibZCbkd0E4Si0i_xcpIeF1B6jA0,26815
116
- fbgemm_gpu/triton/quantize_ref.py,sha256=q4RBmFaqPVPELU52lbSgB0n26Aun7apeK7bRF2MWS80,11553
117
- fbgemm_gpu/triton/jagged/__init__.py,sha256=om0yhjuzKuE1UQakFMWHsXN4WNb8mvNkZtYofQ8hdn4,246
118
- fbgemm_gpu/triton/jagged/triton_jagged_tensor_ops.py,sha256=AIC1G6_QBQtMVTyOyEV4ZKJyDzu36UI_9HDgWmZIRaA,29884
119
- fbgemm_gpu/utils/__init__.py,sha256=JQQNdcTTaEU6ptK-OW-ZQBwTFxEZZpWOtBXWwEZm39o,354
120
- fbgemm_gpu/utils/filestore.py,sha256=LB0GtToszUGHEPHKge8F48T6JUxQCev33mtOOgD5bvI,4775
121
- fbgemm_gpu/utils/loader.py,sha256=1hCEhNvkflniH46fGcrguLeP1z-6uyOu2QFwqKU5CIM,990
122
- fbgemm_gpu/utils/torch_library.py,sha256=dQcHv1qgpu5QYlJjxjd6oeHjtxnmmXzx3PL6vjCmxL4,4199
123
- fbgemm_gpu_nightly_cpu-2025.3.27.dist-info/METADATA,sha256=dQMCTDqjQJl54bm6w7bupnXez7peZypjkVDT4xVTvE4,2793
124
- fbgemm_gpu_nightly_cpu-2025.3.27.dist-info/WHEEL,sha256=IaW-egZU3n4QvS-XsoO31KhIl6_BixcQGgBPEoTC6GI,109
125
- fbgemm_gpu_nightly_cpu-2025.3.27.dist-info/top_level.txt,sha256=2tlbTWLkPjhqvLF_6BbqKzkcPluSE-oPRVjI8axK76I,11
126
- fbgemm_gpu_nightly_cpu-2025.3.27.dist-info/RECORD,,
@@ -1 +0,0 @@
1
- fbgemm_gpu