Lines Matching defs:gpu

16 MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(GPU, gpu, gpu::GPUDialect)
23   return isa<gpu::AsyncTokenType>(unwrap(type));
27   return wrap(gpu::AsyncTokenType::get(unwrap(ctx)));
35   return llvm::isa<gpu::ObjectAttr>(unwrap(attr));
46   return wrap(gpu::ObjectAttr::get(
47       ctx, unwrap(target), static_cast<gpu::CompilationTarget>(format),
62   gpu::KernelTableAttr kernels;
64     kernels = llvm::cast<gpu::KernelTableAttr>(unwrap(mlirKernelsAttr));
65   return wrap(gpu::ObjectAttr::get(
66       ctx, unwrap(target), static_cast<gpu::CompilationTarget>(format),
71   gpu::ObjectAttr objectAttr =
72       llvm::cast<gpu::ObjectAttr>(unwrap(mlirObjectAttr));
77   gpu::ObjectAttr objectAttr =
78       llvm::cast<gpu::ObjectAttr>(unwrap(mlirObjectAttr));
83   gpu::ObjectAttr objectAttr =
84       llvm::cast<gpu::ObjectAttr>(unwrap(mlirObjectAttr));
90   gpu::ObjectAttr objectAttr =
91       llvm::cast<gpu::ObjectAttr>(unwrap(mlirObjectAttr));
96   gpu::ObjectAttr objectAttr =
97       llvm::cast<gpu::ObjectAttr>(unwrap(mlirObjectAttr));
102   gpu::ObjectAttr objectAttr =
103       llvm::cast<gpu::ObjectAttr>(unwrap(mlirObjectAttr));
108   gpu::ObjectAttr objectAttr =
109       llvm::cast<gpu::ObjectAttr>(unwrap(mlirObjectAttr));