File tree Expand file tree Collapse file tree 2 files changed +5
-5
lines changed 
backends/vulkan/test/op_tests/utils Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -177,6 +177,8 @@ def generate_benchmark_fixture(self) -> str:
177177
178178vkapi::ScalarType from_at_scalartype(c10::ScalarType at_scalartype) {{ 
179179  switch (at_scalartype) {{ 
180+     case c10::kDouble: 
181+       return vkapi::kDouble; 
180182    case c10::kFloat: 
181183      return vkapi::kFloat; 
182184    case c10::kHalf: 
@@ -187,6 +189,8 @@ def generate_benchmark_fixture(self) -> str:
187189      return vkapi::kInt; 
188190    case c10::kChar: 
189191      return vkapi::kChar; 
192+     case c10::kBool: 
193+       return vkapi::kBool; 
190194    default: 
191195      VK_THROW("Unsupported at::ScalarType!"); 
192196  }} 
Original file line number Diff line number Diff line change @@ -119,15 +119,11 @@ def gen_parameterization(self) -> str:
119119    case c10::kInt: 
120120      return vkapi::kInt; 
121121    case c10::kLong: 
122-       return vkapi::kLong; 
123-     case c10::kShort: 
124-       return vkapi::kShort; 
122+       return vkapi::kInt; 
125123    case c10::kChar: 
126124      return vkapi::kChar; 
127125    case c10::kBool: 
128126      return vkapi::kBool; 
129-     case c10::kByte: 
130-       return vkapi::kByte; 
131127    default: 
132128      VK_THROW("Unsupported at::ScalarType!"); 
133129  } 
    
 
   
 
     
   
   
          
     
  
    
     
 
    
      
     
 
     
    You can’t perform that action at this time.
  
 
    
  
     
    
      
        
     
 
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments