mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
7f1b672b7a
Test Plan: revert-hammer
Differential Revision:
D29952381 (8737e17af2)
Original commit changeset: e60e70b80ccf
fbshipit-source-id: 59dc2f920b7ceaf94ba8f5f36024e7cc710f6645
31 lines
668 B
C++
31 lines
668 B
C++
// (c) Facebook, Inc. and its affiliates. Confidential and proprietary.
|
|
|
|
#pragma once
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace c10 {
|
|
struct IValue;
|
|
}
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
|
|
struct Node;
|
|
class StaticModule;
|
|
|
|
namespace test {
|
|
|
|
// Given a model/function in jit or IR script, run the model/function
|
|
// with the jit interpreter and static runtime, and compare the results
|
|
void testStaticRuntime(
|
|
const std::string& source,
|
|
const std::vector<c10::IValue>& args,
|
|
const std::vector<c10::IValue>& args2 = {},
|
|
const bool use_allclose = false,
|
|
const bool use_equalnan = false);
|
|
|
|
} // namespace test
|
|
} // namespace jit
|
|
} // namespace torch
|