Why Gemfury?
Push, build, and install
RubyGems
npm packages
Python packages
Maven artifacts
PHP packages
Go Modules
Debian packages
RPM packages
NuGet packages
duality-group / torch python
Repository URL to install this package:
|
Version:
1.12.1+cpu ▾
|
#pragma once
#include <ATen/Parallel.h>
#include <c10/core/thread_pool.h>
namespace at {
class TORCH_API PTThreadPool : public c10::ThreadPool {
public:
explicit PTThreadPool(
int pool_size,
int numa_node_id = -1)
: c10::ThreadPool(pool_size, numa_node_id, [](){
c10::setThreadName("PTThreadPool");
at::init_num_threads();
}) {}
};
} // namespace at