Refined peer-access implementation of CPUAccelerator.

m4rs-mt · Jan 26, 2023 · bdb1a03 · bdb1a03
1 parent c3caf01
commit bdb1a03
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/Src/ILGPU/Runtime/CPU/CPUAccelerator.cs b/Src/ILGPU/Runtime/CPU/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.Diagnostics.CodeAnalysis;
@@ -250,12 +251,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);
@@ -266,7 +268,7 @@ protected override void EnablePeerAccessInternal(Accelerator otherAccelerator)
         protected override void DisablePeerAccessInternal(
             Accelerator otherAccelerator) =>
             Debug.Assert(
-                otherAccelerator is CPUAccelerator,
+                CanAccessPeerInternal(otherAccelerator),
                 "Invalid EnablePeerAccess method");
 
         #endregion
@@ -482,7 +484,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)
         {