changed use_fused_bwd_kernel to true

Author: prishajain1

src/maxdiffusion/tests/wan_magcache_test.py

@@ -91,7 +91,7 @@ def setUpClass(cls):
             "ici_context_parallelism=4",
             "ici_tensor_parallelism=1",
             "flash_min_seq_length=0",
-            'flash_block_sizes={"block_q":3024,"block_kv_compute":1024,"block_kv":2048,"block_q_dkv":3024,"block_kv_dkv":2048,"block_kv_dkv_compute":1024,"use_fused_bwd_kernel":false}',
+            'flash_block_sizes={"block_q":3024,"block_kv_compute":1024,"block_kv":2048,"block_q_dkv":3024,"block_kv_dkv":2048,"block_kv_dkv_compute":1024,"use_fused_bwd_kernel":true}',
         ],
         unittest=True,
     )
@@ -167,7 +167,7 @@ def setUpClass(cls):
             "ici_context_parallelism=4",
             "ici_tensor_parallelism=1",
             "flash_min_seq_length=0",
-            'flash_block_sizes={"block_q":3024,"block_kv_compute":1024,"block_kv":2048,"block_q_dkv":3024,"block_kv_dkv":2048,"block_kv_dkv_compute":1024,"use_fused_bwd_kernel":false}',
+            'flash_block_sizes={"block_q":3024,"block_kv_compute":1024,"block_kv":2048,"block_q_dkv":3024,"block_kv_dkv":2048,"block_kv_dkv_compute":1024,"use_fused_bwd_kernel":true}',
         ],
         unittest=True,
     )
@@ -246,7 +246,7 @@ def setUpClass(cls):
             "ici_context_parallelism=4",
             "ici_tensor_parallelism=1",
             "flash_min_seq_length=0",
-            'flash_block_sizes={"block_q":3024,"block_kv_compute":1024,"block_kv":2048,"block_q_dkv":3024,"block_kv_dkv":2048,"block_kv_dkv_compute":1024,"use_fused_bwd_kernel":false}',
+            'flash_block_sizes={"block_q":3024,"block_kv_compute":1024,"block_kv":2048,"block_q_dkv":3024,"block_kv_dkv":2048,"block_kv_dkv_compute":1024,"use_fused_bwd_kernel":true}',
         ],
         unittest=True,
     )
@@ -327,7 +327,7 @@ def setUpClass(cls):
             "ici_context_parallelism=4",
             "ici_tensor_parallelism=1",
             "flash_min_seq_length=0",
-            'flash_block_sizes={"block_q":3024,"block_kv_compute":1024,"block_kv":2048,"block_q_dkv":3024,"block_kv_dkv":2048,"block_kv_dkv_compute":1024,"use_fused_bwd_kernel":false}',
+            'flash_block_sizes={"block_q":3024,"block_kv_compute":1024,"block_kv":2048,"block_q_dkv":3024,"block_kv_dkv":2048,"block_kv_dkv_compute":1024,"use_fused_bwd_kernel":true}',
         ],
         unittest=True,
     )