mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-07-31 10:52:50 +08:00
Fix distutils failure in Triton Beta testing
Summary: Fixes the distutils issues similar to D73934713 Reviewed By: bottler Differential Revision: D75631611 fbshipit-source-id: 09c354d8cc51ff2c46f4688d7f674370e3f48f1e
This commit is contained in:
parent
3f327a516b
commit
fc08621879
@ -21,8 +21,6 @@ import logging
|
||||
import warnings
|
||||
from collections.abc import Mapping
|
||||
from dataclasses import dataclass, field
|
||||
|
||||
from distutils.version import LooseVersion
|
||||
from typing import Any, Callable, ClassVar, Dict, Iterator, List, Optional, Tuple, Type
|
||||
|
||||
import torch
|
||||
@ -222,7 +220,8 @@ class VoxelGridBase(ReplaceableBase, torch.nn.Module):
|
||||
+ "| 'bicubic' | 'linear' | 'area' | 'nearest-exact'"
|
||||
)
|
||||
|
||||
interpolate_has_antialias = LooseVersion(torch.__version__) >= "1.11"
|
||||
# We assume PyTorch 1.11 and newer.
|
||||
interpolate_has_antialias = True
|
||||
|
||||
if antialias and not interpolate_has_antialias:
|
||||
warnings.warn("Antialiased interpolation requires PyTorch 1.11+; ignoring")
|
||||
|
Loading…
x
Reference in New Issue
Block a user