File tree Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Original file line number Diff line number Diff line change 11import  tilelang .testing 
22
3- import  example_warp_specialize_flashmla 
43import  example_warp_specialize_gemm_barrierpipe_stage2 
54import  example_warp_specialize_gemm_copy_0_gemm_1 
65import  example_warp_specialize_gemm_copy_1_gemm_0 
76import  example_warp_specialize_gemm_softpipe_stage2 
87
9- 
10- @tilelang .testing .requires_cuda  
11- @tilelang .testing .requires_cuda_compute_version_eq (9 , 0 ) 
12- def  test_example_warp_specialize_flashmla ():
13-     example_warp_specialize_flashmla .main ()
8+ # TODO: skip for now as non-deterministic on H20 
9+ # CC @cunxiao 
10+ # @tilelang.testing.requires_cuda 
11+ # @tilelang.testing.requires_cuda_compute_version_eq(9, 0) 
12+ # def test_example_warp_specialize_flashmla(): 
13+ #     import example_warp_specialize_flashmla 
14+ #     example_warp_specialize_flashmla.main() 
1415
1516
1617@tilelang .testing .requires_cuda  
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments