Lines Matching defs:useNvGpu

78                                           bool useNvGpu) {
93   if (!useNvGpu &&
96   if (useNvGpu &&
277 static bool supportsMMaMatrixType(Operation *op, bool useNvGpu) {
281     return useNvGpu ? nvgpu::canLowerToWarpMatrixOperation(transferRead)
284     return useNvGpu ? nvgpu::canLowerToWarpMatrixOperation(transferWrite)
287     return useNvGpu &&
290     return contractSupportsMMAMatrixType(contract, useNvGpu);
346                                              bool useNvGpu) {
368     if (llvm::any_of(dependentOps, [useNvGpu](Operation *op) {
369           if (!supportsMMaMatrixType(op, useNvGpu)) {
1232                                               bool useNvGpu) {
1233   if (!useNvGpu) {
1244   SetVector<Operation *> ops = getOpToConvert(rootOp, /*useNvGpu=*/false);
1277   SetVector<Operation *> ops = getOpToConvert(rootOp, /*useNvGpu=*/true);
1323     useNvGpu.setValue(useNvGpu_);
1328     populatePrepareVectorToMMAPatterns(patterns, useNvGpu.getValue());
1333     if (useNvGpu) {
1345 std::unique_ptr<Pass> mlir::createConvertVectorToGPUPass(bool useNvGpu) {
1346   return std::make_unique<ConvertVectorToGPUPass>(useNvGpu);