erfanzar
diff --git a/‎src/jax_flash_attn2/__init__.py‎ renamed to ‎jax_flash_attn2/__init__.py‎
Lines changed: 2 additions & 0 deletions b/‎src/jax_flash_attn2/__init__.py‎ renamed to ‎jax_flash_attn2/__init__.py‎
Lines changed: 2 additions & 0 deletions
diff --git a/‎src/jax_flash_attn2/_custom_call_lib/__init__.py‎ renamed to ‎jax_flash_attn2/_custom_call_lib/__init__.py‎ b/‎src/jax_flash_attn2/_custom_call_lib/__init__.py‎ renamed to ‎jax_flash_attn2/_custom_call_lib/__init__.py‎
diff --git a/‎src/jax_flash_attn2/_custom_call_lib/lib.py‎ renamed to ‎jax_flash_attn2/_custom_call_lib/lib.py‎ b/‎src/jax_flash_attn2/_custom_call_lib/lib.py‎ renamed to ‎jax_flash_attn2/_custom_call_lib/lib.py‎
diff --git a/‎src/jax_flash_attn2/cpu_calls/__init__.py‎ renamed to ‎jax_flash_attn2/cpu_calls/__init__.py‎ b/‎src/jax_flash_attn2/cpu_calls/__init__.py‎ renamed to ‎jax_flash_attn2/cpu_calls/__init__.py‎
diff --git a/‎src/jax_flash_attn2/cpu_calls/mha.py‎ renamed to ‎jax_flash_attn2/cpu_calls/mha.py‎ b/‎src/jax_flash_attn2/cpu_calls/mha.py‎ renamed to ‎jax_flash_attn2/cpu_calls/mha.py‎
diff --git a/‎src/jax_flash_attn2/flash_attention.py‎ renamed to ‎jax_flash_attn2/flash_attention.py‎ b/‎src/jax_flash_attn2/flash_attention.py‎ renamed to ‎jax_flash_attn2/flash_attention.py‎
diff --git a/‎src/jax_flash_attn2/pallas_kernels/__init__.py‎ renamed to ‎jax_flash_attn2/pallas_kernels/__init__.py‎ b/‎src/jax_flash_attn2/pallas_kernels/__init__.py‎ renamed to ‎jax_flash_attn2/pallas_kernels/__init__.py‎
diff --git a/‎src/jax_flash_attn2/pallas_kernels/gpu_mha_kernel.py‎ renamed to ‎jax_flash_attn2/pallas_kernels/gpu_mha_kernel.py‎ b/‎src/jax_flash_attn2/pallas_kernels/gpu_mha_kernel.py‎ renamed to ‎jax_flash_attn2/pallas_kernels/gpu_mha_kernel.py‎
diff --git a/‎src/jax_flash_attn2/triton_kernels/__init__.py‎ renamed to ‎jax_flash_attn2/triton_kernels/__init__.py‎ b/‎src/jax_flash_attn2/triton_kernels/__init__.py‎ renamed to ‎jax_flash_attn2/triton_kernels/__init__.py‎
diff --git a/‎src/jax_flash_attn2/triton_kernels/gqa_kernel.py‎ renamed to ‎jax_flash_attn2/triton_kernels/gqa_kernel.py‎ b/‎src/jax_flash_attn2/triton_kernels/gqa_kernel.py‎ renamed to ‎jax_flash_attn2/triton_kernels/gqa_kernel.py‎
@@ -25,3 +25,5 @@
 	"FlashAttention",
 	"AttentionConfig",
 ]
+
+__version__ = "0.0.1"
Original file line number	Diff line number	Diff line change
`@@ -25,3 +25,5 @@`
`25`	`25`	`"FlashAttention",`
`26`	`26`	`"AttentionConfig",`
`27`	`27`	`]`
	`28`	`+`
	`29`	`+__version__ = "0.0.1"`