Productive, portable, and performant GPU programming in Python.
#pragma once
#include <memory>
#include "taichi/program/program.h"
namespace taichi::lang {
class TestProgram {
public:
void setup(Arch arch = Arch::x64);
Program *prog() {
return prog_.get();
}
private:
std::unique_ptr<Program> prog_{nullptr};
};
} // namespace taichi::lang
