Wheelodex » Projects » flashrnn

flashrnn

View on PyPI — Reverse Dependencies (0)

1.0.3 flashrnn-1.0.3-py3-none-any.whl

Wheel Details

Project: flashrnn
Version: 1.0.3
Filename: flashrnn-1.0.3-py3-none-any.whl
Download: [link]
Size: 154305
MD5: 5546953333bafcd36505e342274bc80d
SHA256: a0d345441862fd5b3e212c33ba8224db9eaae8cb5e81e51985e9ca51be282666
Uploaded: 2024-12-11 09:40:48 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: flashrnn
Version: 1.0.3
Summary: FlashRNN: Optimizing Traditional RNNs on Modern Hardware
Author-Email: Korbinian Pöppel <korbinian.poeppel[at]nx-ai.com>, Maximilian Beck <maximilian.beck[at]nx-ai.com>
Project-Url: Homepage, https://github.com/NX-AI/flashrnn
Project-Url: Issues, https://github.com/NX-AI/flashrnn/issues
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Requires-Python: >=3.8
Requires-Dist: torch
Requires-Dist: einops
Requires-Dist: triton
Requires-Dist: ninja
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 4851 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.6.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
flashrnn/__init__.py sha256=6yjtLp7M45Zb31l8OEab9ebAa6-L3k3maj1yMOyfDZM 89
flashrnn/autotune/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
flashrnn/autotune/caching.py sha256=bHxLVHmpo9HfNYvCYhOHn8lzW0vFrGR6X8fB-eGyEfM 1578
flashrnn/autotune/constrint.py sha256=9bp7AaRfG2ZwHCIflMm83X9quJm3s5WzR0d_yAoWt04 75834
flashrnn/autotune/runner.py sha256=RUuyW1WELabBIEPzUOQBy3Lp2HiOJFGeVb6hbamw16A 1371
flashrnn/autotune/submit.py sha256=JfYpJ6c9ewvj__CFWL5zMVQrDTo7MOm7bsLn8N2RBPA 4605
flashrnn/flashrnn/__init__.py sha256=6yjtLp7M45Zb31l8OEab9ebAa6-L3k3maj1yMOyfDZM 89
flashrnn/flashrnn/cuda_init.py sha256=5w5RjOnY5p9gQZxMOLruIJu0W9aCvrRD2sz0o_SEjv8 3313
flashrnn/flashrnn/cuda_init_parametric.py sha256=9Wg8UOJBas5kqVmtRVBwheuU1rP6HRXur7v4cR-Lz1A 13853
flashrnn/flashrnn/flashrnn.py sha256=7SKF9tdyaIfwq4Uyj8S9sMR9u0oKvBVB3Fpl_K3wVF0 49013
flashrnn/flashrnn/alternating/elman_pointwise.cu sha256=oX1sSVZ5sgxm5XdjivXNQwkqTxMz4VaQQqngPnHn500 4739
flashrnn/flashrnn/alternating/flashrnn.cc sha256=a_WlZfRH_NwdObds9_JjOUeetD1MmVEH5KmkW7eh1Ao 12740
flashrnn/flashrnn/alternating/flashrnn.h sha256=0L6YC3Pn6rl7l3S4thXLfhpPwknAixlWswBZfR7AqwM 18961
flashrnn/flashrnn/alternating/flashrnn_backward.cu sha256=iGrOMt3lWeAzgloY-eOW4fAUxn0XMaxOeFM_FdZe-kY 16716
flashrnn/flashrnn/alternating/flashrnn_backward_cut.cu sha256=NDL5B8upKCDsCyzoMmJNJwzBtKLf8cGF3x5b0-aGcgQ 16138
flashrnn/flashrnn/alternating/flashrnn_forward.cu sha256=EH5GVzUAMT9EIjNd9cGiS80RIfA6L9iZop1_2-IfpGQ 9578
flashrnn/flashrnn/alternating/flashrnn_pointwise.cuh sha256=LzAJJUqkXcBjb02rgytiaFUv5LWge3HHCYrNYNvDhb4 2994
flashrnn/flashrnn/alternating/gru_pointwise.cu sha256=3V-m657oS1Lws7UQ2RE0ABuBHNPFo2mZYA8YbZePR-Q 8027
flashrnn/flashrnn/alternating/lstm_pointwise.cu sha256=31NxDLUIoRPTx92DLX2wFMkzuvaWWNx9Y1u83djJ-MU 7623
flashrnn/flashrnn/alternating/slstm_pointwise.cu sha256=_yRgjuDiTQTGoFC4fgKmDGst-kHT8FRfDxBSB3QuOfU 9204
flashrnn/flashrnn/fused/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
flashrnn/flashrnn/fused/elman_fused_pointwise.cuh sha256=Sa892MqxV1e_KNhR7gizyvLZXkxdW7bB1HniFGUzmJs 3249
flashrnn/flashrnn/fused/flashrnn.cc sha256=g-LV1apRArhT9g0r-GeM_UCA5cTy05t35YDJs2u0vrs 14491
flashrnn/flashrnn/fused/flashrnn.h sha256=PtbKLs91eduVmzCv8Uvbtbk5yOc1yXrtYd1Z846lx90 7453
flashrnn/flashrnn/fused/flashrnn_fused_backward.cu sha256=cWzdtEw0VLwPQ-D2TjfqgYoy5kGYNoGuWwkN3IzrDWs 47353
flashrnn/flashrnn/fused/flashrnn_fused_backward_cut.cu sha256=pYJ7NEteYOTst1M0SJVNj19XYEVxnopg_n8AMSyiI4A 16891
flashrnn/flashrnn/fused/flashrnn_fused_forward.cu sha256=jzEY8Ps7DU0OQijS8rJaIrZHga0Oq857mlxDQ1dOJsk 41086
flashrnn/flashrnn/fused/flashrnn_fused_pointwise_base.cuh sha256=0uB5KdnrEd2kyCGkjKbeCQdSD9-D9o0B7APzno_MCiI 1395
flashrnn/flashrnn/fused/gru_fused_pointwise.cuh sha256=Ua0DGKpExY5u1kcGXWpWIYVNAPAU-eVwsu6zCwH2sfI 5149
flashrnn/flashrnn/fused/lstm_fused_pointwise.cuh sha256=gpkHLGAHHlWl5NsIR8ITV6o18dGf32MCshyWaplmjsg 5466
flashrnn/flashrnn/fused/slstm_fused_pointwise.cuh sha256=g3KhzX5sN2uizJrosXnnoy9bUQ8zH4Fkg9rk6tu8Hro 7297
flashrnn/flashrnn/gpu_info/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
flashrnn/flashrnn/gpu_info/assign_struct.py sha256=wqmXj-ZjqQRpR3_13SgRdPz_A0G-7xVN9TWG35kj7_M 1648
flashrnn/flashrnn/gpu_info/defs_cudaDeviceProperties.cc sha256=5gB23v52MpuTtlBX9vL2yKFC1viCsG-cV4rkKdZm3gY 8037
flashrnn/flashrnn/gpu_info/gpu_info.cc sha256=M5okx2uTDB0XLlexaz0BuaAqgMqnK_5wls5bhk3t9r8 13518
flashrnn/flashrnn/gpu_info/gpu_info.cu sha256=K7kRlsNB9ymCHFk8EBs8K-x7clgbhFurAj9hNGZ1G4Q 973
flashrnn/flashrnn/gpu_info/gpu_info.h sha256=fRPFilaEcSJzbbXPU64EXSAG6O0ueuUg28Fo5Kkcprc 952
flashrnn/flashrnn/gpu_info/gpu_info.py sha256=z4lAwUcvJ_-of_aGADQPaff83TkpaXxHfpyz-ZQ7Xuo 1089
flashrnn/flashrnn/gpu_info/gpu_info_test.cc sha256=-YUSydkbWumZVgJkAYVOjKOj8O-2xnGNm2DLUzeMF4I 1690
flashrnn/flashrnn/triton_fused/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
flashrnn/flashrnn/triton_fused/fwbw.py sha256=pQy-32wdCXUYNrSc75gcqEfjuCwmchkxb6hu_jNPZoA 6050
flashrnn/flashrnn/triton_fused/lstm_bw.py sha256=pXTUKHo8lIMeqScT0EzBh4QXZX9-C6kmpxaH2qQCE6E 22531
flashrnn/flashrnn/triton_fused/lstm_fw.py sha256=mBj02w8uNDtdRtqW8Id7T-jrhUyoATggFcFzeIn49OE 16108
flashrnn/flashrnn/triton_fused/slstm_bw.py sha256=ED6gfCqALb_0Dll-tDDfVMK8GDq6g9gkJYARY7N5bfg 26588
flashrnn/flashrnn/triton_fused/slstm_fw.py sha256=VGPcBVaDh6n5Ew1XXKbSPc9d7sQE-r--gFci98A-RCE 18894
flashrnn/flashrnn/triton_fused/triton_utils.py sha256=u8dfYZuD6UgSTYVaWo3FQ_pyE7M8mEBIIg4xQP54e4g 495
flashrnn/flashrnn/util/blas.cu sha256=iezo-xZbXTlHByNpP--Tq4UM0QYVs5RfSryjCbXP5Dg 10890
flashrnn/flashrnn/util/blas.h sha256=Cyhe3-PitNazRNw1YMOVghE-7OxxFBl6u0IfBlnGud4 7813
flashrnn/flashrnn/util/cuda_error.cu sha256=Eu4h0-PGmj3WpG-1X8QtT0mdZOuPwQeO24YaEJRd-TE 2156
flashrnn/flashrnn/util/cuda_error.h sha256=CeX4B8HJT3WNvNfRCNUF9DayFOvsgs2bwRF9MqDjBgg 320
flashrnn/flashrnn/util/device_assert.h sha256=8iHwMeoDfQmgim6Xh2c7m2wVmkE_0OsMT0qYzulczrE 1148
flashrnn/flashrnn/util/inline_ops.cuh sha256=izrspqFhU29CoOBMMpp6wT8RpSrXpiMSuoQY4biC3mY 7708
flashrnn/flashrnn/util/inline_ops_2bf16.cuh sha256=u8CyikEl8C_rtVBBQlpPIWFxClgeoShoVbEehKPrbEA 9844
flashrnn/flashrnn/util/inline_ops_2fp16.cuh sha256=WD7CW-_L1J_enFP_M8-ISu0jS3GobAgBrU3Rf3dESn4 8586
flashrnn/flashrnn/util/inline_ops_bf16.cuh sha256=cWTEXkVAjbXMiXyw8uQdNd4cKC82_UhpuI7WO6s6EBE 6628
flashrnn/flashrnn/util/inline_ops_fp16.cuh sha256=FtQ-e522DTihCrOfZMUT-YLknV_ac4MojbmD_EDnUuY 5658
flashrnn/flashrnn/util/inline_print.cuh sha256=xbvRjdqDdG6m4YWL-OAPbvJ4JafHoYBZWj7Ix7zy9wE 1978
flashrnn/flashrnn/util/support.h sha256=Tmuhc3oTSGAfbZijX9XmnmP-_omL-KeVQlRH0v1KYug 2775
flashrnn/flashrnn/util/util.h sha256=RkFiDRr-hFuSv67RDQQjqeC9uP18_kR9ZK7r8ggHcfk 61
flashrnn/flashrnn/vanilla/__init__.py sha256=3r7Aqs_BIEL1I0K-5DyN8hAxOmC7ngxaIgyInoZjkGo 4214
flashrnn/flashrnn/vanilla/elman.py sha256=r6zaAQP86uGm3radjU6rs-vjCgONKmxzlITHnvjcsy8 727
flashrnn/flashrnn/vanilla/gru.py sha256=gFJDEVFS_JNqHPmxeQqtcflcRDZPXNLRnaeIZK_NApA 964
flashrnn/flashrnn/vanilla/lstm.py sha256=NmMM8gv6J-eEj0-20jf7eFrfizJnNh5x_LMrtVEVKPE 988
flashrnn/flashrnn/vanilla/slstm.py sha256=QUG091SIeNJ4TkJJ8hUIJGIY1XQA1L0dFIvQewdLgSk 1142
flashrnn/flashrnn/vanilla_fwbw/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
flashrnn/flashrnn/vanilla_fwbw/bw.py sha256=pSWI2__JRqNTpl92I_WfvR5L72wnOkhbTJ7Lai5WyVA 6944
flashrnn/flashrnn/vanilla_fwbw/fw.py sha256=6KBwgBl7CLSqz-wcx1_zDKgLSgEZRcrSXyRnk5XLUWw 5046
flashrnn/flashrnn/vanilla_fwbw/fwbw.py sha256=7G_ZW-zeUHz2BRSI0JG1TYS9gKbB8P9BgnkETY66Bjk 5548
flashrnn/speed_experiments/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
flashrnn/speed_experiments/kernel_speed_benchmark.py sha256=0Wt7XpQ2a8At35P1ZjHJwd2HrKFLlocBPdj6UuZHYxU 25123
flashrnn/speed_experiments/plot_config.py sha256=4hGs2M5ZnHtELtohPckiSu9WA_VyOp1ssn8SWtO8YHE 6396
flashrnn/speed_experiments/plot_results.py sha256=9ws5g1rPV1kmTtxyBwWNKQC3dS8xB7_yEsqpdr9zKhA 8426
flashrnn/speed_experiments/baseline_kernels/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
flashrnn/speed_experiments/baseline_kernels/torch_lstm.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
flashrnn/speed_experiments/baseline_kernels/torch_sdp_attention.py sha256=GzrhqoF2NxR21r44gFhZhsce8TcsrMQWfeFibxIXcUA 1716
flashrnn-1.0.3.dist-info/LICENSE sha256=YrhgqlpUW_P6Fj4xuGhjIL-UKq7YpzeOPR22kzu_rRM 10283
flashrnn-1.0.3.dist-info/METADATA sha256=YEziA3uDDkB-nvwIiMeluUHJZfze9bbHnj1YHq8OPtA 5468
flashrnn-1.0.3.dist-info/WHEEL sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U 91
flashrnn-1.0.3.dist-info/top_level.txt sha256=Z8Lgr9k0Lx822336xmYEdo9kYpk9QlR9MD_NCQngHV0 9
flashrnn-1.0.3.dist-info/RECORD — —

top_level.txt

flashrnn