// 32 bit integers are supported for the values.
class IsConstantValueOf<int value> : Constraint<CPred<
  "IsConstantValueOf($0," # value # ")">>;

def IsOnGpuDevicePred : CPred<"IsOnGpuDevice($0.getOwner())">;

    return std::nullopt;
  }
  if (!parsed_name.has_type) {
    return std::nullopt;
  }
  return parsed_name.type;
}

}  // namespace

bool IsOnGpuDevice(mlir::Operation *op) {
  std::optional<std::string> device = GetOpDevice(op);
  if (!device) return false;
  return *device == kDeviceGpu;
}

void CopyDeviceAndUnderscoredAttributesAdaptor(mlir::OpResult src,

      return true;
  }

  return false;
}

// Returns true if `op` is placed on GPU device, and false if it's on other
// devices or the device is not specified.
bool IsOnGpuDevice(mlir::Operation *op);

// Wrappers for CopyDeviceAndUnderscoredAttributes
void CopyDeviceAndUnderscoredAttributesAdaptor(mlir::OpResult src,
                                               mlir::OpResult dest);