search

microsoft / triton-shared

Shared Middle-Layer for Triton Compilation
MIT License
132 stars 26 forks source link

Use structured data class for CPU backend target #131

Closed nhat-nguyen closed 2 months ago

nhat-nguyen commented 2 months ago

Update the sample CPU backend to use structured data class (GPUTarget) following https://github.com/openai/triton/commit/5162346487b3e3ebc062d9697429bafad25f22f6.

Fixes #130

nhat-nguyen commented 2 months ago

Yeah this is not ideal but is good to start. It's probably best to have a base class TritonTarget and let GPUTarget and CPUTarget inherit from it instead.