Skip to content

Instantly share code, notes, and snippets.

@YakunZhou
Created February 22, 2022 07:12
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save YakunZhou/7370c732248302e5b20a245370f0450a to your computer and use it in GitHub Desktop.
Save YakunZhou/7370c732248302e5b20a245370f0450a to your computer and use it in GitHub Desktop.
class CpuLocalCallOpKernelInstructionType final : public LocalCallOpKernelInstructionType {
public:
CpuLocalCallOpKernelInstructionType() = default;
~CpuLocalCallOpKernelInstructionType() override = default;
using stream_type = vm::CpuStreamType; // 绑定stream_type
private:
const char* device_tag() const override { return stream_type().device_tag(); }
};
COMMAND(vm::RegisterInstructionType<CpuLocalCallOpKernelInstructionType>("cpu.LocalCallOpKernel"));
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment