@@ -254,6 +254,9 @@ class HalideBuilder(BuilderConfig):
254254 def has_nvidia (self ):
255255 return self .arch == "x86" and self .os in ["windows" , "linux" ]
256256
257+ def handles_d3d12 (self ):
258+ return self .arch == "x86" and self .os == "windows"
259+
257260 def handles_vulkan (self ):
258261 # Stick with Windows on x86-64 for now. Others TBD.
259262 return self .arch == "x86" and self .os == "windows"
@@ -693,14 +696,13 @@ def get_gpu_dsp_targets(builder_type):
693696 yield "host-cuda" , False
694697 yield "host-opencl" , False
695698
696- # TODO: temporarily disabled because our only windows bot doesn't support it...
697- # if builder_type.os == 'windows':
698- # yield 'host-d3d12compute', False
699+ if builder_type .handles_d3d12 ():
700+ yield "host-d3d12compute" , False
699701
700- # If we're running on a capable GPU, add all optional feature flags to the vulkan target
701- # which are required to get all the correctness tests to pass
702- if builder_type .handles_vulkan ():
703- yield "host-vulkan-vk_int8-vk_int16-vk_int64-vk_float16-vk_float64-vk_v13" , False
702+ # If we're running on a capable GPU, add all optional feature flags to the vulkan target
703+ # which are required to get all the correctness tests to pass
704+ if builder_type .handles_vulkan ():
705+ yield "host-vulkan-vk_int8-vk_int16-vk_int64-vk_float16-vk_float64-vk_v13" , False
704706
705707 if builder_type .handles_webgpu ():
706708 yield "host-webgpu" , False
0 commit comments