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.
- fbgemm_gpu/__init__.py +118 -23
- fbgemm_gpu/asmjit.so +0 -0
- fbgemm_gpu/batched_unary_embeddings_ops.py +3 -3
- fbgemm_gpu/config/feature_list.py +7 -1
- fbgemm_gpu/docs/jagged_tensor_ops.py +0 -1
- fbgemm_gpu/docs/sparse_ops.py +142 -1
- fbgemm_gpu/docs/target.default.json.py +6 -0
- fbgemm_gpu/enums.py +3 -4
- fbgemm_gpu/fbgemm.so +0 -0
- fbgemm_gpu/fbgemm_gpu_config.so +0 -0
- fbgemm_gpu/fbgemm_gpu_embedding_inplace_ops.so +0 -0
- fbgemm_gpu/fbgemm_gpu_py.so +0 -0
- fbgemm_gpu/fbgemm_gpu_sparse_async_cumsum.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_cache.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_common.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_index_select.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_inference.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_optimizers.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward_dense.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward_gwd.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward_pt2.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward_split_host.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_training_backward_vbe.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_training_forward.so +0 -0
- fbgemm_gpu/fbgemm_gpu_tbe_utils.so +0 -0
- fbgemm_gpu/permute_pooled_embedding_modules.py +5 -4
- fbgemm_gpu/permute_pooled_embedding_modules_split.py +4 -4
- fbgemm_gpu/quantize/__init__.py +2 -0
- fbgemm_gpu/quantize/quantize_ops.py +1 -0
- fbgemm_gpu/quantize_comm.py +29 -12
- fbgemm_gpu/quantize_utils.py +88 -8
- fbgemm_gpu/runtime_monitor.py +9 -5
- fbgemm_gpu/sll/__init__.py +3 -0
- fbgemm_gpu/sll/cpu/cpu_sll.py +8 -8
- fbgemm_gpu/sll/triton/__init__.py +0 -10
- fbgemm_gpu/sll/triton/triton_jagged2_to_padded_dense.py +2 -3
- fbgemm_gpu/sll/triton/triton_jagged_bmm.py +2 -2
- fbgemm_gpu/sll/triton/triton_jagged_dense_elementwise_add.py +1 -0
- fbgemm_gpu/sll/triton/triton_jagged_dense_flash_attention.py +5 -6
- fbgemm_gpu/sll/triton/triton_jagged_flash_attention_basic.py +1 -2
- fbgemm_gpu/sll/triton/triton_multi_head_jagged_flash_attention.py +1 -2
- fbgemm_gpu/sparse_ops.py +244 -76
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/__init__.py +26 -0
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_adagrad.py +208 -105
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_adam.py +261 -53
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_args.py +9 -58
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_args_ssd.py +10 -59
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_lamb.py +225 -41
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_lars_sgd.py +211 -36
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_none.py +195 -26
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_partial_rowwise_adam.py +225 -41
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_partial_rowwise_lamb.py +225 -41
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad.py +216 -111
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad_ssd.py +221 -37
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_rowwise_adagrad_with_counter.py +259 -53
- fbgemm_gpu/split_embedding_codegen_lookup_invokers/lookup_sgd.py +192 -96
- fbgemm_gpu/split_embedding_configs.py +287 -3
- fbgemm_gpu/split_embedding_inference_converter.py +7 -6
- fbgemm_gpu/split_embedding_optimizer_codegen/optimizer_args.py +2 -0
- fbgemm_gpu/split_embedding_optimizer_codegen/split_embedding_optimizer_rowwise_adagrad.py +2 -0
- fbgemm_gpu/split_table_batched_embeddings_ops_common.py +275 -9
- fbgemm_gpu/split_table_batched_embeddings_ops_inference.py +44 -37
- fbgemm_gpu/split_table_batched_embeddings_ops_training.py +900 -126
- fbgemm_gpu/split_table_batched_embeddings_ops_training_common.py +44 -1
- fbgemm_gpu/ssd_split_table_batched_embeddings_ops.py +0 -1
- fbgemm_gpu/tbe/bench/__init__.py +13 -2
- fbgemm_gpu/tbe/bench/bench_config.py +37 -9
- fbgemm_gpu/tbe/bench/bench_runs.py +301 -12
- fbgemm_gpu/tbe/bench/benchmark_click_interface.py +189 -0
- fbgemm_gpu/tbe/bench/eeg_cli.py +138 -0
- fbgemm_gpu/tbe/bench/embedding_ops_common_config.py +4 -5
- fbgemm_gpu/tbe/bench/eval_compression.py +3 -3
- fbgemm_gpu/tbe/bench/tbe_data_config.py +116 -198
- fbgemm_gpu/tbe/bench/tbe_data_config_bench_helper.py +332 -0
- fbgemm_gpu/tbe/bench/tbe_data_config_loader.py +158 -32
- fbgemm_gpu/tbe/bench/tbe_data_config_param_models.py +16 -8
- fbgemm_gpu/tbe/bench/utils.py +129 -5
- fbgemm_gpu/tbe/cache/__init__.py +1 -0
- fbgemm_gpu/tbe/cache/kv_embedding_ops_inference.py +385 -0
- fbgemm_gpu/tbe/cache/split_embeddings_cache_ops.py +4 -5
- fbgemm_gpu/tbe/ssd/common.py +27 -0
- fbgemm_gpu/tbe/ssd/inference.py +15 -15
- fbgemm_gpu/tbe/ssd/training.py +2930 -195
- fbgemm_gpu/tbe/ssd/utils/partially_materialized_tensor.py +34 -3
- fbgemm_gpu/tbe/stats/__init__.py +10 -0
- fbgemm_gpu/tbe/stats/bench_params_reporter.py +349 -0
- fbgemm_gpu/tbe/utils/offsets.py +6 -6
- fbgemm_gpu/tbe/utils/quantize.py +8 -8
- fbgemm_gpu/tbe/utils/requests.py +53 -28
- fbgemm_gpu/tbe_input_multiplexer.py +16 -7
- fbgemm_gpu/triton/common.py +0 -1
- fbgemm_gpu/triton/jagged/triton_jagged_tensor_ops.py +11 -11
- fbgemm_gpu/triton/quantize.py +14 -9
- fbgemm_gpu/utils/filestore.py +56 -5
- fbgemm_gpu/utils/torch_library.py +2 -2
- fbgemm_gpu/utils/writeback_util.py +124 -0
- fbgemm_gpu/uvm.py +3 -0
- {fbgemm_gpu_nightly_cpu-2025.3.27.dist-info → fbgemm_gpu_nightly_cpu-2026.1.29.dist-info}/METADATA +3 -6
- fbgemm_gpu_nightly_cpu-2026.1.29.dist-info/RECORD +135 -0
- fbgemm_gpu_nightly_cpu-2026.1.29.dist-info/top_level.txt +2 -0
- fbgemm_gpu/docs/version.py → list_versions/__init__.py +5 -3
- list_versions/cli_run.py +161 -0
- fbgemm_gpu_nightly_cpu-2025.3.27.dist-info/RECORD +0 -126
- fbgemm_gpu_nightly_cpu-2025.3.27.dist-info/top_level.txt +0 -1
- {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,,
|
|
@@ -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
|
-
|
|
10
|
-
|
|
8
|
+
# pyre-strict
|
|
9
|
+
|
|
10
|
+
from .cli_run import CLI, CLIOutput
|
|
11
|
+
|
|
12
|
+
__all__ = ["CLI", "CLIOutput"]
|
list_versions/cli_run.py
ADDED
|
@@ -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
|
{fbgemm_gpu_nightly_cpu-2025.3.27.dist-info → fbgemm_gpu_nightly_cpu-2026.1.29.dist-info}/WHEEL
RENAMED
|
File without changes
|