Skip to content
Permalink
Browse files

Make multiple torch_glow forks (#3992)

Summary:
Pull Request resolved: #3992

Make it possible to import torch_glow python lib for different glow forks.

This should let us `import torch_glow` anywhere and the underlying glow fork can be changed transparently

Reviewed By: bertmaher

Differential Revision: D19326216

fbshipit-source-id: d428137390d7f10ab414f3f3a8a4e937b6196e6f
  • Loading branch information
jackm321 authored and facebook-github-bot committed Jan 10, 2020
1 parent 336ffa8 commit b25af25a5d0e16bb5119c21daa255dcf02ad8343
Showing with 4 additions and 0 deletions.
  1. +4 −0 torch_glow/src/binding.cpp
@@ -33,7 +33,11 @@ namespace py = pybind11;
using namespace glow;

/// The torch_glow pybind11 module.
#ifdef TORCH_GLOW_MODULE_NAME
PYBIND11_MODULE(TORCH_GLOW_MODULE_NAME, m) {
#else
PYBIND11_MODULE(_torch_glow, m) {
#endif
/// Register Glow op and FusionPass, enable the fusion pass if
/// fusionPassEnabled is set in PyTorchLoaderSettings.
registerGlowFusionOpAndPass(

0 comments on commit b25af25

Please sign in to comment.
You can’t perform that action at this time.