From c76d3e7e3daf7f1c1fe10f339942d2fa8bbdba7a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Seungjun=20Park=28=EB=B0=95=EC=8A=B9=EC=A4=80=29?=
 <77137245+seungjun-Park@users.noreply.github.com>
Date: Thu, 26 Jun 2025 19:44:11 +0900
Subject: [PATCH] modified setup.py. now flash-swin-attention can be installed
 on Windows.

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 7ccd914..8ad4b5d 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,5 @@
 from setuptools import setup, find_packages
+import platform
 
 setup(
     name='flash_swin_attn',
@@ -7,8 +8,9 @@
     install_requires=[
         'torch',
         'einops',
-        'triton'
+        'triton-windows' if platform.system() == 'Windows' else 'triton'
     ],
+
     entry_points={
         'console_scripts': [
             # Add command line scripts here