Revision 2b3aaa8d6a5bb54a3da4e32395bb80ca09b4e4b3 authored by xndcn on 08 January 2021, 19:46:56 UTC, committed by GitHub on 08 January 2021, 19:46:56 UTC
* Add max threads checking for Metal Originally, this checking will be asserted by Metal API Validation in Xcode, otherwise the program will crash or output wrong results. * Disable the max threads checking for Metal in non-debug runtime * Disable error/metal_threads_too_large test for non-OSX target
1 parent 081f472
PythonExtensionGen.h
#ifndef HALIDE_PYTHON_EXTENSION_GEN_H_
#define HALIDE_PYTHON_EXTENSION_GEN_H_
#include "Module.h"
#include "Target.h"
#include <string>
namespace Halide {
namespace Internal {
class PythonExtensionGen {
public:
PythonExtensionGen(std::ostream &dest);
void compile(const Module &module);
private:
std::ostream &dest;
std::vector<std::string> buffer_refs;
void compile(const LoweredFunc &f);
void convert_buffer(const std::string &name, const LoweredArgument *arg);
void release_buffers(const std::string &prefix);
};
} // namespace Internal
} // namespace Halide
#endif // HALIDE_PYTHON_EXTENSION_GEN_H_
Computing file changes ...
