Lines Matching refs:spirv

39     context.getOrLoadDialect<mlir::spirv::SPIRVDialect>();  in SerializationTest()
47 spirv::ModuleOp::getOperationName()); in initModuleOp()
50 spirv::AddressingModel::Logical))); in initModuleOp()
53 static_cast<uint32_t>(spirv::MemoryModel::GLSL450))); in initModuleOp()
55 spirv::VerCapExtAttr::get( in initModuleOp()
56 spirv::Version::V_1_0, ArrayRef<spirv::Capability>(), in initModuleOp()
57 ArrayRef<spirv::Extension>(), &context)); in initModuleOp()
58 spirv::ModuleOp::build(builder, state); in initModuleOp()
59 module = cast<spirv::ModuleOp>(Operation::create(state)); in initModuleOp()
63 spirv::StructType getFloatStructType() { in getFloatStructType()
66 llvm::SmallVector<spirv::StructType::OffsetInfo, 1> offsetInfo{0}; in getFloatStructType()
67 return spirv::StructType::get(elementTypes, offsetInfo); in getFloatStructType()
71 spirv::GlobalVariableOp addGlobalVar(Type type, llvm::StringRef name) { in addGlobalVar()
73 auto ptrType = spirv::PointerType::get(type, spirv::StorageClass::Uniform); in addGlobalVar()
74 return builder.create<spirv::GlobalVariableOp>( in addGlobalVar()
81 using HandleFn = llvm::function_ref<bool(spirv::Opcode opcode,
88 auto currOffset = spirv::kHeaderWordCount; in scanInstruction()
95 spirv::Opcode opcode = in scanInstruction()
96 static_cast<spirv::Opcode>(binary[currOffset] & 0xffff); in scanInstruction()
109 OwningOpRef<spirv::ModuleOp> module;
121 ASSERT_TRUE(succeeded(spirv::serialize(module.get(), binary))); in TEST_F()
123 auto hasBlockDecoration = [](spirv::Opcode opcode, in TEST_F()
125 return opcode == spirv::Opcode::OpDecorate && operands.size() == 2 && in TEST_F()
126 operands[1] == static_cast<uint32_t>(spirv::Decoration::Block); in TEST_F()
138 ASSERT_TRUE(succeeded(spirv::serialize(module.get(), binary))); in TEST_F()
141 auto countBlockDecoration = [&count](spirv::Opcode opcode, in TEST_F()
143 if (opcode == spirv::Opcode::OpDecorate && operands.size() == 2 && in TEST_F()
144 operands[1] == static_cast<uint32_t>(spirv::Decoration::Block)) in TEST_F()
156 spirv::SerializationOptions options; in TEST_F()
158 ASSERT_TRUE(succeeded(spirv::serialize(module.get(), binary, options))); in TEST_F()
160 auto hasVarName = [](spirv::Opcode opcode, ArrayRef<uint32_t> operands) { in TEST_F()
162 return opcode == spirv::Opcode::OpName && in TEST_F()
163 spirv::decodeStringLiteral(operands, index) == "var0"; in TEST_F()
172 spirv::SerializationOptions options; in TEST_F()
174 ASSERT_TRUE(succeeded(spirv::serialize(module.get(), binary, options))); in TEST_F()
176 auto hasVarName = [](spirv::Opcode opcode, ArrayRef<uint32_t> operands) { in TEST_F()
178 return opcode == spirv::Opcode::OpName && in TEST_F()
179 spirv::decodeStringLiteral(operands, index) == "var0"; in TEST_F()