14df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni/* 24df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * Copyright 2017, The Android Open Source Project 34df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * 44df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * Licensed under the Apache License, Version 2.0 (the "License"); 54df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * you may not use this file except in compliance with the License. 64df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * You may obtain a copy of the License at 74df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * 84df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * http://www.apache.org/licenses/LICENSE-2.0 94df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * 104df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * Unless required by applicable law or agreed to in writing, software 114df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * distributed under the License is distributed on an "AS IS" BASIS, 124df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 134df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * See the License for the specific language governing permissions and 144df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni * limitations under the License. 154df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni */ 164df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 174df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni#include "transformer.h" 184df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 194df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni#include "file_utils.h" 204df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni#include "spirit.h" 214df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni#include "test_utils.h" 224df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni#include "gtest/gtest.h" 234df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 243f30b6202dd5ad6ff66959131d216405850ed152Yang Ni#include <stdint.h> 253f30b6202dd5ad6ff66959131d216405850ed152Yang Ni 264df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ninamespace android { 274df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ninamespace spirit { 284df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 294df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ninamespace { 304df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 314df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Niclass MulToAddTransformer : public Transformer { 324df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Nipublic: 334df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni Instruction *transform(IMulInst *mul) override { 344df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni auto ret = new IAddInst(mul->mResultType, mul->mOperand1, mul->mOperand2); 354df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni ret->setId(mul->getId()); 364df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni return ret; 374df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni } 384df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni}; 394df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 404df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Niclass Deleter : public Transformer { 414df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Nipublic: 424df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni Instruction *transform(IMulInst *) override { return nullptr; } 434df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni}; 444df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 453f30b6202dd5ad6ff66959131d216405850ed152Yang Niclass NewDataTypeTransformer : public Transformer { 463f30b6202dd5ad6ff66959131d216405850ed152Yang Nipublic: 473f30b6202dd5ad6ff66959131d216405850ed152Yang Ni Instruction *transform(IMulInst *mul) override { 483f30b6202dd5ad6ff66959131d216405850ed152Yang Ni insert(mul); 493f30b6202dd5ad6ff66959131d216405850ed152Yang Ni auto *DoubleTy = getModule()->getFloatType(64); 503f30b6202dd5ad6ff66959131d216405850ed152Yang Ni ConstantInst *ConstDouble2 = getModule()->getConstant(DoubleTy, 2.0); 513f30b6202dd5ad6ff66959131d216405850ed152Yang Ni auto ret = new IAddInst(DoubleTy, mul, ConstDouble2); 523f30b6202dd5ad6ff66959131d216405850ed152Yang Ni 533f30b6202dd5ad6ff66959131d216405850ed152Yang Ni IdResult id = ret->getId(); 543f30b6202dd5ad6ff66959131d216405850ed152Yang Ni ret->setId(mul->getId()); 553f30b6202dd5ad6ff66959131d216405850ed152Yang Ni mul->setId(id); 563f30b6202dd5ad6ff66959131d216405850ed152Yang Ni 573f30b6202dd5ad6ff66959131d216405850ed152Yang Ni return ret; 583f30b6202dd5ad6ff66959131d216405850ed152Yang Ni } 593f30b6202dd5ad6ff66959131d216405850ed152Yang Ni}; 603f30b6202dd5ad6ff66959131d216405850ed152Yang Ni 614df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni} // annonymous namespace 624df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 634df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Niclass TransformerTest : public ::testing::Test { 644df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Niprotected: 654df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni virtual void SetUp() { mWordsGreyscale = readWords("greyscale.spv"); } 664df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 674df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni std::vector<uint32_t> mWordsGreyscale; 684df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 694df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Niprivate: 704df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni std::vector<uint32_t> readWords(const char *testFile) { 714df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni static const std::string testDataPath( 724df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni "frameworks/rs/rsov/compiler/spirit/test_data/"); 734df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni const std::string &fullPath = getAbsolutePath(testDataPath + testFile); 744df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni return readFile<uint32_t>(fullPath); 754df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni } 764df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni}; 774df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 784df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang NiTEST_F(TransformerTest, testMulToAdd) { 793e5fbb6faf66488ba49c9d2396350ae567dba315Yang Ni std::unique_ptr<Module> m(Deserialize<Module>(mWordsGreyscale)); 804df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 814df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni ASSERT_NE(nullptr, m); 824df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 834df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni EXPECT_EQ(1, countEntity<IAddInst>(m.get())); 844df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni EXPECT_EQ(1, countEntity<IMulInst>(m.get())); 854df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 864df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni MulToAddTransformer trans; 873f30b6202dd5ad6ff66959131d216405850ed152Yang Ni std::unique_ptr<Module> m1(trans.run(m.get())); 884df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 894df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni ASSERT_NE(nullptr, m1); 904df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 914df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni ASSERT_TRUE(m1->resolveIds()); 924df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 934df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni EXPECT_EQ(2, countEntity<IAddInst>(m1.get())); 944df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni EXPECT_EQ(0, countEntity<IMulInst>(m1.get())); 954df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni} 964df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 974df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang NiTEST_F(TransformerTest, testDeletion) { 983e5fbb6faf66488ba49c9d2396350ae567dba315Yang Ni std::unique_ptr<Module> m(Deserialize<Module>(mWordsGreyscale)); 994df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 1004df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni ASSERT_NE(nullptr, m.get()); 1014df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 1024df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni EXPECT_EQ(1, countEntity<IMulInst>(m.get())); 1034df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 1044df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni Deleter trans; 1053f30b6202dd5ad6ff66959131d216405850ed152Yang Ni std::unique_ptr<Module> m1(trans.run(m.get())); 1064df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 1074df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni ASSERT_NE(nullptr, m1.get()); 1084df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 1094df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni EXPECT_EQ(1, countEntity<IAddInst>(m1.get())); 1104df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni EXPECT_EQ(0, countEntity<IMulInst>(m1.get())); 1114df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni} 1124df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni 1133f30b6202dd5ad6ff66959131d216405850ed152Yang NiTEST_F(TransformerTest, testAddInstructionUsingNewDataType) { 1143e5fbb6faf66488ba49c9d2396350ae567dba315Yang Ni std::unique_ptr<Module> m(Deserialize<Module>(mWordsGreyscale)); 1153f30b6202dd5ad6ff66959131d216405850ed152Yang Ni 1163f30b6202dd5ad6ff66959131d216405850ed152Yang Ni ASSERT_NE(nullptr, m.get()); 1173f30b6202dd5ad6ff66959131d216405850ed152Yang Ni 1183f30b6202dd5ad6ff66959131d216405850ed152Yang Ni EXPECT_EQ(5, countEntity<ConstantInst>(m.get())); 1193f30b6202dd5ad6ff66959131d216405850ed152Yang Ni EXPECT_EQ(1, countEntity<TypeFloatInst>(m.get())); 1203f30b6202dd5ad6ff66959131d216405850ed152Yang Ni EXPECT_EQ(1, countEntity<IMulInst>(m.get())); 1213f30b6202dd5ad6ff66959131d216405850ed152Yang Ni 1223f30b6202dd5ad6ff66959131d216405850ed152Yang Ni NewDataTypeTransformer trans; 1233f30b6202dd5ad6ff66959131d216405850ed152Yang Ni std::unique_ptr<Module> m1(trans.run(m.get())); 1243f30b6202dd5ad6ff66959131d216405850ed152Yang Ni 1253f30b6202dd5ad6ff66959131d216405850ed152Yang Ni ASSERT_NE(nullptr, m1.get()); 1263f30b6202dd5ad6ff66959131d216405850ed152Yang Ni 1273f30b6202dd5ad6ff66959131d216405850ed152Yang Ni EXPECT_EQ(6, countEntity<ConstantInst>(m.get())); 1283f30b6202dd5ad6ff66959131d216405850ed152Yang Ni EXPECT_EQ(2, countEntity<TypeFloatInst>(m1.get())); 1293f30b6202dd5ad6ff66959131d216405850ed152Yang Ni EXPECT_EQ(2, countEntity<IAddInst>(m1.get())); 1303f30b6202dd5ad6ff66959131d216405850ed152Yang Ni EXPECT_EQ(1, countEntity<IMulInst>(m1.get())); 1313f30b6202dd5ad6ff66959131d216405850ed152Yang Ni} 1323f30b6202dd5ad6ff66959131d216405850ed152Yang Ni 1334df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni} // namespace spirit 1344df77d18bf57187f8e7142c6f7a70a9cdd3d581eYang Ni} // namespace android 135