@@ -36,46 +36,16 @@ limitations under the License.
36
36
namespace stream_executor {
37
37
namespace gpu {
38
38
39
- // An empty struct to be used as a handle for all unsupported features in
40
- // current CUDA/HIP/SYCL version.
41
- struct UnsupportedGpuFeature {
42
- // This makes the struct the same size as all the other handles.
43
- void * payload;
44
- };
45
-
46
39
#if TENSORFLOW_USE_SYCL
47
40
48
41
using GpuStreamHandle = ::sycl::queue*;
49
- using GpuFunctionHandle = ::sycl::kernel*;
50
- using GpuDevicePtr = void *;
51
- using GpuGraphHandle = UnsupportedGpuFeature;
52
- using GpuGraphExecHandle = UnsupportedGpuFeature;
53
- using GpuGraphNodeHandle = UnsupportedGpuFeature;
54
- using GpuGraphConditionalHandle = UnsupportedGpuFeature;
55
42
56
43
#elif TENSORFLOW_USE_ROCM
57
44
58
45
using GpuStreamHandle = hipStream_t;
59
- using GpuFunctionHandle = hipFunction_t;
60
- using GpuDevicePtr = hipDeviceptr_t;
61
- using GpuGraphHandle = hipGraph_t;
62
- using GpuGraphExecHandle = hipGraphExec_t;
63
- using GpuGraphNodeHandle = hipGraphNode_t;
64
- using GpuGraphConditionalHandle = UnsupportedGpuFeature;
65
46
#else // CUDA
66
47
67
48
using GpuStreamHandle = CUstream;
68
- using GpuFunctionHandle = CUfunction;
69
- using GpuDevicePtr = CUdeviceptr;
70
- using GpuGraphHandle = CUgraph;
71
- using GpuGraphExecHandle = CUgraphExec;
72
- using GpuGraphNodeHandle = CUgraphNode;
73
-
74
- #if CUDA_VERSION >= 12030
75
- using GpuGraphConditionalHandle = CUgraphConditionalHandle;
76
- #else
77
- using GpuGraphConditionalHandle = UnsupportedGpuFeature;
78
- #endif // #if CUDA_VERSION >= 12030
79
49
80
50
#endif
81
51
0 commit comments