基于C#反射手动内联方法以适配ILGPU GPU内核

我使用ILGPU在GPU上运行计算密集型代码，通过接口划分模块化类以实现不同的模块化算法。但ILGPU仅允许将静态方法作为GPU内核运行，因此需要提取嵌套类中的IL代码并手动内联，生成可作为内核调用的单个静态方法。

示例代码 #

public interface IFunction
{
    public float DoSum(float[] input);
}

public class NormalSum : IFunction
{
    public float DoSum(float[] input)
    {
        float sum = 0.0f;
        for(int i = 0; i < input.Length; i++)
            sum += input[i];
        return sum;
    }
}

public interface IAlgorithm
{
    public void DoAlgorithm(float[] input, float[] output);
}

public class MyAlgorithm : IAlgorithm
{
    IFunction fun;
    public MyAlgorithm(IFunction fun)
    {
        this.fun = fun;
    }
    public void DoAlgorithm(float[] input, float[] output)
    {
          for(int j = 0; j < output.Length; j++)
          {
               output[j] = 2.0f * fun.DoSum(input);
          }
    }
}

注：修正了原代码中NormalSum类的方法定义错误与Length拼写错误

期望内联结果 #

希望将实例化后的算法逻辑内联为单个静态方法，示例如下：

var staticMethod = Inline(new MyAlgorithm(new NormalSum));

// 生成的IL对应的C#代码
public static void GeneratedCode(float[] input, float[] output)
{
          for(int j = 0; j < output.Length; j++)
          {
              float sum = 0.0f;
              for(int i = 0; i < input.Length; i++)
                  sum += input[i];
              float result = sum;
              output[j] = 2.0f * result;
          }
}

当前尝试的代码 #

目前只能复制单个静态方法的IL，无法深入内部实现内联，尝试代码如下：

MethodInfo existingMethod = typeof(MyStaticClass).GetMethod("MethodToCopy", BindingFlags.Static | BindingFlags.Public);

AssemblyName assemblyName = new AssemblyName("DynamicAssembly");
AssemblyBuilder assemblyBuilder = AppDomain.CurrentDomain.DefineDynamicAssembly(assemblyName, AssemblyBuilderAccess.Run);

ModuleBuilder moduleBuilder = assemblyBuilder.DefineDynamicModule("DynamicModule");

TypeBuilder typeBuilder = moduleBuilder.DefineType("DynamicClass", TypeAttributes.Public);

MethodBuilder methodBuilder = typeBuilder.DefineMethod("DynamicMethod", MethodAttributes.Public | MethodAttributes.Static, existingMethod.ReturnType, new Type[] { });

ILGenerator ilGenerator = methodBuilder.GetILGenerator();

// 复制IL代码
foreach (var instruction in existingMethod.GetMethodBody().GetILAsByteArray())
{
    ilGenerator.Emit(instruction);
}

Type type = typeBuilder.CreateType();

type.GetMethod("DynamicMethod").Invoke(null, null);

内容的提问来源于stack exchange，提问作者CoffeDeveloper