Enable allocation tracing for classes owned by vm isolate

runtime/vm/stub_code_x64.cc

Index: runtime/vm/stub_code_x64.cc
diff --git a/runtime/vm/stub_code_x64.cc b/runtime/vm/stub_code_x64.cc
index 4fe0d9bc63dce897a7220b1ce6a9ad07ba1c88b4..d7793d2a34d0e18f63e01a516309ce6f2ec465aa 100644
--- a/runtime/vm/stub_code_x64.cc
+++ b/runtime/vm/stub_code_x64.cc
@@ -593,7 +593,7 @@ void StubCode::GenerateAllocateArrayStub(Assembler* assembler) {
   // Check for allocation tracing.
   __ MaybeTraceAllocation(kArrayCid,
                           &slow_case,
-                          /* near_jump = */ false,
+                          Assembler::kFarJump,
                           /* inline_isolate = */ false);
 
   const intptr_t fixed_size = sizeof(RawArray) + kObjectAlignment - 1;
@@ -848,6 +848,12 @@ void StubCode::GenerateAllocateContextStub(Assembler* assembler) {
     __ leaq(R13, Address(R10, TIMES_8, fixed_size));
     __ andq(R13, Immediate(-kObjectAlignment));
 
+    // Check for allocation tracing.
+    __ MaybeTraceAllocation(kContextCid,
+                            &slow_case,
+                            Assembler::kFarJump,
+                            /* inline_isolate = */ false);
+
     // Now allocate the object.
     // R10: number of context variables.
     const intptr_t cid = kContextCid;
@@ -1054,8 +1060,9 @@ void StubCode::GenerateAllocationStubForClass(
     __ movq(RDX, Address(RSP, kObjectTypeArgumentsOffset));
     // RDX: instantiated type arguments.
   }
+  Isolate* isolate = Isolate::Current();
   if (FLAG_inline_alloc && Heap::IsAllocatableInNewSpace(instance_size) &&
-      !cls.trace_allocation()) {
+      !cls.TraceAllocation(isolate)) {
     Label slow_case;
     // Allocate the object and update top to point to
     // next object start and initialize the allocated object.