Refined peer-access implementation of CPUAccelerator.

m4rs-mt · Nov 26, 2022 · 35e7f72 · 35e7f72
1 parent e13209b
commit 35e7f72
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/Src/ILGPU/Runtime/CPU/CPUAccelerator.cs b/Src/ILGPU/Runtime/CPU/CPUAccelerator.cs
@@ -1,6 +1,6 @@
 // ---------------------------------------------------------------------------------------
 //                                        ILGPU
-//                        Copyright (c) 2016-2021 ILGPU Project
+//                        Copyright (c) 2016-2022 ILGPU Project
 //                                    www.ilgpu.net
 //
 // File: CPUAccelerator.cs
@@ -12,6 +12,7 @@
 using ILGPU.Backends;
 using ILGPU.Backends.IL;
 using ILGPU.Resources;
+using ILGPU.Runtime.Velocity;
 using System;
 using System.Diagnostics;
 using System.Reflection;
@@ -240,12 +241,13 @@ protected override void OnUnbind() { }
 
         /// <summary cref="Accelerator.CanAccessPeerInternal(Accelerator)"/>
         protected override bool CanAccessPeerInternal(Accelerator otherAccelerator) =>
-            otherAccelerator as CPUAccelerator != null;
+            otherAccelerator is CPUAccelerator ||
+            otherAccelerator is VelocityAccelerator;
 
         /// <summary cref="Accelerator.EnablePeerAccessInternal(Accelerator)"/>
         protected override void EnablePeerAccessInternal(Accelerator otherAccelerator)
         {
-            if (otherAccelerator as CPUAccelerator == null)
+            if (!CanAccessPeerInternal(otherAccelerator))
             {
                 throw new InvalidOperationException(
                     RuntimeErrorMessages.CannotEnablePeerAccessToOtherAccelerator);
@@ -256,7 +258,7 @@ protected override void EnablePeerAccessInternal(Accelerator otherAccelerator)
         protected override void DisablePeerAccessInternal(
             Accelerator otherAccelerator) =>
             Debug.Assert(
-                otherAccelerator is CPUAccelerator,
+                CanAccessPeerInternal(otherAccelerator),
                 "Invalid EnablePeerAccess method");
 
         #endregion
@@ -472,7 +474,7 @@ protected override int EstimateGroupSizeInternal(
         #region Page Lock Scope
 
         /// <inheritdoc/>
-        protected unsafe override PageLockScope<T> CreatePageLockFromPinnedInternal<T>(
+        protected override PageLockScope<T> CreatePageLockFromPinnedInternal<T>(
             IntPtr pinned,
             long numElements)
         {