File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change 39
39
#define __SYCL_TYPE (x )
40
40
#endif
41
41
42
- // joint matrix should only be included by default for SPIR or NVPTX backends
42
+ // joint matrix should only be included by default for SPIR, NVPTX or HIP(GFX90A
43
+ // only) backends
43
44
#if defined __SPIR__ || defined __NVPTX__ || !defined __SYCL_DEVICE_ONLY__ || \
44
45
defined __gfx90a__
45
46
#ifndef SYCL_EXT_ONEAPI_MATRIX_VERSION
Original file line number Diff line number Diff line change 253
253
devices = set ()
254
254
sp = subprocess .check_output (sycl_ls , text = True )
255
255
for line in sp .splitlines ():
256
+ if "gfx90a" in line :
257
+ config .available_features .add ("gpu-amd-gfx90a" )
256
258
(backend , device , _ ) = line [1 :].split (':' , 2 )
257
259
devices .add ('{}:{}' .format (backend , device ))
258
260
config .sycl_devices = list (devices )
281
283
if "ext_oneapi_hip:gpu" in config .sycl_devices and config .hip_platform == "AMD" :
282
284
config .available_features .add ('hip_amd' )
283
285
arch_flag = '-Xsycl-target-backend=amdgcn-amd-amdhsa --offload-arch=' + config .amd_arch
284
- if "gfx90a" in config .sycl_devices :
285
- config .available_features .add ("gpu-amd-gfx90a" )
286
286
elif "ext_oneapi_hip:gpu" in config .sycl_devices and config .hip_platform == "NVIDIA" :
287
287
config .available_features .add ('hip_nvidia' )
288
288
arch_flag = ""
You can’t perform that action at this time.
0 commit comments