[spirv-writer] Add preliminary support for GLSL methods

This CL adds the type determination and builder code to support
outputting the GLSL Round call.

Bug: tint:5
Change-Id: I84dadebaf19aee3361fb13b5f32ce1a9f1b0c421
Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/19923
Reviewed-by: David Neto <dneto@google.com>
This commit is contained in:
dan sinclair 2020-04-20 15:46:18 +00:00 committed by dan sinclair
parent d4d87edc85
commit fd5d4ca16c
12 changed files with 404 additions and 20 deletions

View File

@ -550,6 +550,7 @@ source_set("tint_unittests_spv_writer_src") {
"src/writer/spirv/binary_writer_test.cc", "src/writer/spirv/binary_writer_test.cc",
"src/writer/spirv/builder_assign_test.cc", "src/writer/spirv/builder_assign_test.cc",
"src/writer/spirv/builder_binary_expression_test.cc", "src/writer/spirv/builder_binary_expression_test.cc",
"src/writer/spirv/builder_call_test.cc",
"src/writer/spirv/builder_constructor_expression_test.cc", "src/writer/spirv/builder_constructor_expression_test.cc",
"src/writer/spirv/builder_entry_point_test.cc", "src/writer/spirv/builder_entry_point_test.cc",
"src/writer/spirv/builder_function_test.cc", "src/writer/spirv/builder_function_test.cc",

View File

@ -423,6 +423,7 @@ if(${TINT_BUILD_SPV_WRITER})
writer/spirv/binary_writer_test.cc writer/spirv/binary_writer_test.cc
writer/spirv/builder_assign_test.cc writer/spirv/builder_assign_test.cc
writer/spirv/builder_binary_expression_test.cc writer/spirv/builder_binary_expression_test.cc
writer/spirv/builder_call_test.cc
writer/spirv/builder_constructor_expression_test.cc writer/spirv/builder_constructor_expression_test.cc
writer/spirv/builder_entry_point_test.cc writer/spirv/builder_entry_point_test.cc
writer/spirv/builder_function_test.cc writer/spirv/builder_function_test.cc

View File

@ -18,6 +18,7 @@
#include <memory> #include <memory>
#include <ostream> #include <ostream>
#include <string> #include <string>
#include <unordered_map>
#include <vector> #include <vector>
#include "src/ast/node.h" #include "src/ast/node.h"
@ -31,13 +32,13 @@ class Import : public Node {
/// Create a new empty import statement /// Create a new empty import statement
Import() = default; Import() = default;
/// Create a new import statement /// Create a new import statement
/// @param path The import path e.g. GLSL.std.430 /// @param path The import path e.g. GLSL.std.450
/// @param name The import reference name e.g. std:: /// @param name The import reference name e.g. std
Import(const std::string& path, const std::string& name); Import(const std::string& path, const std::string& name);
/// Create a new import statement /// Create a new import statement
/// @param source The input source for the import statement /// @param source The input source for the import statement
/// @param path The import path e.g. GLSL.std.430 /// @param path The import path e.g. GLSL.std.430
/// @param name The import reference name e.g. std:: /// @param name The import reference name e.g. std
Import(const Source& source, Import(const Source& source,
const std::string& path, const std::string& path,
const std::string& name); const std::string& name);
@ -57,6 +58,24 @@ class Import : public Node {
/// @returns the import name /// @returns the import name
const std::string& name() const { return name_; } const std::string& name() const { return name_; }
/// Add the given |name| to map to the given |id|
/// @param name the name to map
/// @param id the id to map too
void AddMethodId(const std::string& name, uint32_t id) {
method_to_id_map_[name] = id;
}
/// Retrieves the id for a given name
/// @param name the name to lookup
/// @returns the id for the given name or 0 on failure
uint32_t GetIdForMethod(const std::string& name) const {
auto val = method_to_id_map_.find(name);
if (val == method_to_id_map_.end()) {
return 0;
}
return val->second;
}
/// @returns true if the name and path are both present /// @returns true if the name and path are both present
bool IsValid() const override; bool IsValid() const override;
@ -70,6 +89,7 @@ class Import : public Node {
std::string path_; std::string path_;
std::string name_; std::string name_;
std::unordered_map<std::string, uint32_t> method_to_id_map_;
}; };
/// A list of unique imports /// A list of unique imports

View File

@ -25,7 +25,7 @@ Module::Module(Module&&) = default;
Module::~Module() = default; Module::~Module() = default;
const Import* Module::FindImportByName(const std::string& name) { Import* Module::FindImportByName(const std::string& name) {
for (const auto& import : imports_) { for (const auto& import : imports_) {
if (import->name() == name) if (import->name() == name)
return import.get(); return import.get();

View File

@ -47,7 +47,7 @@ class Module {
/// Find the import of the given name /// Find the import of the given name
/// @param name The import name to search for /// @param name The import name to search for
/// @returns the import with the given name if found, nullptr otherwise. /// @returns the import with the given name if found, nullptr otherwise.
const Import* FindImportByName(const std::string& name); Import* FindImportByName(const std::string& name);
/// Add a global variable to the module /// Add a global variable to the module
/// @param var the variable to add /// @param var the variable to add

View File

@ -16,6 +16,7 @@
#include <memory> #include <memory>
#include "spirv/unified1/GLSL.std.450.h"
#include "src/ast/array_accessor_expression.h" #include "src/ast/array_accessor_expression.h"
#include "src/ast/as_expression.h" #include "src/ast/as_expression.h"
#include "src/ast/assignment_statement.h" #include "src/ast/assignment_statement.h"
@ -214,6 +215,15 @@ bool TypeDeterminer::DetermineResultType(ast::Statement* stmt) {
return false; return false;
} }
bool TypeDeterminer::DetermineResultType(const ast::ExpressionList& list) {
for (const auto& expr : list) {
if (!DetermineResultType(expr.get())) {
return false;
}
}
return true;
}
bool TypeDeterminer::DetermineResultType(ast::Expression* expr) { bool TypeDeterminer::DetermineResultType(ast::Expression* expr) {
// This is blindly called above, so in some cases the expression won't exist. // This is blindly called above, so in some cases the expression won't exist.
if (!expr) { if (!expr) {
@ -285,14 +295,43 @@ bool TypeDeterminer::DetermineAs(ast::AsExpression* expr) {
} }
bool TypeDeterminer::DetermineCall(ast::CallExpression* expr) { bool TypeDeterminer::DetermineCall(ast::CallExpression* expr) {
for (const auto& param : expr->params()) { if (!DetermineResultType(expr->params())) {
if (!DetermineResultType(param.get())) { return false;
return false;
}
} }
if (!DetermineResultType(expr->func())) { // The expression has to be an identifier as you can't store function pointers
return false; // but, if it isn't we'll just use the normal result determination to be on
// the safe side.
if (expr->func()->IsIdentifier()) {
auto* ident = expr->func()->AsIdentifier();
if (ident->has_path()) {
auto* imp = mod_->FindImportByName(ident->path());
if (imp == nullptr) {
error_ = "Unable to find import for " + ident->name();
return false;
}
uint32_t ext_id = 0;
auto* result_type =
GetImportData(imp->path(), ident->name(), expr->params(), &ext_id);
if (result_type == nullptr) {
return false;
}
imp->AddMethodId(ident->name(), ext_id);
expr->func()->set_result_type(result_type);
} else {
// An identifier with a single name is a function call, not an import
// lookup which we can handle with the regular identifier lookup.
if (!DetermineResultType(ident)) {
return false;
}
}
} else {
if (!DetermineResultType(expr->func())) {
return false;
}
} }
expr->set_result_type(expr->func()->result_type()); expr->set_result_type(expr->func()->result_type());
return true; return true;
@ -314,8 +353,8 @@ bool TypeDeterminer::DetermineConstructor(ast::ConstructorExpression* expr) {
bool TypeDeterminer::DetermineIdentifier(ast::IdentifierExpression* expr) { bool TypeDeterminer::DetermineIdentifier(ast::IdentifierExpression* expr) {
if (expr->has_path()) { if (expr->has_path()) {
// TODO(dsinclair): Handle imports set_error(expr->source(),
set_error(expr->source(), "imports not handled in type determination"); "determine identifier should not be called with imports");
return false; return false;
} }
@ -459,10 +498,8 @@ bool TypeDeterminer::DetermineUnaryDerivative(
} }
bool TypeDeterminer::DetermineUnaryMethod(ast::UnaryMethodExpression* expr) { bool TypeDeterminer::DetermineUnaryMethod(ast::UnaryMethodExpression* expr) {
for (const auto& param : expr->params()) { if (!DetermineResultType(expr->params())) {
if (!DetermineResultType(param.get())) { return false;
return false;
}
} }
switch (expr->op()) { switch (expr->op()) {
@ -530,4 +567,32 @@ bool TypeDeterminer::DetermineUnaryOp(ast::UnaryOpExpression* expr) {
return true; return true;
} }
ast::type::Type* TypeDeterminer::GetImportData(
const std::string& path,
const std::string& name,
const ast::ExpressionList& params,
uint32_t* id) {
if (path != "GLSL.std.450") {
return nullptr;
}
if (name == "round") {
if (params.size() != 1) {
error_ = "incorrect number of parameters for round. Expected 1 got " +
std::to_string(params.size());
return nullptr;
}
if (!params[0]->result_type()->is_float_scalar_or_vector()) {
error_ =
"incorrect type for round. Requires a float scalar or a float vector";
return nullptr;
}
*id = GLSLstd450Round;
return params[0]->result_type();
}
return nullptr;
}
} // namespace tint } // namespace tint

View File

@ -71,6 +71,10 @@ class TypeDeterminer {
/// @param stmt the statement to check /// @param stmt the statement to check
/// @returns true if the determination was successful /// @returns true if the determination was successful
bool DetermineResultType(ast::Statement* stmt); bool DetermineResultType(ast::Statement* stmt);
/// Determines type information for an expression list
/// @param list the expression list to check
/// @returns true if the determination was successful
bool DetermineResultType(const ast::ExpressionList& list);
/// Determines type information for an expression /// Determines type information for an expression
/// @param expr the expression to check /// @param expr the expression to check
/// @returns true if the determination was successful /// @returns true if the determination was successful
@ -87,6 +91,17 @@ class TypeDeterminer {
variable_stack_.set(var->name(), var); variable_stack_.set(var->name(), var);
} }
/// Retrieves information for the requested import.
/// @param path the import path
/// @param name the method name to get information on
/// @param params the parameters to the method call
/// @param id out parameter for the external call ID. Must not be a nullptr.
/// @returns the return type of |name| in |path| or nullptr on error.
ast::type::Type* GetImportData(const std::string& path,
const std::string& name,
const ast::ExpressionList& params,
uint32_t* id);
private: private:
void set_error(const Source& src, const std::string& msg); void set_error(const Source& src, const std::string& msg);

View File

@ -16,8 +16,10 @@
#include <memory> #include <memory>
#include <utility> #include <utility>
#include <vector>
#include "gtest/gtest.h" #include "gtest/gtest.h"
#include "spirv/unified1/GLSL.std.450.h"
#include "src/ast/array_accessor_expression.h" #include "src/ast/array_accessor_expression.h"
#include "src/ast/as_expression.h" #include "src/ast/as_expression.h"
#include "src/ast/assignment_statement.h" #include "src/ast/assignment_statement.h"
@ -570,6 +572,27 @@ TEST_F(TypeDeterminerTest, Expr_Call_WithParams) {
EXPECT_TRUE(param_ptr->result_type()->IsF32()); EXPECT_TRUE(param_ptr->result_type()->IsF32());
} }
TEST_F(TypeDeterminerTest, Expr_Call_GLSLImport) {
ast::type::F32Type f32;
mod()->AddImport(std::make_unique<ast::Import>("GLSL.std.450", "std"));
// Register the function
EXPECT_TRUE(td()->Determine());
ast::ExpressionList call_params;
call_params.push_back(std::make_unique<ast::ScalarConstructorExpression>(
std::make_unique<ast::FloatLiteral>(&f32, 2.4)));
std::vector<std::string> name{"std", "round"};
ast::CallExpression call(std::make_unique<ast::IdentifierExpression>(name),
std::move(call_params));
EXPECT_TRUE(td()->DetermineResultType(&call));
ASSERT_NE(call.result_type(), nullptr);
EXPECT_TRUE(call.result_type()->IsF32());
}
TEST_F(TypeDeterminerTest, Expr_Cast) { TEST_F(TypeDeterminerTest, Expr_Cast) {
ast::type::F32Type f32; ast::type::F32Type f32;
ast::CastExpression cast(&f32, ast::CastExpression cast(&f32,
@ -1456,5 +1479,92 @@ TEST_F(TypeDeterminerTest, StorageClass_NonFunctionClassError) {
"function variable has a non-function storage class"); "function variable has a non-function storage class");
} }
TEST_F(TypeDeterminerTest, ImportData_Round_Scalar) {
ast::type::F32Type f32;
ast::ExpressionList params;
params.push_back(std::make_unique<ast::ScalarConstructorExpression>(
std::make_unique<ast::IntLiteral>(&f32, 1.f)));
ASSERT_TRUE(td()->DetermineResultType(params)) << td()->error();
uint32_t id = 0;
auto* type = td()->GetImportData("GLSL.std.450", "round", params, &id);
ASSERT_NE(type, nullptr);
EXPECT_TRUE(type->is_float_scalar());
EXPECT_EQ(id, GLSLstd450Round);
}
TEST_F(TypeDeterminerTest, ImportData_Round_Vector) {
ast::type::F32Type f32;
ast::type::VectorType vec(&f32, 3);
ast::ExpressionList vals;
vals.push_back(std::make_unique<ast::ScalarConstructorExpression>(
std::make_unique<ast::FloatLiteral>(&f32, 1.0f)));
vals.push_back(std::make_unique<ast::ScalarConstructorExpression>(
std::make_unique<ast::FloatLiteral>(&f32, 1.0f)));
vals.push_back(std::make_unique<ast::ScalarConstructorExpression>(
std::make_unique<ast::FloatLiteral>(&f32, 3.0f)));
ast::ExpressionList params;
params.push_back(
std::make_unique<ast::TypeConstructorExpression>(&vec, std::move(vals)));
ASSERT_TRUE(td()->DetermineResultType(params)) << td()->error();
uint32_t id = 0;
auto* type = td()->GetImportData("GLSL.std.450", "round", params, &id);
ASSERT_NE(type, nullptr);
EXPECT_TRUE(type->is_float_vector());
EXPECT_EQ(type->AsVector()->size(), 3);
EXPECT_EQ(id, GLSLstd450Round);
}
TEST_F(TypeDeterminerTest, ImportData_Round_Error_Integer) {
ast::type::I32Type i32;
ast::ExpressionList params;
params.push_back(std::make_unique<ast::ScalarConstructorExpression>(
std::make_unique<ast::IntLiteral>(&i32, 1)));
ASSERT_TRUE(td()->DetermineResultType(params)) << td()->error();
uint32_t id = 0;
auto* type = td()->GetImportData("GLSL.std.450", "round", params, &id);
ASSERT_EQ(type, nullptr);
EXPECT_EQ(
td()->error(),
"incorrect type for round. Requires a float scalar or a float vector");
}
TEST_F(TypeDeterminerTest, ImportData_Round_Error_NoParams) {
ast::ExpressionList params;
uint32_t id = 0;
auto* type = td()->GetImportData("GLSL.std.450", "round", params, &id);
ASSERT_EQ(type, nullptr);
EXPECT_EQ(td()->error(),
"incorrect number of parameters for round. Expected 1 got 0");
}
TEST_F(TypeDeterminerTest, ImportData_Round_Error_MultipleParams) {
ast::type::F32Type f32;
ast::ExpressionList params;
params.push_back(std::make_unique<ast::ScalarConstructorExpression>(
std::make_unique<ast::FloatLiteral>(&f32, 1.f)));
params.push_back(std::make_unique<ast::ScalarConstructorExpression>(
std::make_unique<ast::FloatLiteral>(&f32, 1.f)));
params.push_back(std::make_unique<ast::ScalarConstructorExpression>(
std::make_unique<ast::FloatLiteral>(&f32, 1.f)));
ASSERT_TRUE(td()->DetermineResultType(params)) << td()->error();
uint32_t id = 0;
auto* type = td()->GetImportData("GLSL.std.450", "round", params, &id);
ASSERT_EQ(type, nullptr);
EXPECT_EQ(td()->error(),
"incorrect number of parameters for round. Expected 1 got 3");
}
} // namespace } // namespace
} // namespace tint } // namespace tint

View File

@ -22,6 +22,7 @@
#include "src/ast/binding_decoration.h" #include "src/ast/binding_decoration.h"
#include "src/ast/bool_literal.h" #include "src/ast/bool_literal.h"
#include "src/ast/builtin_decoration.h" #include "src/ast/builtin_decoration.h"
#include "src/ast/call_expression.h"
#include "src/ast/constructor_expression.h" #include "src/ast/constructor_expression.h"
#include "src/ast/decorated_variable.h" #include "src/ast/decorated_variable.h"
#include "src/ast/else_statement.h" #include "src/ast/else_statement.h"
@ -430,10 +431,18 @@ bool Builder::GenerateGlobalVariable(ast::Variable* var) {
uint32_t Builder::GenerateIdentifierExpression( uint32_t Builder::GenerateIdentifierExpression(
ast::IdentifierExpression* expr) { ast::IdentifierExpression* expr) {
// TODO(dsinclair): handle names with namespaces in them ...
uint32_t val = 0; uint32_t val = 0;
if (!scope_stack_.get(expr->name(), &val)) { if (expr->has_path()) {
auto* imp = mod_->FindImportByName(expr->path());
if (imp == nullptr) {
error_ = "unable to find import for " + expr->path();
return 0;
}
val = imp->GetIdForMethod(expr->name());
if (val == 0) {
error_ = "unable to lookup: " + expr->name() + " in " + expr->path();
}
} else if (!scope_stack_.get(expr->name(), &val)) {
error_ = "unable to find name for identifier: " + expr->name(); error_ = "unable to find name for identifier: " + expr->name();
return 0; return 0;
} }
@ -691,6 +700,59 @@ uint32_t Builder::GenerateBinaryExpression(ast::BinaryExpression* expr) {
return result_id; return result_id;
} }
uint32_t Builder::GenerateCallExpression(ast::CallExpression* expr) {
// TODO(dsinclair): Support regular function calls
if (!expr->func()->IsIdentifier() ||
!expr->func()->AsIdentifier()->has_path()) {
error_ = "function calls not supported yet.";
return 0;
}
auto* ident = expr->func()->AsIdentifier();
auto type_id = GenerateTypeIfNeeded(expr->func()->result_type());
if (type_id == 0) {
return 0;
}
auto set_iter = import_name_to_id_.find(ident->path());
if (set_iter == import_name_to_id_.end()) {
error_ = "unknown import " + ident->path();
return 0;
}
auto set_id = set_iter->second;
auto* imp = mod_->FindImportByName(ident->path());
if (imp == nullptr) {
error_ = "unknown import " + ident->path();
return 0;
}
auto inst_id = imp->GetIdForMethod(ident->name());
if (inst_id == 0) {
error_ = "unknown method " + ident->name();
return 0;
}
auto result = result_op();
auto result_id = result.to_i();
std::vector<Operand> ops{Operand::Int(type_id), result, Operand::Int(set_id),
Operand::Int(inst_id)};
for (const auto& param : expr->params()) {
auto id = GenerateExpression(param.get());
if (id == 0) {
return 0;
}
ops.push_back(Operand::Int(id));
}
push_function_inst(spv::Op::OpExtInst, std::move(ops));
return result_id;
}
bool Builder::GenerateConditionalBlock( bool Builder::GenerateConditionalBlock(
ast::Expression* cond, ast::Expression* cond,
const ast::StatementList& true_body, const ast::StatementList& true_body,

View File

@ -202,6 +202,10 @@ class Builder {
/// @param expr the expression to generate /// @param expr the expression to generate
/// @returns the expression ID on success or 0 otherwise /// @returns the expression ID on success or 0 otherwise
uint32_t GenerateBinaryExpression(ast::BinaryExpression* expr); uint32_t GenerateBinaryExpression(ast::BinaryExpression* expr);
/// Generates a call expression
/// @param expr the expression to generate
/// @returns the expression ID on success or 0 otherwise
uint32_t GenerateCallExpression(ast::CallExpression* expr);
/// Generates a loop statement /// Generates a loop statement
/// @param stmt the statement to generate /// @param stmt the statement to generate
/// @returns true on successful generation /// @returns true on successful generation

View File

@ -0,0 +1,81 @@
// Copyright 2020 The Tint Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include <memory>
#include "gtest/gtest.h"
#include "src/ast/call_expression.h"
#include "src/ast/float_literal.h"
#include "src/ast/identifier_expression.h"
#include "src/ast/scalar_constructor_expression.h"
#include "src/ast/type/f32_type.h"
#include "src/ast/type/void_type.h"
#include "src/context.h"
#include "src/type_determiner.h"
#include "src/writer/spirv/builder.h"
#include "src/writer/spirv/spv_dump.h"
namespace tint {
namespace writer {
namespace spirv {
namespace {
using BuilderTest = testing::Test;
TEST_F(BuilderTest, Call_GLSLMethod) {
ast::type::F32Type f32;
ast::type::VoidType void_type;
ast::ExpressionList params;
params.push_back(std::make_unique<ast::ScalarConstructorExpression>(
std::make_unique<ast::FloatLiteral>(&f32, 1.f)));
ast::CallExpression expr(std::make_unique<ast::IdentifierExpression>(
std::vector<std::string>{"std", "round"}),
std::move(params));
Context ctx;
ast::Module mod;
TypeDeterminer td(&ctx, &mod);
auto imp = std::make_unique<ast::Import>("GLSL.std.450", "std");
auto* glsl = imp.get();
mod.AddImport(std::move(imp));
ASSERT_TRUE(td.DetermineResultType(&expr)) << td.error();
ast::Function func("a_func", {}, &void_type);
Builder b(&mod);
b.GenerateImport(glsl);
ASSERT_TRUE(b.GenerateFunction(&func)) << b.error();
EXPECT_EQ(b.GenerateCallExpression(&expr), 7u) << b.error();
EXPECT_EQ(DumpBuilder(b), R"(%1 = OpExtInstImport "GLSL.std.450"
OpName %4 "a_func"
%3 = OpTypeVoid
%2 = OpTypeFunction %3
%6 = OpTypeFloat 32
%8 = OpConstant %6 1
%4 = OpFunction %3 None %2
%5 = OpLabel
%7 = OpExtInst %6 %1 Round %8
OpFunctionEnd
)");
}
} // namespace
} // namespace spirv
} // namespace writer
} // namespace tint

View File

@ -217,6 +217,31 @@ TEST_F(BuilderTest, IdentifierExpression_NoLoadConst) {
)"); )");
} }
TEST_F(BuilderTest, IdentifierExpression_ImportMethod) {
auto imp = std::make_unique<ast::Import>("GLSL.std.450", "std");
imp->AddMethodId("round", 42u);
ast::Module mod;
mod.AddImport(std::move(imp));
Builder b(&mod);
ast::IdentifierExpression expr(std::vector<std::string>({"std", "round"}));
EXPECT_EQ(b.GenerateIdentifierExpression(&expr), 42u) << b.error();
}
TEST_F(BuilderTest, IdentifierExpression_ImportMethod_NotFound) {
auto imp = std::make_unique<ast::Import>("GLSL.std.450", "std");
ast::Module mod;
mod.AddImport(std::move(imp));
Builder b(&mod);
ast::IdentifierExpression expr(std::vector<std::string>({"std", "ceil"}));
EXPECT_EQ(b.GenerateIdentifierExpression(&expr), 0u);
ASSERT_TRUE(b.has_error());
EXPECT_EQ(b.error(), "unable to lookup: ceil in std");
}
} // namespace } // namespace
} // namespace spirv } // namespace spirv
} // namespace writer } // namespace writer