14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24
25
26 package org.graalvm.compiler.hotspot.amd64.test;
27
28 import static org.graalvm.compiler.lir.LIRInstruction.OperandFlag.REG;
29
30 import org.graalvm.compiler.api.replacements.Snippet;
31 import org.graalvm.compiler.asm.amd64.AMD64Address;
32 import org.graalvm.compiler.asm.amd64.AMD64MacroAssembler;
33 import org.graalvm.compiler.bytecode.BytecodeProvider;
34 import org.graalvm.compiler.core.common.LIRKind;
35 import org.graalvm.compiler.core.common.spi.ForeignCallDescriptor;
36 import org.graalvm.compiler.core.common.type.DataPointerConstant;
37 import org.graalvm.compiler.hotspot.HotSpotForeignCallLinkage;
38 import org.graalvm.compiler.hotspot.meta.HotSpotForeignCallsProviderImpl;
39 import org.graalvm.compiler.hotspot.meta.HotSpotProviders;
40 import org.graalvm.compiler.hotspot.stubs.SnippetStub;
41 import org.graalvm.compiler.lir.LIRInstructionClass;
42 import org.graalvm.compiler.lir.Variable;
43 import org.graalvm.compiler.lir.amd64.AMD64LIRInstruction;
44 import org.graalvm.compiler.lir.asm.ArrayDataPointerConstant;
45 import org.graalvm.compiler.lir.asm.CompilationResultBuilder;
46 import org.graalvm.compiler.lir.gen.LIRGeneratorTool;
47 import org.graalvm.compiler.lir.jtt.LIRTest;
48 import org.graalvm.compiler.lir.jtt.LIRTestSpecification;
49 import org.graalvm.compiler.nodes.extended.ForeignCallNode;
50 import org.graalvm.compiler.nodes.graphbuilderconf.GraphBuilderConfiguration;
51 import org.graalvm.compiler.nodes.graphbuilderconf.GraphBuilderContext;
52 import org.graalvm.compiler.nodes.graphbuilderconf.InvocationPlugin;
53 import org.graalvm.compiler.nodes.graphbuilderconf.InvocationPlugins;
54 import org.graalvm.compiler.options.OptionValues;
55 import org.graalvm.compiler.replacements.ReplacementsImpl;
56 import org.graalvm.compiler.replacements.classfile.ClassfileBytecodeProvider;
57 import org.junit.Assume;
58 import org.junit.Before;
59 import org.junit.Test;
60
61 import jdk.vm.ci.amd64.AMD64;
62 import jdk.vm.ci.amd64.AMD64.CPUFeature;
63 import jdk.vm.ci.amd64.AMD64Kind;
64 import jdk.vm.ci.code.Register;
65 import jdk.vm.ci.code.ValueUtil;
66 import jdk.vm.ci.meta.AllocatableValue;
67 import jdk.vm.ci.meta.MetaAccessProvider;
68 import jdk.vm.ci.meta.ResolvedJavaMethod;
69 import jdk.vm.ci.meta.Value;
70 import org.graalvm.compiler.hotspot.HotSpotBackend;
71
72 public class StubAVXTest extends LIRTest {
73
74 @Before
75 public void checkAMD64() {
76 Assume.assumeTrue("skipping AMD64 specific test", getTarget().arch instanceof AMD64);
77 Assume.assumeTrue("skipping AVX test", ((AMD64) getTarget().arch).getFeatures().contains(CPUFeature.AVX));
78 if (getBackend() instanceof HotSpotBackend) {
79 HotSpotBackend backend = (HotSpotBackend) getBackend();
80 Assume.assumeTrue("skipping because of MaxVectorSize", backend.getRuntime().getVMConfig().maxVectorSize >= 32);
81 }
82 }
83
84 private static final DataPointerConstant avxConstant = new ArrayDataPointerConstant(new float[]{1f, 2f, 3f, 4f, 5f, 6f, 7f, 8f}, 32);
85
86 private static class LoadAVXConstant extends AMD64LIRInstruction {
87 public static final LIRInstructionClass<LoadAVXConstant> TYPE = LIRInstructionClass.create(LoadAVXConstant.class);
88
89 @Def({REG}) AllocatableValue result;
90
166
167 @Override
168 public void generate(LIRGeneratorTool gen, Value arg0, Value arg1) {
169 Variable ret = gen.newVariable(LIRKind.value(AMD64Kind.DWORD));
170 gen.append(new CompareAVXRegister(ret, gen.asAllocatable(arg0), gen.asAllocatable(arg1), gen.newVariable(LIRKind.value(AMD64Kind.V256_QWORD))));
171 setResult(ret);
172 }
173 };
174
175 private static class TestStub extends SnippetStub {
176
177 TestStub(OptionValues options, HotSpotProviders providers, HotSpotForeignCallLinkage linkage) {
178 super("testStub", options, providers, linkage);
179 }
180
181 @Snippet
182 static void testStub() {
183 }
184
185 @Override
186 protected BytecodeProvider getReplacementsBytecodeProvider() {
187 ReplacementsImpl d = (ReplacementsImpl) providers.getReplacements();
188 MetaAccessProvider metaAccess = d.providers.getMetaAccess();
189 return new ClassfileBytecodeProvider(metaAccess, d.snippetReflection, ClassLoader.getSystemClassLoader());
190 }
191 }
192
193 public static final ForeignCallDescriptor TEST_STUB = new ForeignCallDescriptor("test_stub", void.class);
194
195 @LIRIntrinsic
196 public static int compareAVXRegister(@SuppressWarnings("unused") LIRTestSpecification spec, Object left, Object right) {
197 return left == right ? 0xff : 0;
198 }
199
200 @Override
201 protected GraphBuilderConfiguration editGraphBuilderConfiguration(GraphBuilderConfiguration conf) {
202 InvocationPlugins invocationPlugins = conf.getPlugins().getInvocationPlugins();
203 InvocationPlugins.Registration r = new InvocationPlugins.Registration(invocationPlugins, TestStub.class);
204 r.register0("testStub", new InvocationPlugin() {
205 @Override
206 public boolean apply(GraphBuilderContext b, ResolvedJavaMethod targetMethod, InvocationPlugin.Receiver receiver) {
207 b.add(new ForeignCallNode(getProviders().getForeignCalls(), TEST_STUB));
208 return true;
209 }
|
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24
25
26 package org.graalvm.compiler.hotspot.amd64.test;
27
28 import static org.graalvm.compiler.lir.LIRInstruction.OperandFlag.REG;
29
30 import org.graalvm.compiler.api.replacements.Snippet;
31 import org.graalvm.compiler.asm.amd64.AMD64Address;
32 import org.graalvm.compiler.asm.amd64.AMD64MacroAssembler;
33 import org.graalvm.compiler.bytecode.BytecodeProvider;
34 import org.graalvm.compiler.core.common.CompilationIdentifier;
35 import org.graalvm.compiler.core.common.LIRKind;
36 import org.graalvm.compiler.core.common.spi.ForeignCallDescriptor;
37 import org.graalvm.compiler.core.common.type.DataPointerConstant;
38 import org.graalvm.compiler.debug.DebugContext;
39 import org.graalvm.compiler.hotspot.HotSpotBackend;
40 import org.graalvm.compiler.hotspot.HotSpotForeignCallLinkage;
41 import org.graalvm.compiler.hotspot.meta.HotSpotForeignCallsProviderImpl;
42 import org.graalvm.compiler.hotspot.meta.HotSpotProviders;
43 import org.graalvm.compiler.hotspot.stubs.SnippetStub;
44 import org.graalvm.compiler.lir.LIRInstructionClass;
45 import org.graalvm.compiler.lir.Variable;
46 import org.graalvm.compiler.lir.amd64.AMD64LIRInstruction;
47 import org.graalvm.compiler.lir.asm.ArrayDataPointerConstant;
48 import org.graalvm.compiler.lir.asm.CompilationResultBuilder;
49 import org.graalvm.compiler.lir.gen.LIRGeneratorTool;
50 import org.graalvm.compiler.lir.jtt.LIRTest;
51 import org.graalvm.compiler.lir.jtt.LIRTestSpecification;
52 import org.graalvm.compiler.nodes.StructuredGraph;
53 import org.graalvm.compiler.nodes.extended.ForeignCallNode;
54 import org.graalvm.compiler.nodes.graphbuilderconf.GraphBuilderConfiguration;
55 import org.graalvm.compiler.nodes.graphbuilderconf.GraphBuilderContext;
56 import org.graalvm.compiler.nodes.graphbuilderconf.InvocationPlugin;
57 import org.graalvm.compiler.nodes.graphbuilderconf.InvocationPlugins;
58 import org.graalvm.compiler.options.OptionValues;
59 import org.graalvm.compiler.replacements.ReplacementsImpl;
60 import org.graalvm.compiler.replacements.classfile.ClassfileBytecodeProvider;
61 import org.junit.Assume;
62 import org.junit.Before;
63 import org.junit.Test;
64
65 import jdk.vm.ci.amd64.AMD64;
66 import jdk.vm.ci.amd64.AMD64.CPUFeature;
67 import jdk.vm.ci.amd64.AMD64Kind;
68 import jdk.vm.ci.code.Register;
69 import jdk.vm.ci.code.ValueUtil;
70 import jdk.vm.ci.meta.AllocatableValue;
71 import jdk.vm.ci.meta.MetaAccessProvider;
72 import jdk.vm.ci.meta.ResolvedJavaMethod;
73 import jdk.vm.ci.meta.Value;
74
75 public class StubAVXTest extends LIRTest {
76
77 @Before
78 public void checkAMD64() {
79 Assume.assumeTrue("skipping AMD64 specific test", getTarget().arch instanceof AMD64);
80 Assume.assumeTrue("skipping AVX test", ((AMD64) getTarget().arch).getFeatures().contains(CPUFeature.AVX));
81 if (getBackend() instanceof HotSpotBackend) {
82 HotSpotBackend backend = (HotSpotBackend) getBackend();
83 Assume.assumeTrue("skipping because of MaxVectorSize", backend.getRuntime().getVMConfig().maxVectorSize >= 32);
84 }
85 }
86
87 private static final DataPointerConstant avxConstant = new ArrayDataPointerConstant(new float[]{1f, 2f, 3f, 4f, 5f, 6f, 7f, 8f}, 32);
88
89 private static class LoadAVXConstant extends AMD64LIRInstruction {
90 public static final LIRInstructionClass<LoadAVXConstant> TYPE = LIRInstructionClass.create(LoadAVXConstant.class);
91
92 @Def({REG}) AllocatableValue result;
93
169
170 @Override
171 public void generate(LIRGeneratorTool gen, Value arg0, Value arg1) {
172 Variable ret = gen.newVariable(LIRKind.value(AMD64Kind.DWORD));
173 gen.append(new CompareAVXRegister(ret, gen.asAllocatable(arg0), gen.asAllocatable(arg1), gen.newVariable(LIRKind.value(AMD64Kind.V256_QWORD))));
174 setResult(ret);
175 }
176 };
177
178 private static class TestStub extends SnippetStub {
179
180 TestStub(OptionValues options, HotSpotProviders providers, HotSpotForeignCallLinkage linkage) {
181 super("testStub", options, providers, linkage);
182 }
183
184 @Snippet
185 static void testStub() {
186 }
187
188 @Override
189 protected void registerSnippet() {
190 }
191
192 @Override
193 protected StructuredGraph buildInitialGraph(DebugContext debug, CompilationIdentifier compilationId, Object[] args) {
194 // Build the snippet graph directly since snippet registration is closed at this point.
195 ReplacementsImpl d = (ReplacementsImpl) providers.getReplacements();
196 MetaAccessProvider metaAccess = d.getProviders().getMetaAccess();
197 BytecodeProvider bytecodes = new ClassfileBytecodeProvider(metaAccess, d.snippetReflection, ClassLoader.getSystemClassLoader());
198 return d.makeGraph(debug, bytecodes, method, args, null, false, null);
199 }
200 }
201
202 public static final ForeignCallDescriptor TEST_STUB = new ForeignCallDescriptor("test_stub", void.class);
203
204 @LIRIntrinsic
205 public static int compareAVXRegister(@SuppressWarnings("unused") LIRTestSpecification spec, Object left, Object right) {
206 return left == right ? 0xff : 0;
207 }
208
209 @Override
210 protected GraphBuilderConfiguration editGraphBuilderConfiguration(GraphBuilderConfiguration conf) {
211 InvocationPlugins invocationPlugins = conf.getPlugins().getInvocationPlugins();
212 InvocationPlugins.Registration r = new InvocationPlugins.Registration(invocationPlugins, TestStub.class);
213 r.register0("testStub", new InvocationPlugin() {
214 @Override
215 public boolean apply(GraphBuilderContext b, ResolvedJavaMethod targetMethod, InvocationPlugin.Receiver receiver) {
216 b.add(new ForeignCallNode(getProviders().getForeignCalls(), TEST_STUB));
217 return true;
218 }
|