mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
4972cf06a2
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/41145 **Summary** This commit adds out-of-source-tree tests for `to_backend`. These tests check that a Module can be lowered to a backend, exported, loaded (in both Python and C++) and executed. **Fixes** This commit fixes #40067. Test Plan: Imported from OSS Reviewed By: jamesr66a Differential Revision: D22510076 Pulled By: SplitInfinity fbshipit-source-id: f65964ef3092a095740f06636ed5b1eb0884492d
15 lines
278 B
C++
15 lines
278 B
C++
#include "custom_backend.h"
|
|
|
|
namespace torch {
|
|
namespace custom_backend {
|
|
namespace {
|
|
constexpr auto kBackendName = "custom_backend";
|
|
static auto cls = torch::jit::backend<CustomBackend>(kBackendName);
|
|
}
|
|
|
|
std::string getBackendName() {
|
|
return std::string(kBackendName);
|
|
}
|
|
}
|
|
}
|